Opto-Electronic Engineering, Volume. 51, Issue 6, 240055-1(2024)

A traffic sign recognition method based on improved YOLOv5

Liguo Qu1...2,*, Xin Zhang1, Zibao Lu1, Yuling Liu1 and Guohao Chen3 |Show fewer author(s)
Author Affiliations
  • 1School of Physics and Electronic Information, Anhui Normal University, Wuhu, Anhui 241002, China
  • 2Anhui Provincial Engineering Research Center for Information Fusion and Control of Intelligent Robots, Wuhu, Anhui 241002, China
  • 3Wuhan Mingke Rail Transit Equipment Co., Ltd., Wuhan, Hubei 430074, China
  • show less

    Traffic sign detection is an important link in the field of autonomous driving, and given the problems of missed detections, false detections, many model parameters, and common and complex representative real environment conditions, such as poor robustness in foggy days, an improved YOLOv5 micro-target traffic sign recognition algorithm was proposed. Firstly, the dataset was atomized to adapt to the accurate identification in the foggy weather, and the PC3 feature extraction module was constructed by using a lighter partial convolution (PConv), and then the Extended Feature Pyramid Network (EFPN) was proposed in the neck network Finally, Focal-EIOU is introduced to replace CIOU as the loss function to solve the problem of false detection and missed detection of micro targets, and the CBAM attention mechanism is embedded to realize the lightweight model and significantly improves the feature extraction ability of the network model. Compared with the original YOLOv5 algorithm, the improved model is increased by 8.9% and 4.4% respectively on P and mAP0.5, the number of parameters is reduced by 44.4%, and the FPS value on NVIDIA 3080 device is 151.5, which can meet the real-time detection of traffic signs in the real scenes.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Liguo Qu, Xin Zhang, Zibao Lu, Yuling Liu, Guohao Chen. A traffic sign recognition method based on improved YOLOv5[J]. Opto-Electronic Engineering, 2024, 51(6): 240055-1

    Download Citation

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

    Category:

    Received: Mar. 7, 2024

    Accepted: Jun. 4, 2024

    Published Online: Oct. 21, 2024

    The Author Email: Qu Liguo (曲立国)

    DOI:10.12086/oee.2024.240055

    Topics