VisionGuard提升头盔违规检测准确率,解决数据不均衡与标注不一致问题。
VisionGuard: Synergistic Framework for Helmet Violation Detection
- 通过追踪算法实现跨帧标签一致性,修正误分类。
- 生成虚拟框提升小样本类召回率,缓解数据不平衡。
- 在真实交通监控中表现更优,适合安防系统部署。
维护摩托车骑行者头盔佩戴规范对提升道路安全和交通管理系统有效性至关重要。然而,由于环境变化、摄像头角度差异及数据标注不一致,自动检测头盔违规面临显著挑战,影响摩托车及骑手的可靠识别与分类一致性。为此,我们提出VisionGuard——一种协同式多阶段框架,克服帧级检测器在类别不平衡和标注不一致场景下的局限性。该框架集成两个核心模块:基于追踪的自适应标注模块,利用追踪算法在多帧间保持标签连续性并修正错误分类;上下文扩展模块,通过生成带置信度的虚拟边界框,提升欠代表类别的召回率,缓解数据不平衡的影响。实验表明,VisionGuard相比基线检测器整体mAP提升3.1%,验证了其有效性和在真实交通监控系统中的应用潜力,有助于推动交通安全与合规管理。
原文摘要 · Abstract (English)
Enforcing helmet regulations among motorcyclists is essential for enhancing road safety and ensuring the effectiveness of traffic management systems. However, automatic detection of helmet violations faces significant challenges due to environmental variability, camera angles, and inconsistencies in the data. These factors hinder reliable detection of motorcycles and riders and disrupt consistent object classification. To address these challenges, we propose VisionGuard, a synergistic multi-stage framework designed to overcome the limitations of frame-wise detectors, especially in scenarios with class imbalance and inconsistent annotations. VisionGuard integrates two key components: Adaptive Labeling and Contextual Expander modules. The Adaptive Labeling module is a tracking-based refinement technique that enhances classification consistency by leveraging a tracking algorithm to assign persistent labels across frames and correct misclassifications. The Contextual Expander module improves recall for underrepresented classes by generating virtual bounding boxes with appropriate confidence scores, effectively addressing the impact of data imbalance. Experimental results show that VisionGuard improves overall mAP by 3.1% compared to baseline detectors, demonstrating its effectiveness and potential for real-world deployment in traffic surveillance systems, ultimately promoting safety and regulatory compliance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。