dsh/dual_camera_system/config/detection.py

"""
检测配置
"""

# 检测配置
DETECTION_CONFIG = {
    'target_classes': ['person'],   # 检测目标类别
    'confidence_threshold': 0.5,     # 置信度阈值
    'detection_interval': 0.1,       # 检测间隔(秒)
}

# 安全检测模型配置
SAFETY_DETECTION_CONFIG = {
    # 模型路径 - 支持三种格式:
    # - YOLO: .pt 文件, 使用 ultralytics
    # - RKNN: .rknn 文件, 使用 rknnlite (RK3588 平台)
    # - ONNX: .onnx 文件, 使用 onnxruntime
    'model_path': '/Users/wenhongquan/Desktop/阿里云同步/项目/dnn/德胜河 AI/dsh/testrk3588/yolo11n_rk3588.pt',
    
    'model_type': 'yolo',           # 模型类型: 'auto', 'yolo', 'rknn', 'onnx'
    'use_gpu': False,                # RKNN 使用 NPU，不依赖 GPU
    'conf_threshold': 0.5,           # 一般物品置信度阈值 (安全帽、反光衣)
    'person_threshold': 0.8,         # 人员检测置信度阈值
    
    # 检测类别映射
    'class_map': {
        0: '安全帽',
        3: '人',
        4: '反光衣'
    },
    
    # 检测间隔
    'detection_interval': 0.1,       # 检测间隔(秒)
    
    # 告警控制
    'alert_cooldown': 3.0,           # 同一目标告警冷却时间(秒)
    'max_alerts_per_minute': 10,     # 每分钟最大告警数
}