YOLO11-CR用轻量模块提升疲劳驾驶检测精度与定位能力。
YOLO11-CR: a Lightweight Convolution-and-Attention Framework for Accurate Fatigue Driving Detection
- 融合卷积与注意力机制,增强特征表达能力。
- 在DSM数据集上达88.09% mAP@50,优于基线模型。
- 适合车载实时系统部署,尤其擅长小目标和侧脸识别。
驾驶员疲劳检测对智能交通系统至关重要,可有效减少交通事故。尽管基于生理信号和车辆动态的方法精度高,但存在侵入性强、依赖硬件、真实环境鲁棒性差等问题。视觉方法提供非侵入式且可扩展的替代方案,但仍面临小目标或遮挡目标检测困难、多尺度特征建模不足等挑战。为此,本文提出YOLO11-CR,一种专为实时疲劳检测设计的轻量高效目标检测模型。该模型引入两个关键模块:卷积与注意力融合模块(CAFM),将局部卷积特征与全局Transformer上下文结合,提升特征表达;矩形校准模块(RCM),捕捉水平与垂直上下文信息,增强空间定位能力,尤其适用于侧脸及手机等小目标。在DSM数据集上的实验表明,YOLO11-CR达到87.17%精度、83.86%召回率、mAP@50为88.09%、mAP@50-95为55.93%,显著优于基线模型。消融实验证实了CAFM与RCM在提升敏感性和定位精度方面的有效性。结果表明,YOLO11-CR为车内疲劳监测提供了实用且高性能的解决方案,具备实际部署潜力,未来可拓展至时序建模、多模态融合与嵌入式优化。
原文摘要 · Abstract (English)
Driver fatigue detection is of paramount importance for intelligent transportation systems due to its critical role in mitigating road traffic accidents. While physiological and vehicle dynamics-based methods offer accuracy, they are often intrusive, hardware-dependent, and lack robustness in real-world environments. Vision-based techniques provide a non-intrusive and scalable alternative, but still face challenges such as poor detection of small or occluded objects and limited multi-scale feature modeling. To address these issues, this paper proposes YOLO11-CR, a lightweight and efficient object detection model tailored for real-time fatigue detection. YOLO11-CR introduces two key modules: the Convolution-and-Attention Fusion Module (CAFM), which integrates local CNN features with global Transformer-based context to enhance feature expressiveness; and the Rectangular Calibration Module (RCM), which captures horizontal and vertical contextual information to improve spatial localization, particularly for profile faces and small objects like mobile phones. Experiments on the DSM dataset demonstrated that YOLO11-CR achieves a precision of 87.17%, recall of 83.86%, mAP@50 of 88.09%, and mAP@50-95 of 55.93%, outperforming baseline models significantly. Ablation studies further validate the effectiveness of the CAFM and RCM modules in improving both sensitivity and localization accuracy. These results demonstrate that YOLO11-CR offers a practical and high-performing solution for in-vehicle fatigue monitoring, with strong potential for real-world deployment and future enhancements involving temporal modeling, multi-modal data integration, and embedded optimization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。