基于YOLOv5的交通标志识别
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

陕西省重点研发计划(2021GY-287)


Traffic Sign Recognition Based on YOLOv5
Author:
Affiliation:

Fund Project:

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

    交通标志识别是自动驾驶技术中的关键一部分. 针对交通标志在道路场景中目标较小且识别精度较低的问题, 提出一种改进的YOLOv5算法. 首先在YOLOv5模型中引入全局注意力机制(GAM), 提高网络捕获不同尺度交通标志特征的能力; 其次将YOLOv5算法中使用的GIoU损失函数更换为更具回归特性的CIoU损失函数来优化模型, 提高对交通标志的识别精度. 最后在Tsinghua-Tencent 100K数据集上进行训练, 实验结果表明, 改进后的YOLOv5算法对交通标志识别的平均精度均值为93.00%, 相比于原算法提升了5.72%, 具有更好的识别性能.

    Abstract:

    Traffic sign recognition is a key part of autonomous driving technology. Given the problems of small targets and low recognition accuracy of traffic signs in road scenes, an improved YOLOv5 algorithm is proposed. First, the global attention mechanism (GAM) is introduced into the YOLOv5 model to improve the network’s ability to capture traffic sign features of different scales. Second, the GIoU loss function used in the YOLOv5 algorithm is replaced with the CIoU loss function which is more regressive to optimize the model and improve the recognition accuracy of traffic signs. Finally, the training is carried out on the Tsinghua-Tencent 100K dataset. The experimental results show that the average accuracy of the improved YOLOv5 algorithm for traffic sign recognition is 93.00%, which is 5.72% higher than that of the original one, indicating that the improved algorithm has better recognition performance.

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

郑红彬,宋晓茹,刘康.基于YOLOv5的交通标志识别.计算机系统应用,2023,32(8):230-237

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

京公网安备 11040202500063号