Update module

app.py

@@ -61,7 +61,6 @@ import tempfile
 from ultralytics import YOLO
 from moviepy import ImageSequenceClip
 from scipy.io import wavfile
-from PIL import Image, ImageOps
 
 # Load face detector
 model = YOLO("yolov8n-face.pt").cuda()  # assumes CUDA available
@@ -167,18 +166,16 @@ def extract_faces(video_file):
             face_crop = frame[y1:y2, x1:x2]
             if face_crop.size != 0:
                 #resized = cv2.resize(face_crop, (224, 224))
-                def pad_to_square(img):
-                    h, w = img.shape[:2]
-                    size = max(h, w)
-                    pad_top = (size - h) // 2
-                    pad_bottom = size - h - pad_top
-                    pad_left = (size - w) // 2
-                    pad_right = size - w - pad_left
-                    return cv2.copyMakeBorder(img, pad_top, pad_bottom, pad_left, pad_right, borderType=cv2.BORDER_CONSTANT, value=0)
-
-                square_crop = pad_to_square(face_crop)
-                resized = cv2.resize(square_crop, (224, 224))
-                frames.append(resized)
+                #frames.append(resized)
+                # Center crop 224x224 without resize
+                cy, cx = (y2 - y1) // 2, (x2 - x1) // 2
+                half = 112
+                start_y = max(0, cy - half)
+                start_x = max(0, cx - half)
+                center_crop = face_crop[start_y:start_y + 224, start_x:start_x + 224]
+
+                if center_crop.shape[0] == 224 and center_crop.shape[1] == 224:
+                    frames.append(center_crop)
                 break  # only one face per frame
 
     cap.release()