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

隐藏视频右上角提示

Browse Source
This commit is contained in:
Qinchuanqi
2026-03-07 12:33:14 +08:00
parent 6a1013d138
commit 3ea6a48652
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
biz/checkpoint/checkpoint_biz.py
View File

@@ -18,6 +18,7 @@ logger = get_logger(__name__)
# ========================= 配置区 ========================= # ========================= 配置区 =========================
# Kadian 模型路径与ROI（可根据实际情况修改） # Kadian 模型路径与ROI（可根据实际情况修改）
DETECT_MODEL_PATH = 'YOLO_Weight/Kadian.onnx' 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' #POSE_MODEL_PATH = 'YOLO_Weight/yolov8l-pose.onnx'
# 默认相对ROI（与原文件一致） # 默认相对ROI（与原文件一致）
@@ -435,7 +436,7 @@ class KadianDetector:
'time': current_time_sec, 'time': current_time_sec,
'action': "Unchecked Trunk", '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 alert_offset += 100
# C. 显示 Ignore (通过过快) # C. 显示 Ignore (通过过快)
@@ -450,7 +451,7 @@ class KadianDetector:
'time': current_time_sec, 'time': current_time_sec,
'action': "Ignore", '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 alert_offset += 100
# D. 显示警察在场状态 (Nobody/Only One) # D. 显示警察在场状态 (Nobody/Only One)