Optics and Precision Engineering, Volume. 31, Issue 9, 1366(2023)

Road traffic sign recognition algorithm based on improved YOLOv4

Daxiang LI... Zhongheng SU* and Ying LIU |Show fewer author(s)
Author Affiliations
  • College of Communication and Information Engineering, Xi'an University of Posts and Telecommunication, Xi'an710121, China
  • show less

    To address the low recognition accuracy resulting from multiple scale changes in the traffic signs of complex scenes, an improved YOLOv4 algorithm is proposed. First, an attention-driven scale-aware feature extraction module is designed, and the range of receptive fields in each layer is widened to obtain more fine-grained multi-scale features by constructing a hierarchical connection mode similar to the residual structure; this is followed by the generation of a pair of attention maps with directional-aware and position-sensitive characteristics under the attention drive so that the network can focus on key areas with more discrimination. Following this, a feature-aligned pyramid convolution feature fusion module is constructed, and the feature offset between adjacent scale feature maps is obtained via convolution for feature alignment. Finally, the network adaptively learns the optimal feature fusion mode through pyramid convolution and constructs a feature pyramid to identify traffic signs with different scales. Experimental results indicate that the recognition accuracy for the TT100K dataset is improved by 5.4% compared with that of the original YOLOv4 algorithm, which is superior to other recognition algorithms, and the FPS reaches 33.17. Thus, the proposed algorithm satisfies the requirements of accuracy and real-time performance for road traffic sign recognition.

    Tools

    Get Citation

    Copy Citation Text

    Daxiang LI, Zhongheng SU, Ying LIU. Road traffic sign recognition algorithm based on improved YOLOv4[J]. Optics and Precision Engineering, 2023, 31(9): 1366

    Download Citation

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

    Category: Information Sciences

    Received: Jul. 13, 2022

    Accepted: --

    Published Online: Jun. 6, 2023

    The Author Email: SU Zhongheng (Szh1998@stu.xupt.edu.cn)

    DOI:10.37188/OPE.20233109.1366

    Topics