基于注意力机制优化的YOLOv7目标检测算法OA
An Optimized YOLOv7 Object Detection Algorithm Based on Attention Mechanisms
为保障自动驾驶系统能准确且迅速地识别道路目标信息,提出在YOLOv7主干网络中分别融合SE、CBAM、ECA三类注意力模块,设计并优化了一种可执行特征重新校准的网络结构.基于公开的KITTI数据集进行网络模型训练,搭建数据处理平台,对比分析融合SE注意力模块优化前后的网络模型性能以及对Car、Pedestrian、Cyclist三类目标的检测效果.横向对比试验结果表明,融合SE注意力模块优化算法在计算效率和精度之间取得了更好的平衡.消融试验进一步证明了SE模块贡献了主要性能增益,融合SE模块能较好地兼顾检测精度和实时性.优化后的网络模型mAP值提升了1.16%,FPS下降幅度控制在7%以内,同时对Car、Pedestrian、Cyclist三类目标检测的精度分别提高了3.17%、3.88%、2.77%,有效降低了漏检率和误检率.试验结果证明了基于注意力机制优化的YOLOv7目标检测算法对复杂环境目标检测的有效性及其实用价值,为自动驾驶系统的安全性与可靠识别提供了技术支持.
To ensure that an autonomous-driving system can accurately and swiftly identify road targets,this paper integrates SE,CBAM,and ECA attention modules into the YOLOv7 backbone network and designs an optimized network capable of feature recalibration.The network model is trained on the publicly available KITTI dataset,and a data processing platform is established.The performance of the network model and its detection of Car,Pedestrian,and Cyclist classes are compared before and after optimization with the SE attention module.Comparative experimental results demonstrate that the optimization algorithm incorporating the SE attention module achieves a better balance between computational efficiency and accuracy.Ablation studies further confirm that the SE module provides the primary performance gain,and the fused network balances detection accuracy with real-time performance.After optimization,the mean Average Precision(mAP)of the network increases by 1.16%,the FPS drop remains within 7%,and detection accuracies for Car,Pedestrian,and Cyclist improve by 3.17%,3.88%,and 2.77%respectively,effectively reducing missed and false detection rates.These results validate the effectiveness and practical value of the attention-optimized YOLOv7 algorithm for object detection in complex environments,providing robust technical support for the safety and reliable perception of autonomous driving systems.
张洪梅;张云飞;杨良义;李斌;张强
智能汽车安全技术全国重点实验室,重庆 401122||中国汽车工程研究院股份有限公司,重庆 401122智能汽车安全技术全国重点实验室,重庆 401122||中国汽车工程研究院股份有限公司,重庆 401122智能汽车安全技术全国重点实验室,重庆 401122||中国汽车工程研究院股份有限公司,重庆 401122智能汽车安全技术全国重点实验室,重庆 401122||中国汽车工程研究院股份有限公司,重庆 401122智能汽车安全技术全国重点实验室,重庆 401122||中国汽车工程研究院股份有限公司,重庆 401122
交通工程
道路目标信息YOLOv7目标检测算法注意力机制KITTI数据集
road target informationYOLOv7 object detection algorithmattention mechanismKITTI data set
《汽车工程学报》 2026 (2)
227-235,9
重庆市人力资源和社会保障局重庆英才计划(CQYC20220207209)
评论