Update utils.py

utils.py

@@ -27,7 +27,6 @@ def add_timecode_to_image(image, timecode):
     draw.text((10, 10), timecode, (255, 0, 0), font=font)
     return np.array(img_pil)
     
-
 def add_timecode_to_image_body(image, timecode):
     from PIL import Image, ImageDraw, ImageFont
     import numpy as np
@@ -40,6 +39,7 @@ def add_timecode_to_image_body(image, timecode):
 
 def create_annotated_video(video_path, df, mse_embeddings, largest_cluster, output_path):
     import cv2
+    from facenet_pytorch import MTCNN
     video = cv2.VideoCapture(video_path)
     fps = video.get(cv2.CAP_PROP_FPS)
     width = int(video.get(cv2.CAP_PROP_FRAME_WIDTH))
@@ -52,6 +52,7 @@ def create_annotated_video(video_path, df, mse_embeddings, largest_cluster, outp
         if not ret:
             break
         # Detect face and draw bounding box
+        mtcnn = MTCNN(keep_all=False, device=device, thresholds=[0.9, 0.9, 0.9], min_face_size=50)
         boxes, _ = mtcnn.detect(frame)
         if boxes is not None and len(boxes) > 0:
             box = boxes[0]