Real-Time Traffic Sign Detection Based on Yolov5-MGC

被引:0
|
作者
Zhu Ningke [1 ]
Ge Qing [2 ]
Wang Hanwen [1 ]
Yu Pengfei [1 ]
机构
[1] Yunnan Univ, Sch Informat Sci & Engn, Kunming 650504, Yunnan, Peoples R China
[2] Kunming Publ Secur Traff Management Informat Appl, Kunming 650000, Yunnan, Peoples R China
关键词
traffic sign detection; lightweight network; CARAFE operator; global and local fusion; real-time detection;
D O I
10.3788/LOP231703
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
For handling low detection accuracy, poor real-time performance, and large model size of small target traffic sign detection, a real-time road traffic sign detection algorithm based on Yolov5 is proposed. First, the inverted residual structure in Mobilenetv3 was improved and applied to the backbone network of Yolov5 to align with the lightweight network design. Then, the lightweight upsampling universal operator CARAFE (content-aware ReAssembly of FEatures) replaced the nearest neighbor interpolation upsampling module of the original network, reducing the loss of upsampling information and increasing the receptive field. Finally, global and local fusion attention (GLFA) was used to focus on the global and local scales to enhance the sensitivity of the network for small target objects. Experiments on the self-made Chinese multiclass traffic sign dataset (CMTSD) show that the enhanced algorithm improves mean accuracy precision (mAP) @0. 5 by 2. 58 percentage points based on the model size reduction of 8. 76 MB compared with the algorithm before enhancement. Furthermore, the detection speed reaches 62. 59 frame/s. Compared with other mainstream object detection algorithms, the proposed algorithm exhibits certain advantages in detection accuracy, speed, and model volume and performs better in real complex traffic scenes.
引用
收藏
页数:10
相关论文
共 29 条
  • [1] Multi-class US traffic signs 3D recognition and localization via image-based point cloud model using color candidate extraction and texture-based recognition
    Balali, Vahid
    Jahangiri, Arash
    Machiani, Sahar Ghanipoor
    [J]. ADVANCED ENGINEERING INFORMATICS, 2017, 32 : 263 - 274
  • [2] Garbage classification system based on improved ShuffleNet v2
    Chen, Zhichao
    Yang, Jie
    Chen, Lifang
    Jiao, Haining
    [J]. RESOURCES CONSERVATION AND RECYCLING, 2022, 178
  • [3] Attentional Feature Fusion
    Dai, Yimian
    Gieseke, Fabian
    Oehmcke, Stefan
    Wu, Yiquan
    Barnard, Kobus
    [J]. 2021 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION WACV 2021, 2021, : 3559 - 3568
  • [4] [郭磊 Guo Lei], 2022, [电子科技大学学报, Journal of University of Electronic Science and Technology of China], V51, P251
  • [5] He KM, 2014, LECT NOTES COMPUT SC, V8691, P346, DOI [arXiv:1406.4729, 10.1007/978-3-319-10578-9_23]
  • [6] Searching for MobileNetV3
    Howard, Andrew
    Sandler, Mark
    Chu, Grace
    Chen, Liang-Chieh
    Chen, Bo
    Tan, Mingxing
    Wang, Weijun
    Zhu, Yukun
    Pang, Ruoming
    Vasudevan, Vijay
    Le, Quoc V.
    Adam, Hartwig
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 1314 - 1324
  • [7] Hu J, 2018, PROC CVPR IEEE, P7132, DOI [10.1109/TPAMI.2019.2913372, 10.1109/CVPR.2018.00745]
  • [8] Hu Junping, 2023, Computer Engineering and Applications, P185, DOI 10.3778/j.issn.1002-8331.2206-0503
  • [9] Hu Zhaohua, 2023, Computer Engineering and Applications, P82, DOI 10.3778/j.issn.1002-8331.2207-0307
  • [10] Ioffe S, 2015, Arxiv, DOI [arXiv:1502.03167, 10.48550/ARXIV.1502.03167]