Update utils.py
Browse files
utils.py
CHANGED
@@ -39,6 +39,7 @@ def add_timecode_to_image_body(image, timecode):
|
|
39 |
|
40 |
def create_annotated_video(video_path, df, mse_embeddings, largest_cluster, output_path):
|
41 |
import cv2
|
|
|
42 |
from facenet_pytorch import MTCNN
|
43 |
video = cv2.VideoCapture(video_path)
|
44 |
fps = video.get(cv2.CAP_PROP_FPS)
|
|
|
39 |
|
40 |
def create_annotated_video(video_path, df, mse_embeddings, largest_cluster, output_path):
|
41 |
import cv2
|
42 |
+
import torch
|
43 |
from facenet_pytorch import MTCNN
|
44 |
video = cv2.VideoCapture(video_path)
|
45 |
fps = video.get(cv2.CAP_PROP_FPS)
|