A road traffic sign recognition method based on improved YOLOv5

被引:0
作者
Shi, Lu [1 ]
Zhang, Haifei [1 ]
机构
[1] Nantong Inst Technol, Sch Comp & Informat Engn, Nantong 226602, Peoples R China
关键词
deep learning; object detection; traffic sign recognition; YOLOv5;
D O I
10.1504/IJSNET.2024.10066756
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
With the rapid development of artificial intelligence technology, the automatic driving of intelligent vehicles has gradually entered people's lives. The traditional vision will fail in many scenarios, such as snow, lane line wear, occlusion, or haze weather. This study aims to provide an accurate and efficient method for recognising traffic signs in their natural surroundings. The advanced convolutional block attention module attention mechanism is embedded in the infrastructure of you only look once version 5 (YOLOv5), which strengthens the network's ability to capture key features of the image. Then, the transformer module is introduced in the core part of YOLOv5, which uses its self-attention mechanism to effectively enhance the overall context connection, thereby significantly enhancing the model's detection accuracy and achieving an impressive 91% high accuracy performance. According to the most recent experimental data, the enhanced YOLOv5 model performs exceptionally well at recognising traffic signs in various natural settings.
引用
收藏
页数:14
相关论文
共 26 条
[1]  
Albawi S, 2017, I C ENG TECHNOL
[2]   PhishTransformer: A Novel Approach to Detect Phishing Attacks Using URL Collection and Transformer [J].
Asiri, Sultan ;
Xiao, Yang ;
Li, Tieshan .
ELECTRONICS, 2024, 13 (01)
[3]  
Bochkovskiy A, 2020, arXiv, DOI DOI 10.48550/ARXIV.2004.10934
[4]   A rotatable battery recognition method based on improved YOLOv5 [J].
Chen, Wenming ;
Liang, Dongtai ;
Ding, Wenhui ;
Wang, Meng ;
Chen, Zizhen .
INTERNATIONAL JOURNAL OF SENSOR NETWORKS, 2024, 45 (02) :101-114
[5]   A transformative deep learning framework for traffic modelling using sensors-based multi-resolution traffic data [J].
Goswami, Shubhashish ;
Kumar, Abhimanyu .
INTERNATIONAL JOURNAL OF SENSOR NETWORKS, 2023, 42 (03) :145-155
[6]  
Han Y.L., 2012, Master thesis, P43
[7]  
He KM, 2020, IEEE T PATTERN ANAL, V42, P386, DOI [10.1109/ICCV.2017.322, 10.1109/TPAMI.2018.2844175]
[8]  
Jiang Jinhong, 2020, Journal of Computer Applications, V40, P2472, DOI 10.11772/j.issn.1001-9081.2020010062
[9]   ImageNet Classification with Deep Convolutional Neural Networks [J].
Krizhevsky, Alex ;
Sutskever, Ilya ;
Hinton, Geoffrey E. .
COMMUNICATIONS OF THE ACM, 2017, 60 (06) :84-90
[10]   Deep learning [J].
LeCun, Yann ;
Bengio, Yoshua ;
Hinton, Geoffrey .
NATURE, 2015, 521 (7553) :436-444