在detector创建的时候,传入参数

This commit is contained in:
zqc
2026-02-26 14:48:06 +08:00
parent 17e131a0c6
commit 51125f354a

View File

@@ -47,7 +47,7 @@ RTSP_TARGET_FPS = 10.0
# ========================= Kadian TrafficMonitor精简版专为服务设计 =========================
class KadianDetector:
def __init__(self, roi_points=ROI_RELATIVE):
def __init__(self, roi_points=ROI_RELATIVE, params=None):
# 模型加载
self.detector = YOLOv8_ONNX(DETECT_MODEL_PATH, conf_threshold=0.25, iou_threshold=0.45,
input_size=PERSON_CAR_INPUT_SIZE)
@@ -70,6 +70,9 @@ class KadianDetector:
#self.roi_points = roi_points.astype(np.int32)
self.roi_points = np.array(roi_points, dtype=np.float64) if roi_points is not None else None
# 摄像头额外参数
self.params = params if params is not None else {}
# ==========================================
# 超参数设置 (Hyperparameters)
# ==========================================
@@ -733,7 +736,9 @@ class FrameProcessorWorker(threading.Thread):
# 获取检测器实例
if cam_id not in self.kadian_detectors:
self.kadian_detectors[cam_id] = KadianDetector()
camera_config = self.cameras.get(cam_id)
params = camera_config.params if camera_config else None
self.kadian_detectors[cam_id] = KadianDetector(params=params)
detector = self.kadian_detectors[cam_id]
# 执行检测