Update video_app.py

video_app.py

@@ -15,8 +15,7 @@ import matplotlib.pyplot as plt
 import utils
 from utils import app
 
-# Function to perform face recognition in the video
-def face_recognition_in_video(video_path, names, embeddings):
+def face_recognition_in_video(video_path, output_path, names, embeddings):
     # Open the video file
     cap = cv2.VideoCapture(video_path)
 
@@ -27,13 +26,8 @@ def face_recognition_in_video(video_path, names, embeddings):
 
     # Define the codec and create VideoWriter object
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-    output_path = "/tmp/output_video.mp4"
     out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
 
-    # Initialize FaceAnalysis app
-    app = FaceAnalysis(name='buffalo_l')
-    app.prepare(ctx_id=0, det_size=(640, 640))
-
     # Process each frame of the video
     while cap.isOpened():
         ret, frame = cap.read()
@@ -49,7 +43,7 @@ def face_recognition_in_video(video_path, names, embeddings):
             detected_embedding = face.normed_embedding
 
             # Calculate similarity scores with known embeddings
-            scores = np.dot(detected_embedding, np.array(embeddings).T)
+            scores = np.dot(detected_embedding, embeddings.T)
             scores = np.clip(scores, 0., 1.)
 
             # Find the index with the highest score
@@ -75,6 +69,4 @@ def face_recognition_in_video(video_path, names, embeddings):
     # Release everything if job is finished
     cap.release()
     out.release()
-
     return output_path
-