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 条
  • [1] UCN-YOLOv5: Traffic Sign Object Detection Algorithm Based on Deep Learning
    Liu, Peilin
    Xie, Zhaoyang
    Li, Taijun
    IEEE ACCESS, 2023, 11 : 110039 - 110050
  • [2] An Improved Traffic Sign Detection and Recognition Deep Model Based on YOLOv5
    Wang, Qianying
    Li, Xiangyu
    Lu, Ming
    IEEE ACCESS, 2023, 11 : 54679 - 54691
  • [3] Traffic Sign Recognition Algorithm Based on Improved YOLOv5
    Sang, Zhengxiao
    Xia, Fuming
    Huang, Han
    Shi, Zhen
    2022 IEEE 7TH INTERNATIONAL CONFERENCE ON INTELLIGENT TRANSPORTATION ENGINEERING, ICITE, 2022, : 468 - 472
  • [4] Traffic Sign Detection and Recognition Based on Deep Learning
    Zhang, H.
    Zhao, J.
    ENGINEERING LETTERS, 2022, 30 (02) : 666 - 673
  • [5] Lightweight Traffic Sign Recognition and Detection Algorithm Based on Improved YOLOv5s
    Liu, Fei
    Zhong, Yanfen
    Qiu, Jiawei
    LASER & OPTOELECTRONICS PROGRESS, 2024, 61 (24)
  • [6] Application of improved YOLOv5s algorithm in traffic sign detection and recognition
    Guo, Junbin
    Yu, Lin
    Yu, Chuanqiang
    Guofang Keji Daxue Xuebao/Journal of National University of Defense Technology, 2024, 46 (06): : 123 - 130
  • [7] Research on traffic sign detection algorithm based on deep learning
    Wang, Quan
    Fu, Weiping
    CONCURRENCY AND COMPUTATION-PRACTICE & EXPERIENCE, 2018, 30 (22):
  • [8] Traffic Sign Detection Algorithm Based on Improved YOLOv5-S
    Liu, Haibin
    Zhang, Youbing
    Zhou, Kui
    Zhang, Yufeng
    Lyu, Sheng
    Computer Engineering and Applications, 2024, 60 (05) : 200 - 209
  • [9] A Lightweight Traffic Sign Detection Algorithm Based on YOLOv5s
    Sun, Jiahao
    Liu, Zhengda
    Liang, Jifeng
    2024 3RD INTERNATIONAL CONFERENCE ON ROBOTICS, ARTIFICIAL INTELLIGENCE AND INTELLIGENT CONTROL, RAIIC 2024, 2024, : 358 - 362
  • [10] Traffic Sign Recognition Based on the YOLOv3 Algorithm
    Gong, Chunpeng
    Li, Aijuan
    Song, Yumin
    Xu, Ning
    He, Weikai
    SENSORS, 2022, 22 (23)