改进YOLOv5s的交通标识检测算法
DOI:
作者:
作者单位:

上海电力大学 电子与信息工程学院 上海 201200

作者简介:

通讯作者:

中图分类号:

基金项目:


Traffic sign detection algorithm based on improved YOLOv5s
Author:
Affiliation:

Shanghai University of Electric Power

Fund Project:

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

    针对交通标识在图像中占比小、检测精度低且周围环境复杂等问题,提出一种基于改进YOLOv5s的算法。首先,在主干网络部分添加注意力机制ECA(Efficient Channel Attention,高效通道注意力),增强网络的特征提取能力,有效解决了周围环境复杂的问题;其次,提出HASPP(Hybrid Atrous Spatial Pyramid Pooling,混合空洞空间金字塔池化),增强了网络结合上下文的能力;最后,修改网络中的Neck结构,使高层特征与底层特征有效融合,同时避免了跨卷积层造成的信息丢失。实验结果表明,改进后的算法在交通标识数据集上取得了94.4%的平均检测精度、74.1%的召回率以及94.0%的精确率,较原始算法分别提升了3.6、2.8、3.4个百分点 。

    Abstract:

    An algorithm based on improved YOLOv5s is proposed to address the problems of small percentage of traffic signs in the image, low detection accuracy and complex surrounding environment. Firstlythe attention mechanism ECA (Efficient Channel Attention) is added to the backbone network part to enhance the feature extraction ability of the network and effectively solve the problem of complex surrounding environment;Secondly, the HASPP (Hybrid Atrous Spatial Pyramid Pooling) is proposed, which enhances the network's ability to combine context; Finally,the Neck structure in the network is modified to allow efficient fusion of high level features with underlying features while avoiding information loss across convolutional layers. Experimental results show that the improved algorithm achieves an average detection accuracy of 94.4%, a recall rate of 74.1% and an accuracy rate of 94.0% on the traffic signage dataset, which were 3.6, 2.8, and 3.4 percentage points higher than the original algorithm, respectively.

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

李孟浩,袁三男.改进YOLOv5s的交通标识检测算法[J].南京信息工程大学学报,,():

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

地址:江苏省南京市宁六路219号    邮编:210044

联系电话:025-58731025    E-mail:nxdxb@nuist.edu.cn

南京信息工程大学学报 ® 2024 版权所有  技术支持:北京勤云科技发展有限公司