本文已被:浏览 289次 下载 795次
Received:February 05, 2024 Revised:March 05, 2024
Received:February 05, 2024 Revised:March 05, 2024
中文摘要: 车辆行驶过程中, 对前方目标的检测速度和检测精度一直是自动驾驶领域研究的重点. 针对现有的目标检测算法模型, 在复杂交通环境下, 传统模型面对重叠目标容易导致误检和漏检的情况发生, 大幅度提高检测精度又会造成计算量过大导致处理速度缓慢, 实时性下降的问题. 本文提出基于YOLOv5模型的改进算法. 首先采用MobileNetV3网络替换原模型中主干网络C3的方案, 实现网络仍保持轻量化的同时, 提高模型响应速度. 其次, 提出一种非极大值抑制算法Adaptive-EIoU-NMS来提高重叠目标的识别精度. 最后采用K-means++聚类算法替换原有聚类算法, 生成更精确的锚框. 实验结果表明, 改进后的模型平均检测精度达到90.1%, 检测速度达到89 f/s. 实验结果可以证实, 改进后的模型针对复杂场景检测精度和检测速度都有显著提高.
Abstract:The detection speed and accuracy of detecting targets ahead during vehicle operation have always been a focus of research in the field of autonomous driving. For existing object detection algorithm models, in complex traffic environments, traditional models are prone to false positives and missed detections when facing overlapping targets. Significantly improving detection accuracy can also lead to increased computational demands, resulting in slower processing speed and decreased real-time performance. This article proposes an improved algorithm based on the YOLOv5 model. Firstly, the MobileNetV3 network is adopted to replace the C3 backbone network in the original model, achieving a lightweight network while improving the model’s response speed. Secondly, a non-maximum suppression algorithm, Adaptive-EIoU-NMS, is proposed to improve the recognition accuracy of overlapping targets. Finally, the K-means++ clustering algorithm is used to replace the original clustering algorithm and generate more accurate anchor boxes. Experimental results show that the improved model achieves an average detection accuracy of 90.1% and a detection speed of 89 frames per second (f/s). The experimental results confirm that the enhanced model significantly improves both detection accuracy and speed for complex scene detection.
文章编号: 中图分类号: 文献标志码:
基金项目:
Author Name | Affiliation | |
QU Ying-Wei | School of Software, Dalian Jiaotong University, Dalian 116052, China | |
LIU Rui | School of Software, Dalian Jiaotong University, Dalian 116052, China | 236422217@qq. com |
Author Name | Affiliation | |
QU Ying-Wei | School of Software, Dalian Jiaotong University, Dalian 116052, China | |
LIU Rui | School of Software, Dalian Jiaotong University, Dalian 116052, China | 236422217@qq. com |
引用文本:
曲英伟,刘锐.基于YOLOv5-MobileNetV3算法的目标检测.计算机系统应用,2024,33(7):213-221
QU Ying-Wei,LIU Rui.Object Detection Based on YOLOv5-MobileNetV3 Algorithm.COMPUTER SYSTEMS APPLICATIONS,2024,33(7):213-221
曲英伟,刘锐.基于YOLOv5-MobileNetV3算法的目标检测.计算机系统应用,2024,33(7):213-221
QU Ying-Wei,LIU Rui.Object Detection Based on YOLOv5-MobileNetV3 Algorithm.COMPUTER SYSTEMS APPLICATIONS,2024,33(7):213-221