• Laser & Optoelectronics Progress
  • Vol. 61, Issue 24, 2412004 (2024)
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
    DOI: 10.3788/LOP240672 Cite this Article Set citation alerts
    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 Copy Citation Text show less

    Abstract

    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.
    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