TSS-YOLO: A timely swift small traffic sign detection model
摘要
Lightweight object detection models are increasingly important for traffic sign recognition, particularly in real-time applications on resource-constrained devices. This paper presents TSS-YOLO (Timely Swift Small-YOLO), a compact detection framework designed to address the challenges of small object detection while maintaining computational efficiency. The proposed model incorporates three core components: a Feature Extraction and Space-to-Depth (FESD) module that improves the representation of fine-grained features with reduced computational cost; a Weighted Feature Fusion (WFF) module that facilitates more balanced multi-scale feature integration; and a detection head based on a Non-Maximum Suppression-Free (NMS-Free) inference strategy with the large-object branch removed, enhancing both efficiency and small-object performance. Experimental results on the TT100K and CCTSDB datasets show that TSS-YOLO achieves better accuracy than YOLOv10s, with reductions of 5.5 million parameters and 10.9 GFLOPs in computation. These results suggest that TSS-YOLO offers a practical balance between accuracy and efficiency for real-time traffic sign detection.