MasterThesis/UNet/utils.py

11 lines
251 B
Python
Raw Permalink Normal View History

2024-04-15 13:46:41 +02:00
from PIL import Image
def keep_image_size_open(path,size=(256,256)):
img = Image.open(path)
tmp = max(img.size)
mask = Image.new('RGB', (tmp, tmp),(0,0,0))
mask.paste(img,(0,0))
mask = mask.resize(size)
return mask