Update module

app.py

@@ -146,6 +146,7 @@ def extract_faces(video_file):
         for box in results.boxes:
             # version 1
             # x1, y1, x2, y2 = map(int, box.xyxy[0])
+
             # version 2
             h, w, _ = frame.shape
             x1, y1, x2, y2 = box.xyxy[0].cpu().numpy()
@@ -158,7 +159,11 @@ def extract_faces(video_file):
             y1 = int(max(0, y1 - dy))
             x2 = int(min(w, x2 + dx))
             y2 = int(min(h, y2 + dy))
-            face_crop = frame[y1:y2, x1:x2]
+            # Added for v3
+            shift_down = int(0.1 * (y2 - y1))
+            y1 = int(min(max(0, y1 + shift_down), h))
+            y2 = int(min(max(0, y2 + shift_down), h))
+                face_crop = frame[y1:y2, x1:x2]
             if face_crop.size != 0:
                 resized = cv2.resize(face_crop, (224, 224))
                 frames.append(resized)