一种基于改进YOLOv5s-Ghost网络的交通标志识别方法
DOI:
作者:
作者单位:

(1.天津理工大学 电气工程与自动化学院,天津市复杂系统控制理论与应用重点实验室,天津 300384; 2.北京航天动力研究所,北京 100076)

作者简介:

许 亮(1980—),男,博士,副教授,博士生导师,主要从事气动光学、人工智能应用等方面的研究.

通讯作者:

中图分类号:

基金项目:

国家自然科学基金(61975151,61308120)资助项目


A traffic sign recognition method based on improved YOLOv5s-Ghost network
Author:
Affiliation:

(1.Tianjin Key Laboratory for Control Theory & Applications in Complicated Systems,School of Electrical Engineering and Automation,Tianjin University of Techn ology,Tianjin 300384, China;2.Beijing Aerospace Propulsion Institute,Beijing 100076, China)

Fund Project:

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

    针对目前自动驾驶过程中对交通标志的识别检测 速度慢的问题,提出一种改进的YOLOv 5s-Ghost网络模型对交通标志进行识别的方法,在3×3运算核Ghost Net模型框架下,通过 两个连续的Ghost模块构建的Ghost Bottleneck模块,并代替C3模块中全部的Bottleneck模 块,与跨阶段局部网络(cross-stage position network,CSPNet)模块结合生成Ghost Bot t leneck CSP模块。通过调整每个模块中加入Ghost Bottleneck模块的数量,对比实验数据得 到最佳网络模型。分别用原网络和新网络对TT100K数据集进行训练,对比实验数据表明,YO LOv5s-Ghost模型的检测精度达95.1%,检测速度达到了52. 6 FPS,模型大小压缩了69.3%,在保证原检测精度的情 况下提高了网络的检测速度。

    Abstract:

    In view of the problem of slow detection speed and large network model of traffic signs in the process of automatic driving,an improved YOLOv5s-Ghost network model is proposed to identify traffic signs.Under the framework of 3× 3 computing core Ghost Net model,the Ghost Bottleneck module is constructed by two consecutive Ghost modules,which replace all the Bottleneck module of C3 modules and combine with cross-stage position network (CSPNet) to generate Ghost Bottleneck CSP module.The best netw o rk model is obtained by comparing the experimental data by adjusting the number of Ghost Bottleneck modules added to each module.The original network and the n ew network are respectively used to train TT100K data set.Compared experimenta l data show that the detection accuracy of the YOLOv5s-Ghost model is 95.1%,th e detection speed reaches 52.6 FPS,and the model size is compressed by 69.3%,whi ch improves the detection speed of the network while ensuring the original detec tion accuracy.

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

徐正军,张强,许亮.一种基于改进YOLOv5s-Ghost网络的交通标志识别方法[J].光电子激光,2023,34(1):52~61

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2022-04-29
  • 最后修改日期:2022-06-30
  • 录用日期:
  • 在线发布日期: 2023-01-16
  • 出版日期: