PSG-Yolov5: A Paradigm for Traffic Sign Detection and Recognition Algorithm Based on Deep Learning

被引:23
|
作者
Hu, Jie [1 ]
Wang, Zhanbin [1 ]
Chang, Minjie [1 ]
Xie, Lihao [1 ]
Xu, Wencai [1 ]
Chen, Nan [1 ]
机构
[1] Wuhan Univ Technol, Coll Automot Engn, Wuhan 430070, Peoples R China
来源
SYMMETRY-BASEL | 2022年 / 14卷 / 11期
关键词
traffic sign detection; deep learning; small object detection; multi-scale fusion;
D O I
10.3390/sym14112262
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
With the gradual popularization of autonomous driving technology, how to obtain traffic sign information efficiently and accurately is very important for subsequent decision-making and planning tasks. Traffic sign detection and recognition (TSDR) algorithms include color-based, shape-based, and machine learning based. However, the algorithms mentioned above are insufficient for traffic sign detection tasks in complex environments. In this paper, we propose a traffic sign detection and recognition paradigm based on deep learning algorithms. First, to solve the problem of insufficient spatial information in high-level features of small traffic signs, the parallel deformable convolution module (PDCM) is proposed in this paper. PDCM adaptively acquires the corresponding receptive field preserving the integrity of the abstract information through symmetrical branches thereby improving the feature extraction capability. Simultaneously, we propose sub-pixel convolution attention module (SCAM) based on the attention mechanism to alleviate the influence of scale distribution. Distinguishing itself from other feature fusion, our proposed method can better focus on the information of scale distribution through the attention module. Eventually, we introduce GSConv to further reduce the computational complexity of our proposed algorithm, better satisfying industrial application. Experimental results demonstrate that our proposed methods can effectively improve performance, both in detection accuracy and mAP@0.5. Specifically, when the proposed PDCM, SCAM, and GSConv are applied to the Yolov5, it achieves 89.2% mAP@0.5 in TT100K, which exceeds the benchmark network by 4.9%.
引用
收藏
页数:16
相关论文
共 50 条
  • [31] Automatic Traffic Sign Recognition Artificial Inteligence - Deep Learning Algorithm
    Radu, Mihai Daniel
    Costea, Ilona Madalina
    Stan, Valentin Alexandru
    PROCEEDINGS OF THE 2020 12TH INTERNATIONAL CONFERENCE ON ELECTRONICS, COMPUTERS AND ARTIFICIAL INTELLIGENCE (ECAI-2020), 2020,
  • [32] Robust detection and recognition of japanese traffic sign in the complex scenes based on deep learning
    Hasegawa, Ryo
    Iwamoto, Yutaro
    Chen, Yen-Wei
    2019 IEEE 8th Global Conference on Consumer Electronics, GCCE 2019, 2019, : 575 - 578
  • [33] Traffic Sign Detection Based on Improved YOLOv5 Model
    Zhao, Yibing
    Wang, Yannan
    Xing, Shuyong
    Guo, Lie
    SMART TRANSPORTATION AND GREEN MOBILITY SAFETY, GITSS 2022, 2024, 1201 : 293 - 307
  • [34] Road Sign Detection and Recognition of Thai Traffic Based on YOLOv3
    Thipsanthia, Paitoon
    Chamchong, Rapeeporn
    Songram, Panida
    MULTI-DISCIPLINARY TRENDS IN ARTIFICIAL INTELLIGENCE, 2019, 11909 : 271 - 279
  • [35] An Attention Based YOLOv5 Network for Small Traffic Sign Recognition
    Chen, Yi
    Wang, Junfan
    Dong, Zhekang
    Yang, Yuxiang
    Luo, Qiang
    Gao, Mingyu
    2022 IEEE 31ST INTERNATIONAL SYMPOSIUM ON INDUSTRIAL ELECTRONICS (ISIE), 2022, : 1158 - 1164
  • [36] A Lightweight Traffic Sign Recognition Model Based on Improved YOLOv5
    Yang, Jie
    Sun, Ting
    Zhu, Wenchao
    Li, Zonghao
    IEEE ACCESS, 2023, 11 : 115998 - 116010
  • [37] A road traffic sign recognition method based on improved YOLOv5
    Shi, Lu
    Zhang, Haifei
    INTERNATIONAL JOURNAL OF SENSOR NETWORKS, 2025, 47 (01)
  • [38] A Novel Lightweight Traffic Sign Recognition Model Based on YOLOv5
    Li, Wenju
    Zhang, Gan
    Cui, Liu
    JOURNAL OF TRANSPORTATION ENGINEERING PART A-SYSTEMS, 2023, 149 (05)
  • [39] Street Sign Recognition Algorithm Based on Deep Learning
    Wang, Jiawei
    Chen, Cen
    Wang, Chongwen
    PROCEEDINGS OF THE 2020 3RD INTERNATIONAL CONFERENCE ON IMAGE AND GRAPHICS PROCESSING (ICIGP 2020), 2020, : 31 - 35
  • [40] Traffic Sign Detection Algorithm Based on Improved YOLOv8n
    Peng, Jun
    Mou, Biao
    Jin, Shangzhu
    Lu, Yiyi
    Li, Chenxi
    Chen, Wei
    Jiang, Aiping
    2024 IEEE 19TH CONFERENCE ON INDUSTRIAL ELECTRONICS AND APPLICATIONS, ICIEA 2024, 2024,