Wavelet guided real time detection transformer with sparse attention
摘要
Real-time detection transformer (RT-DETR) has been proposed as a promising approach for real-time object detection. However, the performance of RT-DETR is limited due to: (a) the insufficient receptive fields of PResnet backbone, (b) lack of multi-level feature capabilities in Transformer encoder, and (c) deficient understanding of channel-wise information. To address these challenges, we introduce a wavelet guided real-time detection transformer with sparse attention (WS-RTDT) that integrates wavelet transform and sparse cubic attention mechanism for real-time object detection. Specifically, it first exploits a wavelet-guided convolution to expand the respective field of backbone to capture both high-frequency details and low-frequency contextual information. A sparse spatial encoder (SSE) is then proposed to learn feature maps of different scales and extract the multi-level features from images which employs a sparse cubic attention mechanism to process spatial and channel-wise feature correlation in an efficient manner. Experiment results demonstrate that WS-RTDT achieves the state-of-the-art performance in real-time object detection tasks, outperforming existing mainstream detectors in terms of efficiency and accuracy. Source code and pre-trained models are available at https://github.com/he13689/WSDETR.