diff --git a/biz/prison/supervision_room_biz.py b/biz/prison/supervision_room_biz.py index 45629e2..ca93869 100644 --- a/biz/prison/supervision_room_biz.py +++ b/biz/prison/supervision_room_biz.py @@ -25,8 +25,8 @@ from utils.logger import get_logger logger = get_logger(__name__) # ========================= 配置区 ========================= -Person_Phone_Model = r'Yolo_Weight/person_phone_model.onnx' # 人和手机的检测模型 -Smoke_Model = r'Yolo_Weight/smoke_model.onnx' # 抽烟检测模型 +Person_Phone_Model = r'YOLO_Weight/person_phone_model.onnx' # 人和手机的检测模型 +Smoke_Model = r'YOLO_Weight/smoke_model.onnx' # 抽烟检测模型 person_phone_input_size = 1280 # 模型输入尺寸,与训练时的模型一致 smoke_input_size = 1280 # 模型输入尺寸,与训练时的模型一致