Laser & Optoelectronics Progress, Volume. 61, Issue 24, 2412004(2024)

Lightweight Traffic Sign Recognition and Detection Algorithm Based on Improved YOLOv5s

Fei Liu1, Yanfen Zhong1,2,3、*, and Jiawei Qiu1
Author Affiliations
  • 1School of Civil Engineering and Transportation, Nanchang Hangkong University, Nanchang 330063, Jiangxi , China
  • 2Jiangxi Intelligent Building Engineering Research Centre, Nanchang 330063, Jiangxi , China
  • 3Nanchang Hangkong University Intelligent Construction Research Centre, Nanchang 330063, Jiangxi , China
  • show less

    To address the inadequate detection precision and computational efficiency of common traffic sign detection methods under poor lighting conditions, capturing small distant targets, and in complex backgrounds, this study introduces an enhanced YOLOv5s algorithm, named BMGE-YOLOv5s. The proposed method employs BoTNet (bottleneck Transformer network) to replace the original backbone network of YOLOv5s. It also designs a lightweight network, C3GBneckv2, which integrates the GhostNetv2 bottleneck and an efficient channel attention mechanism, reducing the number of parameters while significantly enhancing the feature extraction capability for traffic signs. To further enhance the accuracy of bounding box localization, the MPDIoU loss function is utilized. Experimental results indicate that the improved network model achieves a mean average precision of 93.1% at an intersection ratio threshold of 0.5, indicating an improvement of 3.3 percentage points over the baseline model on the same dataset. Moreover, the proposed model demonstrates a 9.375% decrease in floating-point operations, a ~25.98% decrease in the number of parameters, and a ~67.40% increase in detection speed. The proposed algorithm effectively balances robustness and real-time performance, showing a clear performance advantage over traditional methods.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Fei Liu, Yanfen Zhong, Jiawei Qiu. Lightweight Traffic Sign Recognition and Detection Algorithm Based on Improved YOLOv5s[J]. Laser & Optoelectronics Progress, 2024, 61(24): 2412004

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Instrumentation, Measurement and Metrology

    Received: Feb. 5, 2024

    Accepted: Apr. 30, 2024

    Published Online: Dec. 13, 2024

    The Author Email: Yanfen Zhong (70016@nchu.edu.cn)

    DOI:10.3788/LOP240672

    CSTR:32186.14.LOP240672

    Topics