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

恢复模型修改

Browse Source
This commit is contained in:
zqc
2026-03-10 10:39:51 +08:00
parent 0c50dfff30
commit d87e960e8b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
biz/checkpoint/checkpoint_biz.py
View File

@@ -17,7 +17,7 @@ from utils.logger import get_logger
from common.constants import MODEL_ROOT_PATH from common.constants import MODEL_ROOT_PATH
logger = get_logger(__name__) logger = get_logger(__name__)
DETECT_MODEL_PATH = 'D:\Python_Save\PoliceProject\Yolo_Weight\Kadian\Kadian_sanshijiazi.onnx' DETECT_MODEL_PATH = 'YOLO_Weight/Kadian.onnx'
# 默认相对ROI（与原文件一致） # 默认相对ROI（与原文件一致）
#ROI_RELATIVE = np.array([ #ROI_RELATIVE = np.array([
@@ -52,7 +52,7 @@ PERSON_CAR_INPUT_SIZE = 640
#POSE_INPUT_SIZE = 640 #POSE_INPUT_SIZE = 640
RTSP_TARGET_FPS = 30.0 RTSP_TARGET_FPS = 10.0
# ========================= Kadian TrafficMonitor（精简版，专为服务设计） ========================= # ========================= Kadian TrafficMonitor（精简版，专为服务设计） =========================
class KadianDetector: class KadianDetector:
@@ -240,7 +240,7 @@ class KadianDetector:
current_time_sec = timestamp current_time_sec = timestamp
print('current_time_sec:', current_time_sec) # print('current_time_sec:', current_time_sec)
# ========= 主检测（删除pose检测）========= # ========= 主检测（删除pose检测）=========
detections = self.detector(frame) detections = self.detector(frame)