YOLOv5 Traffic Sign Recognition Algorithm Combined with RepVGG

Hua-ling GUO , Jia-shuai LIU , Bin ZHENG , Yun-hua YIN , Di-yu ZHAO

Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (9) : 3869 -3875.

PDF
Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (9) : 3869-3875 DOI: 10.12404/j.issn.1671-1815.2302814
research-article
YOLOv5 Traffic Sign Recognition Algorithm Combined with RepVGG
Author information +
History +
PDF

Abstract

In order to achieve the safety of autonomous driving, it is crucial to accurately detect traffic signs. Addressing the issue of low accuracy in traditional methods for traffic sign detection, a modified YOLOv5 algorithm incorporating the RepVGG module for traffic sign recognition was proposed. Firstly, by replacing certain CBS modules in the original algorithm with the RepVGG module, the feature extraction capability was enhanced. Additionally, the channel block attention module (CBAM) attention mechanism was introduced in the feature fusion module to strengthen the detection model's resistance to interference. Finally, during network training, the efficient-IoU (EIoU) loss function was utilized to achieve more precise target positioning, thereby improving the algorithm's detection accuracy and iteration speed. Experimental results demonstrate that the improved YOLOv5 algorithm exhibits faster iteration speed. On the CCTSDB traffic sign dataset, it achieves improvements of 4.99% in accuracy, 3.62% in recall rate, and 1.73% in average accuracy compared to the original YOLOv5 algorithm, making it more suitable for practical applications.

Cite this article

Download citation ▾
Hua-ling GUO, Jia-shuai LIU, Bin ZHENG, Yun-hua YIN, Di-yu ZHAO. YOLOv5 Traffic Sign Recognition Algorithm Combined with RepVGG. Science Technology and Engineering, 2024, 24(9): 3869-3875 DOI:10.12404/j.issn.1671-1815.2302814

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (2795KB)

17

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉