yipai-tech/SupervisorAI
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

新增监狱人脸识别的biz

Browse Source
This commit is contained in:
zqc
2026-01-08 16:47:17 +08:00
parent a183d650b3
commit 6679c8ea34
4 changed files with 304 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

211
api/routes/algorithm_router.py
View File

@@ -20,6 +20,7 @@ from algorithm.face_recognition_algorithm import FaceRecognitionAlgorithm
from biz.base_face_biz import BaseFaceBiz
from biz.video_check_biz import VideoCheckBiz
from biz.video_face_biz import VideoFaceBiz
from biz.video_face_prison_biz import VideoFacePrisonBiz
from repositories.video_check_repository import VideoCheckTaskRepository
# 创建路由器
@@ -34,6 +35,9 @@ face_algorithm_for_rtsp = FaceRecognitionAlgorithm(use_gpu=settings.FACE_USE_GPU
# 初始化RTSP专用VideoFaceBiz实例
video_face_biz = VideoFaceBiz(face_algorithm_for_rtsp.get_app())
# 初始化RTSP专用VideoFacePrisonBiz实例
video_face_prison_biz = VideoFacePrisonBiz(face_algorithm_for_rtsp.get_app())
logger = logging.getLogger(__name__)
@@ -827,5 +831,210 @@ async def get_videofacebiz_status():
logger.error(f"获取VideoFaceBiz状态失败: {e}")
raise HTTPException(status_code=500, detail=f"获取状态失败: {str(e)}")
def sync_videofaceprisonbiz_params():
"""
同步VideoFacePrisonBiz的参数
"""
try:
with db_manager.get_session() as session:
# 查询人脸识别配置（根据实际表结构）
config_records = session.query(SurConfigBase).filter(
SurConfigBase.config_type == settings.SUR_CONFIG_TYPE_FACE
).all()
# 构建配置参数字典
config_params = {}
for record in config_records:
if record.config_key and record.config_value:
config_params[record.config_key] = record.config_value
# 配置键映射关系
config_mapping = {
"face.list_mode": "list_mode",
"face.clarity_threshold": "clarity_threshold",
"face.min_face_size": "min_face_size",
"face.pitch_threshold": "pitch_threshold",
"face.yaw_threshold": "yaw_threshold",
"face.similarity_threshold": "similarity_threshold"
}
updated_count = 0
for config_key, param_name in config_mapping.items():
if config_key in config_params:
config_value = config_params[config_key]
# 根据参数类型进行转换和设置
if param_name == "list_mode":
if config_value in ["0", "1"]:
video_face_prison_biz.set_list_mode(config_value)
updated_count += 1
elif param_name == "clarity_threshold":
try:
threshold = float(config_value)
video_face_prison_biz.set_clarity_threshold(threshold)
updated_count += 1
except ValueError:
logger.error(f"无效的清晰度阈值: {config_value}")
elif param_name == "min_face_size":
try:
size = int(config_value)
video_face_prison_biz.set_min_face_size(size)
updated_count += 1
except ValueError:
logger.error(f"无效的最小人脸尺寸: {config_value}")
elif param_name == "pitch_threshold":
try:
threshold = float(config_value)
video_face_prison_biz.set_pitch_threshold(threshold)
updated_count += 1
except ValueError:
logger.error(f"无效的俯仰角阈值: {config_value}")
elif param_name == "yaw_threshold":
try:
threshold = float(config_value)
video_face_prison_biz.set_yaw_threshold(threshold)
updated_count += 1
except ValueError:
logger.error(f"无效的偏航角阈值: {config_value}")
elif param_name == "similarity_threshold":
try:
threshold = float(config_value)
video_face_prison_biz.set_similarity_threshold(threshold)
updated_count += 1
except ValueError:
logger.error(f"无效的相似度阈值: {config_value}")
logger.info(f"✅ 同步VideoFacePrisonBiz参数完成，更新了 {updated_count} 个参数")
return updated_count
except Exception as e:
logger.error(f"❌ 同步VideoFacePrisonBiz参数失败: {e}")
return 0
def sync_videofaceprisonbiz_blacklist():
"""
同步VideoFacePrisonBiz的黑名单
"""
try:
with db_manager.get_session() as session:
# 查询启用的黑名单人员
blacklist_persons = session.query(SurPersonBlacklist).filter(
SurPersonBlacklist.status == 1
).all()
if not blacklist_persons:
logger.info("⚠️ 黑名单为空，清空当前黑名单")
video_face_prison_biz.set_registered_faces({})
return 0
person_ids = [person.person_id for person in blacklist_persons]
# 查询对应的人脸特征
face_features = session.query(SurFaceFeature).filter(
SurFaceFeature.person_id.in_(person_ids),
SurFaceFeature.feature_type == settings.FACE_MODEL_VERSION,
SurFaceFeature.status == 2 # 计算成功的特征
).all()
# 构建特征字典
registered_faces = {}
loaded_count = 0
for feature in face_features:
if feature.feature_data:
try:
# 将bytea转换为numpy数组
import numpy as np
feature_array = np.frombuffer(feature.feature_data, dtype=np.float32)
# 使用person_id作为标识符
person_name = f"blacklist_{feature.person_id}"
registered_faces[person_name] = feature_array
loaded_count += 1
except Exception as e:
logger.error(f"❌ 解析黑名单人员 {feature.person_id} 的特征数据失败: {e}")
continue
# 设置黑名单
success = video_face_prison_biz.set_registered_faces(registered_faces)
if success:
logger.info(f"✅ 同步黑名单完成，加载了 {loaded_count} 个黑名单人员")
else:
logger.error("❌ 设置黑名单失败")
return loaded_count
except Exception as e:
logger.error(f"❌ 同步黑名单失败: {e}")
return 0
@router.post("/sync-videofaceprisonbiz-params", summary="同步VideoFacePrisonBiz参数")
async def sync_videofaceprisonbiz_params_endpoint():
"""
同步VideoFacePrisonBiz的参数
从sur_config表同步参数到VideoFacePrisonBiz实例
"""
try:
updated_count = sync_videofaceprisonbiz_params()
return {
"success": True,
"message": f"同步参数完成，更新了 {updated_count} 个参数",
"updated_count": updated_count
}
except Exception as e:
logger.error(f"同步VideoFacePrisonBiz参数失败: {e}")
raise HTTPException(status_code=500, detail=f"同步参数失败: {str(e)}")
@router.post("/sync-videofaceprisonbiz-blacklist", summary="同步VideoFacePrisonBiz黑名单")
async def sync_videofaceprisonbiz_blacklist_endpoint():
"""
同步VideoFacePrisonBiz的黑名单
从sur_person_blacklist表同步黑名单到VideoFacePrisonBiz实例
"""
try:
loaded_count = sync_videofaceprisonbiz_blacklist()
return {
"success": True,
"message": f"同步黑名单完成，加载了 {loaded_count} 个黑名单人员",
"loaded_count": loaded_count
}
except Exception as e:
logger.error(f"同步VideoFacePrisonBiz黑名单失败: {e}")
raise HTTPException(status_code=500, detail=f"同步黑名单失败: {str(e)}")
@router.get("/videofaceprisonbiz-status", summary="获取VideoFacePrisonBiz状态")
async def get_videofaceprisonbiz_status():
"""
获取VideoFacePrisonBiz的当前状态
"""
try:
status = {
"list_mode": video_face_prison_biz.get_list_mode(),
"clarity_threshold": video_face_prison_biz.get_clarity_threshold(),
"min_face_size": video_face_prison_biz.get_min_face_size(),
"pitch_threshold": video_face_prison_biz.get_pitch_threshold(),
"yaw_threshold": video_face_prison_biz.get_yaw_threshold(),
"similarity_threshold": video_face_prison_biz.get_similarity_threshold(),
"blacklist_count": video_face_prison_biz.get_registered_face_count()
}
return {
"success": True,
"data": status
}
except Exception as e:
logger.error(f"获取VideoFacePrisonBiz状态失败: {e}")
raise HTTPException(status_code=500, detail=f"获取状态失败: {str(e)}")
# 导出路由器
__all__ = ["router"]
__all__ = ["router", "sync_videofacebiz_params", "sync_videofacebiz_blacklist", "sync_videofaceprisonbiz_params", "sync_videofaceprisonbiz_blacklist"]