新增设置人脸名单接口

This commit is contained in:
zqc
2025-12-20 14:59:15 +08:00
parent 3859fe13e4
commit de5ac3ef22

View File

@@ -111,6 +111,29 @@ class FaceRecognitionAlgorithm:
self.det_size = size self.det_size = size
print(f"✅ 检测尺寸设置为: {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): def load_registered_faces(self, register_dir: str):
""" """
从目录加载注册的人脸图片 从目录加载注册的人脸图片