45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
|
from core.ocr import load_model as ocrLoadModel, detect as ocrDetect
|
|||
|
|
from core.face import load_model as faceLoadModel, detect as faceDetect
|
|||
|
|
from core.yolo import load_model as yoloLoadModel, detect as yoloDetect
|
|||
|
|
|
|||
|
|
# 添加一个标记变量，用于监控load_model是否已被调用
|
|||
|
|
_model_loaded = False
|
|||
|
|
|
|||
|
|
|
|||
|
|
def load_model():
|
|||
|
|
global _model_loaded
|
|||
|
|
|
|||
|
|
# 如果已经调用过，直接忽略
|
|||
|
|
if _model_loaded:
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 首次调用时加载模型
|
|||
|
|
ocrLoadModel()
|
|||
|
|
faceLoadModel()
|
|||
|
|
yoloLoadModel()
|
|||
|
|
|
|||
|
|
# 标记为已调用
|
|||
|
|
_model_loaded = True
|
|||
|
|
|
|||
|
|
|
|||
|
|
def detect(frame):
|
|||
|
|
# 先进行YOLO检测
|
|||
|
|
yolo_flag, yolo_result = yoloDetect(frame)
|
|||
|
|
print("YOLO检测结果：", yolo_result)
|
|||
|
|
if yolo_flag:
|
|||
|
|
return (True, yolo_result, "yolo")
|
|||
|
|
|
|||
|
|
# YOLO未检测到，进行人脸检测
|
|||
|
|
face_flag, face_result = faceDetect(frame)
|
|||
|
|
print("人脸检测结果：", face_result)
|
|||
|
|
if face_flag:
|
|||
|
|
return (True, face_result, "face")
|
|||
|
|
|
|||
|
|
# 人脸未检测到，进行OCR检测
|
|||
|
|
ocr_flag, ocr_result = ocrDetect(frame)
|
|||
|
|
print("OCR检测结果：", ocr_result)
|
|||
|
|
if ocr_flag:
|
|||
|
|
return (True, ocr_result, "ocr")
|
|||
|
|
|
|||
|
|
# 所有检测都未检测到
|
|||
|
|
return (False, "未检测到任何内容", "none")
|