As the transportation industry continues to grow and traffic conditions become increasingly complex, ensuring driving safety has become a pressing issue of concern. Traffic sign recognition can offer real-time evaluation of road conditions and driving safety alerts for vehicles and ultimately reduce traffic safety accidents. This study proposes a class-imbalanced traffic sign recognition method based on improved YOLOv7. Firstly, the Coordinate Attention is incorporated into the YOLOv7 backbone to enhance the representation and localization ability of the target to be tested. Then, the larger feature layer is introduced to improve the recognition ability. Finally, the class imbalance problem in the dataset is addressed by using Cost-sensitive softmax cross-entropy loss. The improved YOLOv7 network has been shown through experiments to enhance the detection capability of traffic signs effectively, reaching 90.54% mAP and 86.78% F1-Score.


    Access

    Check access

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

    Class-imbalanced traffic sign recognition based on improved YOLOv7


    Contributors:

    Conference:

    International Conference on Image, Signal Processing, and Pattern Recognition (ISPP 2023) ; 2023 ; Changsha, China


    Published in:

    Proc. SPIE ; 12707


    Publication date :

    2023-06-08





    Type of media :

    Conference paper


    Type of material :

    Electronic Resource


    Language :

    English



    Traffic Sign Detection Algorithm Based on Improved YOLOv7

    Wang, Haodong / Zhang, Xindong | IEEE | 2024


    YOLOv7-SC:A Traffic Sign Detection Algorithm Based On SimAM and improved SPPCSPC-C

    Huang, Qian / Nong, Haiyuan / Peng, Wenquan et al. | IEEE | 2024


    Traffic Sign Recognition Algorithm Based on Improved YOLOv5

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


    Traffic sign recognition algorithm based on improved ResNet18

    Hu, Yixin / Ye, Qingyang / Zhu, Xuanqi et al. | SPIE | 2024


    Road pothole detection based on improved YOLOv7

    Zhang, Jianli / Lei, Jiaofei | SPIE | 2023