dsh/dual_camera_system/config/detection.py

"""
检测配置
"""

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

# 安全检测模型配置
SAFETY_DETECTION_CONFIG = {
    'model_path': '/home/wen/dsh/yolo/yolo11m_safety.pt',  # 安全检测模型路径
    'use_gpu': True,                 # 是否使用 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,     # 每分钟最大告警数
}