YOLOv8s-DDA: An Improved Small Traffic Sign Detection Algorithm Based on YOLOv8s

Abstract

In the realm of traffic sign detection, challenges arise due to the small size of objects, complex scenes, varying scales of signs, and dispersed objects. To address these problems, this paper proposes a small object detection algorithm, YOLOv8s-DDA, for traffic signs based on an improved YOLOv8s. Specifically, the C2f-DWR-DRB module is introduced, which utilizes an efficient two-step method to capture multi-scale contextual information and employs a dilated re-parameterization block to enhance feature extraction quality while maintaining computational efficiency. The neck network is improved by incorporating ideas from ASF-YOLO, enabling the fusion of multi-scale object features and significantly boosting small object detection capabilities. Finally, the original IoU is replaced with Wise-IoU to further improve detection accuracy. On the TT100K dataset, the YOLOv8s-DDA algorithm achieves mAP@0.5 of 87.2%, mAP@0.5:0.95 of 68.3%, precision of 85.2%, and recall of 80.0%, with a 5.4% reduction in parameter count. The effectiveness of this algorithm is also validated on the publicly available Chinese traffic sign detection dataset, CCTSDB2021.