Upload crop.py

crop.py

@@ -3,7 +3,8 @@ from pathlib import Path
 from PIL import Image
 import numpy as np
 
-def crop(img, point, mode=0, size=(512, 512), box=None, face_ratio=3, shreshold=1.5):
+def crop(image_path, point, mode=0, size=(512, 512), box=None, face_ratio=3, shreshold=1.5):
+    img = Image.open(image_path)
     img_width, img_height = img.size
     tgt_width, tgt_height = size
     point = (point[0]*img_width, point[1]*img_height)
@@ -81,4 +82,4 @@ def crop(img, point, mode=0, size=(512, 512), box=None, face_ratio=3, shreshold=
 
     cropped_img = img.crop((left, top, right, bottom))
     cropped_img = cropped_img.resize(size)
-    return np.ndarray(cropped_img)
+    return cropped_img