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 条
  • [41] Traffic Sign Detection Algorithm Based on Improved YOLOv8s
    Zhang, Xiaoming
    Tian, Ying
    ENGINEERING LETTERS, 2024, 32 (01) : 168 - 178
  • [42] Traffic sign detection algorithm based on improved YOLOv4-Tiny
    Yao, Yingbiao
    Han, Li
    Du, Chenjie
    Xu, Xin
    Jiang, Xianyang
    SIGNAL PROCESSING-IMAGE COMMUNICATION, 2022, 107
  • [43] A Method for Traffic Sign Detection and Recognition Based on Genetic Algorithm
    Kobayashi, Mitsuhiro
    Baba, Mitsuru
    Ohtani, Kozo
    Li, Li
    2015 IEEE/SICE INTERNATIONAL SYMPOSIUM ON SYSTEM INTEGRATION (SII), 2015, : 455 - 460
  • [44] Enhancing small traffic sign recognition based on an improved YOLOv8 algorithm
    Zhang, Xiaoqing
    Wang, Weixi
    Wang, Qi
    Shan, Liang
    Yamane, Satoshi
    Zheng, Bochao
    Liu, Sifan
    Shen, Tianxing
    Ge, Dongming
    SIGNAL IMAGE AND VIDEO PROCESSING, 2025, 19 (06)
  • [45] A Lightweight Deep Learning Model for Real-time Detection and Recognition of Traffic Signs Images Based on YOLOv5
    He, Hui
    Chen, Qihong
    Xie, Guoping
    Yang, Boxiong
    Li, Shelei
    Zhou, Bo
    Gu, Yuye
    2022 INTERNATIONAL CONFERENCE ON CYBER-ENABLED DISTRIBUTED COMPUTING AND KNOWLEDGE DISCOVERY, CYBERC, 2022, : 206 - 212
  • [46] Traffic sign recognition using deep learning
    Patel V.
    Mehta J.
    Iyer S.
    Sharma A.K.
    International Journal of Vehicle Autonomous Systems, 2023, 16 (2-4) : 97 - 107
  • [47] Deep Transfer Learning for Traffic Sign Recognition
    Rosario, Grant
    Sonderman, Thomas
    Zhu, Xingquan
    2018 IEEE INTERNATIONAL CONFERENCE ON INFORMATION REUSE AND INTEGRATION (IRI), 2018, : 178 - 185
  • [48] An Enhanced Deep Learning Model for Obstacle and Traffic Light Detection Based on YOLOv5
    Li, Zhenwei
    Zhang, Wei
    Yang, Xiaoli
    ELECTRONICS, 2023, 12 (10)
  • [49] Semantic segmentation-based traffic sign detection and recognition using deep learning techniques
    Timbus, Calin
    Miclea, Vlad
    Lemnaru, Camelia
    2018 IEEE 14TH INTERNATIONAL CONFERENCE ON INTELLIGENT COMPUTER COMMUNICATION AND PROCESSING (ICCP), 2018, : 325 - 331
  • [50] High-Precision Traffic Sign Detection and Recognition Using an Enhanced YOLOv5
    Luo, Jiaomin
    Li, Ying
    Wei, Li
    Nie, Gang
    JOURNAL OF CIRCUITS SYSTEMS AND COMPUTERS, 2024,