TCF-DETR: multi-scale token-channel fusion transformer for enhanced small object detection
摘要
Small object detection is crucial in areas such as autonomous driving and surveillance. However, it faces challenges of feature information loss and semantic insufficiency in multi-scale scenarios. To address these issues, we propose a Multi-scale Token-Channel Fusion Detection Transformer (TCF-DETR) that leverages Channel Aggregation Token Mixer (CATM) and Squeeze-Excitation Token Mixer (SETM) modules. First, multi-scale features are extracted from the backbone network and the highest semantic feature is refined through context intra-scale interactions, effectively enhancing feature representation. Second, a cross-scale fusion network based on the token-channel mixer is proposed to fuse multi-scale features. In the top-down path, CATM enhances the semantic fusion of low-level features, while SETM, used in the bottom-up path, improves the fusion of high-level features with localization information. Finally, the fused features are decoded and mapped to category labels and bounding box coordinates. Extensive experiments on the VisDrone, TinyPerson, and Cityscapes datasets shows that TCF-DETR achieves mAP50 values of 49.9%, 24.2%, and 59.5%, respectively, while maintaining a relatively low parameter count of 19.8M. The proposed model not only enhances small object detection but also shows broader applicability. The code and dataset are available at: https://github.com/Leihuan42/TCF-DETR.