MSCRep-YOLO: A new method for traffic sign recognition
This paper primarily addresses the challenges of small-scale traffic signs and real-time requirements in traffic sign recognition. We propose an improved YOLOv8n-based algorithm that incorporates a small object detection layer, a multi-scale convolutional attention mechanism, and a lightweight RepViT architecture, effectively enhancing detection capabilities for smallsized traffic signs while maintaining real-time performance. Additionally, the MPDIoU loss function replaces the Complete Intersection over Union loss function to further improve boundary regression performance. Tested on the TT100K dataset, the model achieves an 6.3% increase in precision, a 7% increase in recall, and an 7% increase in mean average precision, with only 78.82% of the original model’s parameters. These results demonstrate the model’s practical value in real-time traffic sign recognition tasks, especially in handling small-scale signs.
Authors
- Chengzhi Ma
- Xinying Chen (ORCID: https://orcid.org/0000-0002-7169-0614)
Institutions
- Dalian Jiaotong University (CN)
Publication Details
- Published
- 2024-12-20
- DOI
- https://doi.org/10.1109/aiim64537.2024.10934373
- Primary Topic
- Vehicle License Plate Recognition
- Type
- article
- Field-Weighted Citation Impact
- 0.00