新增设置人脸名单接口
This commit is contained in:
@@ -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):
|
||||||
"""
|
"""
|
||||||
从目录加载注册的人脸图片
|
从目录加载注册的人脸图片
|
||||||
|
|||||||
Reference in New Issue
Block a user