zzx/video
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

目前可以成功动态更换模型运行的

Browse Source
This commit is contained in:
ninghongbin
2025-09-12 14:05:09 +08:00
parent 435b2a0e6c
commit 4be7f7bf14
13 changed files with 1518 additions and 325 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
core/all.py
View File

@ -57,50 +57,39 @@ def save_db(model_type, client_ip, result):
# 修正后的 detect 函数关键部分
def detect(client_ip, frame):
"""
执行模型检测，检测到违规时按指定格式保存图片
参数：
frame: 待检测的图像帧（OpenCV格式，numpy.ndarray类型）
返回：
(检测结果布尔值, 检测详情, 检测模型类型)
"""
# 1. YOLO检测（优先级1）
# 1. YOLO检测
yolo_flag, yolo_result = yoloDetect(frame)
print(f"YOLO检测结果：{yolo_result}")
if yolo_flag:
# model_type 传入 "yolo"（正确）
full_save_path, display_path = get_image_save_path(model_type="yolo", client_ip=client_ip)
if full_save_path: # 只判断完整路径是否有效（用于保存）
if full_save_path:
cv2.imwrite(full_save_path, frame)
# 打印时使用「显示用短路径」，符合需求格式
print(f"✅ YOLO违规图片已保存：{display_path}")
print(f"✅ yolo违规图片已保存：{display_path}") # 日志也修正
save_db(model_type="yolo", client_ip=client_ip, result=str(full_save_path))
return (True, yolo_result, "yolo")
#
# # 2. 人脸检测（优先级2）
# 2. 人脸检测
face_flag, face_result = faceDetect(frame)
print(f"人脸检测结果：{face_result}")
if face_flag:
full_save_path, display_path = get_image_save_path(model_type="yolo", client_ip=client_ip)
if full_save_path: # 只判断完整路径是否有效（用于保存）
full_save_path, display_path = get_image_save_path(model_type="face", client_ip=client_ip) # 这里改了
if full_save_path:
cv2.imwrite(full_save_path, frame)
# 打印时使用「显示用短路径」，符合需求格式
print(f"✅ face违规图片已保存：{display_path}")
print(f"✅ face违规图片已保存：{display_path}") # 日志也修正
save_db(model_type="face", client_ip=client_ip, result=str(full_save_path))
return (True, face_result, "face")
# 3. OCR检测（优先级3）
# 3. OCR检测
ocr_flag, ocr_result = ocrDetect(frame)
print(f"OCR检测结果：{ocr_result}")
if ocr_flag:
# 解构元组，保存用完整路径，打印用短路径
full_save_path, display_path = get_image_save_path(model_type="yolo", client_ip=client_ip)
if full_save_path: # 只判断完整路径是否有效（用于保存）
full_save_path, display_path = get_image_save_path(model_type="ocr", client_ip=client_ip) # 这里改了
if full_save_path:
cv2.imwrite(full_save_path, frame)
# 打印时使用「显示用短路径」，符合需求格式
print(f"✅ ocr违规图片已保存：{display_path}")
print(f"✅ ocr违规图片已保存：{display_path}") # 日志也修正
save_db(model_type="ocr", client_ip=client_ip, result=str(full_save_path))
return (True, ocr_result, "ocr")
# 4. 无违规内容（不保存图片）
print(f"❌ 未检测到任何违规内容，不保存图片")
return (False, "未检测到任何内容", "none")