Research on traffic sign detection based on improved YOLOv9
Aiming at the current mainstream traffic sign target detection algorithm's low accuracy, misdetection and omission of small target detection in complicated environments, this paper presents an improved traffic sign detection algorithm based on YOLOv9. AKConv is used to replace the Conv module in RepNCSPELAN4, which maintains the detection accuracy while lightening the weight. Meanwhile, Focal-EIoU Loss is proposed instead of the original regression loss function Clou Loss, this accelerates convergence and raises the accuracy of the regression by dividing the aspect ratio's loss term into the difference between the minimum outer frame's width and height and the anticipated width and height. In addition, the feature extraction capability of the network and the detection accuracy are further strengthened by adding the Convolutional Block Attention Module (CBAM) attention mechanism. On the TT100k traffic signage dataset, the improved algorithm achieves performance metrics of 92.3% and 91.5% in terms of accuracy and [email protected], which are 3.1% and 3.7% higher compared to the original YOLOv9 algorithm. Moreover, the algorithm's misdetection and missed detection problems in complex environments are significantly improved, and the comprehensive detection performance is significantly higher than that of the comparison algorithms, which has greater practical value.
Authors
- Mingxuan ZHANG
Institutions
- Chongqing University of Posts and Telecommunications (CN)
Publication Details
- Published
- 2024-10-09
- DOI
- https://doi.org/10.1117/12.3045722
- Citations
- 3
- Primary Topic
- E-commerce and Technology Innovations
- Type
- article
- Field-Weighted Citation Impact
- 2.03