diff --git a/biz/checkpoint/checkpoint_biz.py b/biz/checkpoint/checkpoint_biz.py index d024c03..d9829f3 100644 --- a/biz/checkpoint/checkpoint_biz.py +++ b/biz/checkpoint/checkpoint_biz.py @@ -18,6 +18,7 @@ logger = get_logger(__name__) # ========================= 配置区 ========================= # Kadian 模型路径与ROI(可根据实际情况修改) DETECT_MODEL_PATH = 'YOLO_Weight/Kadian.onnx' +#DETECT_MODEL_PATH = r'D:\Python_Save\PoliceProject\TestForLiveStream\AIMonitor_KaDian_0304\YOLO_Weight\Kadian.onnx' #POSE_MODEL_PATH = 'YOLO_Weight/yolov8l-pose.onnx' # 默认相对ROI(与原文件一致) @@ -435,7 +436,7 @@ class KadianDetector: 'time': current_time_sec, 'action': "Unchecked Trunk", }) - self.draw_alert(frame, alert_text, (0, 0, 255), offset_y=alert_offset) + #self.draw_alert(frame, alert_text, (0, 0, 255), offset_y=alert_offset) alert_offset += 100 # C. 显示 Ignore (通过过快) @@ -450,7 +451,7 @@ class KadianDetector: 'time': current_time_sec, 'action': "Ignore", }) - self.draw_alert(frame, alert_text, (0, 0, 255), offset_y=alert_offset) + #self.draw_alert(frame, alert_text, (0, 0, 255), offset_y=alert_offset) alert_offset += 100 # D. 显示警察在场状态 (Nobody/Only One)