基于注意力机制的交通标志识别
DOI:
CSTR:
作者:
作者单位:

中南林业科技大学机电工程学院,湖南 长沙410000

作者简介:

通讯作者:

中图分类号:

TP389.1

基金项目:


Traffic sign recognition based on Attention Mechanism
Author:
Affiliation:

School of Mechanical and Electrical Engineering, Central South University of Forestry and Technology, Changsha 410000, Hunan, China

Fund Project:

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

    针对实际场景中的交通标志大多小而密集,导致小目标交通标志识别准确度较低的问题,提出一种改进YOLOv5算法。首先将CBAM同时嵌入YOLOv5网络的Backbone和Head部分,以提升网络特征提取能力。其次为解决GIoU Loss可能造成的模型收敛速度较慢问题,改用DIoU Loss作为网络回归损失函数。实验结果表明,改进后的算法对于交通标志图像的识别平均准确率达到96.40%,相较于原算法有了6.83%的提升。最后为验证模型的实时可行性,在TX2嵌入式系统中利用本文改进YOLOv5算法对实景视频中的交通标志进行识别,结果表明本文改进算法能在嵌入式系统中流畅运行。

    Abstract:

    Aiming at the problem of low accuracy of small target detection in traffic sign recognition tasks,which caused by that most of traffic signs in actual scene are small and dense,this paper proposes an improved YOLOv5 algorithm.firstly,embedding the CBAM into the Backbone and Neck of YOLOv5 network to improve the network feature extraction ability.and in order to solve the problem of slow network converge caused by GIOU Loss, DIoU Loss was used as the regression Loss function of the network. Experimental results show that the improved algorithm reaches 96.40% mAP in traffic sign recognition task,which is 6.83% higher than the original YOLOv5 algorithm. Finally, sending the improved network into TX2 embedded system to recognize traffic signs in real video,the result shows that the improved algorithm can run smoothly in embedded system.

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

李宇琼,周永军,蒋淑霞,梁杨.基于注意力机制的交通标志识别[J].电子测量技术,2022,45(8):116-120

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-05-10
  • 出版日期:
文章二维码