CAC-DETR: bridging heavy and light for real-time small object detection on UAVs
摘要
Real-time small object detection in unmanned aerial vehicle imagery serves critical applications but faces significant deployment challenges. The primary constraint lies in achieving optimal trade-offs between detection accuracy and computational efficiency on resource-limited edge-AI devices. To address small object detection accuracy while maintaining computational efficiency, we propose Convolutional Additive Context DEtection TRansformer (CAC-DETR), a novel detection framework based on Real-Time DEtection TRansformer (RT-DETR). Our framework introduces the CAC block, which comprises three core components: C2f, Convolutional Additive Token Mixer (CATM), and Convolutional Gated Linear Unit (CGLU). The diverse kernel sizes employed in CATM and CGLU provide flexible receptive field adjustment, while the pure convolutional architecture ensures optimal computational efficiency for edge deployment. Our proposed CAC-DETR achieves substantial performance improvements while reducing computational overhead compared to the benchmark RT-DETR. Experimental results on VisDrone demonstrate computational reductions of 30.3% in parameter complexity and 20.4% in GFLOPs, accompanied by significant accuracy improvements of 3.6%