基于改进YOLOv8n的轻量化分心驾驶行为检测方法
DOI:
CSTR:
作者:
作者单位:

淮阴工学院自动化学院 淮安 223003

作者简介:

通讯作者:

中图分类号:

TN0;TP391

基金项目:

国家自然科学基金面上项目(62173159)、淮安市自然科学研究项目(HAB202362)资助


Lightweight distracted driving behavior detection method based on improved YOLOv8n
Author:
Affiliation:

School of Automation, Huaiyin Institute of Technology,Huai′an 223003, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    分心驾驶行为是导致出现道路交通安全问题的主要原因之一。针对现有基于深度学习的检测算法计算复杂度高、泛化能力有限、检测精度不理想等问题,本文构建了一种基于改进YOLOv8n的轻量化分心驾驶行为检测算法。首先,将CAA注意力机制融合进StarNet,形成StarNet-CAA,并且将StarNet-CAA集成到YOLOv8n的主干网络中,提高了模型的全局特征提取能力,显著降低了计算复杂度。随后,将FasterBlock与CGLU相结合加入到颈部网络中,形成C2f-Faster-CGLU模块,降低计算成本。此外,在检测头中引入共享卷积,进一步减少了计算量和参数量。实验结果表明,改进的YOLOv8n算法显著提高了分心驾驶行为检测的效率,在StateFarm数据集上达到了99.4%的准确率。模型的参数量减少46.7%,计算量减少41.5%。此外,在100-Driver数据集上进行了泛化实验,结果表明,与YOLOv8n相比,该方案的泛化效果有所提高。因此,该算法在保持高可靠性和泛化能力的同时,显著降低了计算量。

    Abstract:

    Distracted driving is one of the main causes of road traffic safety problems. Aiming at the problems of high computational complexity, limited generalization ability and unsatisfactory detection accuracy of existing detection algorithms based on deep learning, this paper constructs a lightweight distracted driving behavior detection algorithm based on improved YOLOv8n. Firstly, the Context Anchor Attention mechanism was fused into StarNet to form StarNet-CAA, and StarNet-CAA was integrated into the backbone network of YOLOv8n to improve the global feature extraction ability of the model and significantly reduce the computational complexity. Subsequently, FasterBlock combined with CGLU is added to the neck network to form the C2f-Faster-CGLU module, which reduces the computational cost. In addition, the shared convolution is introduced into the detection head to further reduce the computational burden and parameter size. Experimental results show that the improved YOLOv8n algorithm significantly improves the efficiency of distracted driving behavior detection, reaching an accuracy of 99.3%on the StateFarm dataset. The number of parameters of the model is reduced by 46.7%, and the amount of calculation is reduced by 41.5%. In addition, the generalization experiment is carried out on the 100-Driver dataset, and the results show that the generalization effect of the proposed scheme is improved compared with YOLOv8n. Therefore, the proposed algorithm significantly reduces the computational burden while maintaining high reliability and generalization ability.

    参考文献
    相似文献
    引证文献
引用本文

沈骞,张磊,张宇翔,李易,刘世豪.基于改进YOLOv8n的轻量化分心驾驶行为检测方法[J].电子测量技术,2024,47(24):65-75

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2025-01-24
  • 出版日期:
文章二维码