feat: automatic image croping to match aspect ratio

This commit is contained in:
Yannik Schmidt
2024-03-24 17:29:26 +01:00
parent c3126e3b22
commit eeeee4c94e
3 changed files with 22 additions and 11 deletions

View File

@@ -1,11 +1,20 @@
import PIL
def load_and_keep_aspect_rescale(path, target_x, target_y):
'''Load an image and resize it while keeping the aspect ratio (crop if it doesnt fit)'''
def smart_resize(img, height, width):
'''Crop to aspect ratio and then resize'''
img = PIL.Image.open(path)
# get size
# img.size()
current_height, current_width = img.size
target_ratio = height/width
current_ratio = current_height/current_width
img = img.resize((x-2*30, y-2*30))
img = PIL.ImageTk.PhotoImage(img)
if current_ratio > target_ratio:
target_crop_height = current_width*target_ratio
diff = current_height - target_crop_height
img = img.crop((diff/2, 0, current_height-diff/2, current_width))
elif current_ratio < target_ratio:
target_crop_width = current_height/target_ratio
diff = current_width - target_crop_width
img = img.crop((0, diff/2, current_height, current_width-diff/2))
img = img.resize((height, width))
return img