From de5ac3ef225f657d5a3309184a4201e57b839fcb Mon Sep 17 00:00:00 2001
From: zqc <835569504@qq.com>
Date: Sat, 20 Dec 2025 14:59:15 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E8=AE=BE=E7=BD=AE=E4=BA=BA?=
 =?UTF-8?q?=E8=84=B8=E5=90=8D=E5=8D=95=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/face_recognition_algorithm.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/src/face_recognition_algorithm.py b/src/face_recognition_algorithm.py
index 0d69fce..c9e63c0 100644
--- a/src/face_recognition_algorithm.py
+++ b/src/face_recognition_algorithm.py
@@ -111,6 +111,29 @@ class FaceRecognitionAlgorithm:
         self.det_size = size
         print(f"✅ 检测尺寸设置为: {size}")
 
+    def set_registered_faces(self, registered_faces: Dict[str, np.ndarray]):
+        """
+        直接设置已注册的人脸数据
+        参数: registered_faces - 字典格式 {name: embedding}
+        """
+        if not isinstance(registered_faces, dict):
+            print("❌ 参数必须是字典格式 {name: embedding}")
+            return False
+        
+        # 验证数据格式
+        valid_count = 0
+        for name, embedding in registered_faces.items():
+            if isinstance(embedding, np.ndarray) and embedding.size > 0:
+                valid_count += 1
+        
+        if valid_count == 0:
+            print("❌ 未找到有效的人脸嵌入数据")
+            return False
+        
+        self.registered_faces = registered_faces
+        print(f"✅ 成功设置 {valid_count} 个注册人脸")
+        return True
+
     def load_registered_faces(self, register_dir: str):
         """
         从目录加载注册的人脸图片