Abstract:In this paper, we provide the observation that too few queries assigned as positive samples in DETR with one-to-one set matching leads to sparse supervision on the encoder's output which considerably hurt the discriminative feature learning of the encoder and vice visa for attention learning in the decoder. To alleviate this, we present a novel collaborative hybrid assignments training scheme, namely $\mathcal{C}$o-DETR, to learn more efficient and effective DETR-based detectors from versatile label assignment manners. This new training scheme can easily enhance the encoder's learning ability in end-to-end detectors by training the multiple parallel auxiliary heads supervised by one-to-many label assignments such as ATSS and Faster RCNN. In addition, we conduct extra customized positive queries by extracting the positive coordinates from these auxiliary heads to improve the training efficiency of positive samples in the decoder. In inference, these auxiliary heads are discarded and thus our method introduces no additional parameters and computational cost to the original detector while requiring no hand-crafted non-maximum suppression (NMS). We conduct extensive experiments to evaluate the effectiveness of the proposed approach on DETR variants, including DAB-DETR, Deformable-DETR, and DINO-Deformable-DETR. The state-of-the-art DINO-Deformable-DETR with Swin-L can be improved from 58.5% to 59.5% AP on COCO val. Surprisingly, incorporated with ViT-L backbone, we achieve 66.0% AP on COCO test-dev and 67.9% AP on LVIS val, outperforming previous methods by clear margins with much fewer model sizes. Codes are available at \url{<a class="link-external link-https" href="https://github.com/Sense-X/Co-DETR" rel="external noopener nofollow">this https URL</a>}.

Teach-DETR: Better Training DETR with Teachers

DETRs with Collaborative Hybrid Assignments Training

OD-DETR: Online Distillation for Stabilizing Training of Detection Transformer

DETRDistill: A Universal Knowledge Distillation Framework for DETR-families.

DISTILLING DETR-LIKE DETECTORS WITH INSTANCE-AWARE FEATURE

DN-DETR: Accelerate DETR Training by Introducing Query DeNoising

DETR Doesn't Need Multi-Scale or Locality Design

DETR Does Not Need Multi-Scale or Locality Design

Enhancing Your Trained DETRs with Box Refinement

D$^3$ETR: Decoder Distillation for Detection Transformer

Revisiting DETR Pre-training for Object Detection

AugDETR: Improving Multi-scale Learning for Detection Transformer

Decoupled DETR: Spatially Disentangling Localization and Classification for Improved End-to-End Object Detection

DETR++: Taming Your Multi-Scale Detection Transformer

Knowledge Distillation for Detection Transformer with Consistent Distillation Points Sampling

Rank-DETR for High Quality Object Detection

MS-DETR: Efficient DETR Training with Mixed Supervision

Language-aware Multiple Datasets Detection Pretraining for DETRs

Group DETR: Fast DETR Training with Group-Wise One-to-Many Assignment

FeatAug-DETR: Enriching One-to-Many Matching for DETRs with Feature Augmentation

Team DETR: Guide Queries As a Professional Team in Detection Transformers