基于YOLOv5-MobileNetV3算法的目标检测
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:


Object Detection Based on YOLOv5-MobileNetV3 Algorithm
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 增强出版
  • |
  • 文章评论
    摘要:

    车辆行驶过程中, 对前方目标的检测速度和检测精度一直是自动驾驶领域研究的重点. 针对现有的目标检测算法模型, 在复杂交通环境下, 传统模型面对重叠目标容易导致误检和漏检的情况发生, 大幅度提高检测精度又会造成计算量过大导致处理速度缓慢, 实时性下降的问题. 本文提出基于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.

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

曲英伟,刘锐.基于YOLOv5-MobileNetV3算法的目标检测.计算机系统应用,2024,33(7):213-221

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-02-05
  • 最后修改日期:2024-03-05
  • 录用日期:
  • 在线发布日期: 2024-06-05
  • 出版日期:
您是第位访问者
版权所有:中国科学院软件研究所 京ICP备05046678号-3
地址:北京海淀区中关村南四街4号 中科院软件园区 7号楼305房间,邮政编码:100190
电话:010-62661041 传真: Email:csa (a) iscas.ac.cn
技术支持:北京勤云科技发展有限公司

京公网安备 11040202500063号