How to strike a balance between detection speed and recognition accuracy has become a major challenge in real-time object detection. In this research, the YOLOv5 (You Only Look Once version 5) model was lightweight and optimized to improve the detection speed and accuracy of the network. To prune the backbone and neck are to simplify the network structure and reduce the parameters. The lightweight structure of the C3 module was designed and incorporated into the attention mechanism to improve the feature extraction capability of the network. For the public traffic sign dataset, the label assignment strategy and the loss function of YOLOv5 were refined to alleviate the imbalance between positive and negative samples and to better compute the loss, resulting in more stable and efficient training. Compared with other mainstream single-stage models, it achieves a better trade-off between speed and accuracy. With only 0.85 M parameters, 91.9% of mAP (mean average precision) and 360 FPS (frames per second) were achieved, which were 16.26% mAP and 26.67 FPS higher than the conventional YOLOv5n, respectively. The performance of our lightweight model in traffic sign detection far exceeds the most advanced achievements.


    Zugriff

    Zugriff prüfen

    Verfügbarkeit in meiner Bibliothek prüfen

    Bestellung bei Subito €


    Exportieren, teilen und zitieren



    Titel :

    A Novel Lightweight Traffic Sign Recognition Model Based on YOLOv5


    Weitere Titelangaben:

    J. Transp. Eng., Part A: Systems


    Beteiligte:
    Li, Wenju (Autor:in) / Zhang, Gan (Autor:in) / Cui, Liu (Autor:in)


    Erscheinungsdatum :

    2023-05-01




    Medientyp :

    Aufsatz (Zeitschrift)


    Format :

    Elektronische Ressource


    Sprache :

    Englisch




    Traffic Sign Recognition Algorithm Based on Improved YOLOv5

    Sang, Zhengxiao / Xia, Fuming / Huang, Han et al. | IEEE | 2022


    Traffic Sign Recognition Application Using Yolov5 Architecture

    Snegireva, Daria / Perkova, Anastasiia | IEEE | 2021


    Road traffic sign recognition based on lightweight neural network

    Li, Le-yang / Yue, Qin / Luo, Rui-wen | SPIE | 2021


    Road traffic sign recognition based on lightweight neural network

    Li, Le-yang / Yue, Qin / Luo, Rui-wen | British Library Conference Proceedings | 2021


    Road Traffic Vehicle Detection Method Using Lightweight YOLOv5 and Attention Mechanism

    Wang, Yunzhen / Ma, Hongbing / Li, Liangliang | IEEE | 2022