Spaces:
Sleeping
Sleeping
import smartcrop | |
from PIL import Image | |
def crop_image(image, width=224, height=224): | |
new_image = Image.new('RGB', image.size) | |
new_image.paste(image) | |
image = new_image | |
sc = smartcrop.SmartCrop() | |
result = sc.crop(image, width=100, height=int(width / height * 100)) | |
box = ( | |
result['top_crop']['x'], | |
result['top_crop']['y'], | |
result['top_crop']['width'] + result['top_crop']['x'], | |
result['top_crop']['height'] + result['top_crop']['y'] | |
) | |
cropped_image = image.crop(box) | |
cropped_image.thumbnail((width, height), Image.ANTIALIAS) | |
return cropped_image |