首页|基于改进YOLOv8n的轻量化分心驾驶行为检测方法

基于改进YOLOv8n的轻量化分心驾驶行为检测方法

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

distracted drivingYOLOv8nobject detectiondeep learning

沈骞、张磊、张宇翔、李易、刘世豪

展开 >

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

分心驾驶 YOLOv8n 目标检测 深度学习

2024

电子测量技术
北京无线电技术研究所

电子测量技术

CSTPCD北大核心
影响因子:1.166
ISSN:1002-7300
年,卷(期):2024.47(24)