Merge pull request #77 from LimaoC/mitch-robuster-register

LimaoC · Oct 6, 2024 · 9a04a16 · 9a04a16
2 parents 337156f + d4c022f
commit 9a04a16
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/client/models/face_recognition/recognition.py b/client/models/face_recognition/recognition.py
@@ -67,6 +67,11 @@ def register_faces(user_id: int, faces: list[np.ndarray]) -> int:
         face_embedding = all_faces_embed[0]
         face_embeddings.append(face_embedding)
 
+    # Ensure that all images contain the same face
+    matches = face_recognition.compare_faces(face_embeddings[1:], face_embeddings[0])
+    if not all(matches):
+        return Status.TOO_MANY_FACES.value
+
     register_face_embeddings(user_id, face_embeddings)
 
     return Status.OK.value