Update module

app.py

@@ -150,7 +150,7 @@ def extract_faces(video_file):
             # version 2
             h, w, _ = frame.shape
             x1, y1, x2, y2 = box.xyxy[0].cpu().numpy()
-            pad_ratio = 0.5  # 30% padding
+            pad_ratio = 0.3  # 30% padding
 
             dx = (x2 - x1) * pad_ratio
             dy = (y2 - y1) * pad_ratio
@@ -167,19 +167,15 @@ def extract_faces(video_file):
             if face_crop.size != 0:
                 #resized = cv2.resize(face_crop, (224, 224))
                 #frames.append(resized)
+
                 h_crop, w_crop = face_crop.shape[:2]
-                cy, cx = h_crop // 2, w_crop // 2
-                half = 112
-
-                start_y = max(0, cy - half)
-                end_y = start_y + 224
-                start_x = max(0, cx - half)
-                end_x = start_x + 224
-
-                # Make sure we don't go out of bounds
-                if end_y <= h_crop and end_x <= w_crop:
-                    center_crop = face_crop[start_y:end_y, start_x:end_x]
-                    frames.append(center_crop)
+                side = min(h_crop, w_crop)
+                start_y = (h_crop - side) // 2
+                start_x = (w_crop - side) // 2
+                square_crop = face_crop[start_y:start_y+side, start_x:start_x+side]
+                resized = cv2.resize(square_crop, (224, 224))
+                frames.append(resized)
+
                 break  # only one face per frame
 
     cap.release()