Computer Engineering and Applications ›› 2023, Vol. 59 ›› Issue (10): 262-269.DOI: 10.3778/j.issn.1002-8331.2211-0359

• Graphics and Image Processing • Previous Articles     Next Articles

Improved YOLOv5 Traffic Sign Detection Algorithm

YANG Guoliang, YANG Hao, YU Shuaiying, WANG Jixiang, NIE Ziling   

  1. School of Electrical Engineering and Automation, Jiangxi University of Science and Technology, Ganzhou, Jiangxi 341000, China
  • Online:2023-05-15 Published:2023-05-15

改进YOLOv5的交通标志检测算法

杨国亮,杨浩,余帅英,王吉祥,聂子玲   

  1. 江西理工大学 电气工程与自动化学院,江西 赣州 341000

Abstract: Traffic sign detection has been widely used in intelligent transportation systems such as automatic driving and assisted driving, and its detection performance is related to driving safety. Aiming at the problem that the existing object detection algorithm has poor detection effect on traffic signs with small size, low resolution and non-obvious features in the image, a traffic sign detection algorithm based on improved YOLOv5s is proposed. The 80×80 small sensing field object detection layer in the original algorithm is changed to a smaller 160×160 detection layer, which improves the detection ability of the network model for small object of traffic signs and reduces the missed detection rate of small object. The attention context module(ACM) is constructed to obtain the characteristic information of the target and its adjacent regions from different receptive fields for each branch, and the attention mechanism is used to make the network pay more attention to the traffic signs in the image and avoid being affected by other complex information. The feature fusion module(FFM) is added to filter out the useless information on different layers and retain only the useful information for the model to detect traffic signs. The tacit knowledge is added to refine the output of the detection layer. Experimental results show that the improved algorithm has a recall rate and average accuracy of 95.2% and 97.2% on the CCTSDB traffic sign detection dataset, which is improved compared with the original model, and the effect is significantly improved under medium and long-distance small object detection, and the simultaneous detection speed is 47.3 FPS to meet the real-time requirements.

Key words: intelligent transportation, traffic sign, attention context, receptive field amplification,  , feature fusion, object detection

摘要: 交通标志检测在自动驾驶、辅助驾驶等智能交通系统已得到广泛应用,其检测性能关乎到行车安全。针对现有目标检测算法对图像中尺寸小、分辨率低和特征不明显的交通标志检测效果较差的问题,提出了一种基于改进YOLOv5s的交通标志检测算法。将原算法中80×80小感受野目标检测层改为感受野更小的160×160检测层,提高网络模型对交通标志小目标的检测能力,降低小目标的漏检率;构建了注意力上下文模块(attention context module,ACM),对各分支获取不同的感受野,得到目标及其相邻区域的特征信息,并且使用注意力机制,让网络更关注于图像中的交通标志,避免受其他复杂信息的影响;加入特征融合模块(feature fusion module,FFM),过滤不同层上的无用信息,只保留对模型检测交通标志有用的信息;加入隐性知识,对检测层进行输出细化。实验结果表明,改进后的算法在CCTSDB交通标志检测数据集上召回率和平均精度达到94.7%、97.6%,相比原模型均有提升,在中远距离小目标检测下效果改善明显,同时检测速度为47.3?FPS,满足实时性要求。

关键词: 智能交通, 交通标志, 注意力上下文, 感受野扩增, 特征融合, 目标检测