From e84b22cd0a6835c4aa9146a4cc0f1dd69eb21669 Mon Sep 17 00:00:00 2001
From: Qinchuanqi <2254943770@qq.com>
Date: Thu, 5 Mar 2026 12:59:22 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=A3=80=E6=B5=8B=E6=A1=86?=
 =?UTF-8?q?=E7=9A=84=E5=B0=BA=E5=AF=B8;=E4=BF=AE=E6=94=B9YOLO=E6=A3=80?=
 =?UTF-8?q?=E6=B5=8B=E7=9A=84=E9=98=88=E5=80=BC;=E4=BF=AE=E6=94=B9?=
 =?UTF-8?q?=E5=90=8E=E5=A4=87=E7=AE=B1=E7=9A=84=E6=A3=80=E6=B5=8B=E9=98=88?=
 =?UTF-8?q?=E5=80=BC=E8=A6=81=E6=B1=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 biz/checkpoint/checkpoint_biz.py | 33 ++++++++++++++++++++------------
 1 file changed, 21 insertions(+), 12 deletions(-)

diff --git a/biz/checkpoint/checkpoint_biz.py b/biz/checkpoint/checkpoint_biz.py
index 13d88b9..d024c03 100644
--- a/biz/checkpoint/checkpoint_biz.py
+++ b/biz/checkpoint/checkpoint_biz.py
@@ -28,13 +28,22 @@ DETECT_MODEL_PATH = 'YOLO_Weight/Kadian.onnx'
 #    [0.3411458333333333, 0.9861111111111112]
 #])
 
+# ROI_RELATIVE=np.array([
+#   [0.12,0.0],
+#   [0.3,0.0],
+#   [0.5,0.2],
+#   [1.0, 0.95],
+#   [1.0,1.0],
+#   [0.42,1.0]
+# ])
+
 ROI_RELATIVE=np.array([
-  [0.12,0.0],
-  [0.3,0.0],
-  [0.5,0.2],
-  [1.0, 0.95],
-  [1.0,1.0],
-  [0.42,1.0]
+  [0.15,0.15],
+  [0.37,0.15],
+  [0.55,0.2],
+
+  [0.9,0.85],
+  [0.35,0.85]
 ])
 
 
@@ -56,14 +65,14 @@ class KadianDetector:
         # 模型加载
         self.detector = YOLOv8_ONNX(
             DETECT_MODEL_PATH,
-            conf_threshold=0.25,
-            iou_threshold=0.45,
+            conf_threshold=0.6,
+            iou_threshold=0.65,
             input_size=PERSON_CAR_INPUT_SIZE
         )
 
         # 跟踪器配置
         class TrackerArgs:
-            track_thresh = 0.3  # 必须大于等于yolo的conf_threshold
+            track_thresh = 0.61  # 必须大于等于yolo的conf_threshold
             track_buffer = 40
             match_thresh = 0.85
             mot20 = True
@@ -79,7 +88,7 @@ class KadianDetector:
 
         # ===================== 超参数设置 (仅保留车/后备箱相关) =====================
         # 后备箱检查判定阈值
-        self.TIME_THRESHOLD_TRUNK_OPEN = 0.1
+        self.TIME_THRESHOLD_TRUNK_OPEN = 0.3
         # 车辆最小停留时间阈值 (小于此时间视为无人检查/直接通过)
         self.TIME_THRESHOLD_CAR_MIN_DURATION = 3.0
         # Car 丢帧/ID维持缓冲
@@ -88,8 +97,8 @@ class KadianDetector:
         # police丢失阈值
         self.TIME_TOLERANCE_POLICE = 3.0
         # police状态判定阈值 (累计秒数)
-        self.TIME_THRESHOLD_NOBODY = 5.0
-        self.TIME_THRESHOLD_ONLY_ONE = 5.0
+        self.TIME_THRESHOLD_NOBODY = 10.0
+        self.TIME_THRESHOLD_ONLY_ONE = 10.0
 
         # --- 计算对应的帧数阈值 ---
         self.frame_thresh_trunk_valid = int(self.TIME_THRESHOLD_TRUNK_OPEN * self.fps)