continue if face recognition did not find face

kex0 · Jul 2, 2023 · 872f2cf · 872f2cf
1 parent ed8b334
commit 872f2cf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/roop/processors/frame/face_swapper.py b/roop/processors/frame/face_swapper.py
@@ -189,6 +189,8 @@ def process_recognition(temp_frame_paths: List[str]) -> dict:
         face_locations = sorted(face_locations, key=lambda x: x[1])
         face_encodings = face_recognition.face_encodings(temp_frame, face_locations)
 
+        if len(face_encodings) == 0:
+            matches_dict[temp_frame_path] = None
         for i, face_encoding in enumerate(face_encodings):
             matches = face_recognition.compare_faces([reference_encoding], face_encoding)
             if True in matches: