FeatAug-DETR

This is the official implementation of the paper "[FeatAug-DETR: Enriching One-to-Many Matching for DETRs with Feature Augmentation]".

Authors: Rongyao Fang, Peng Gao, Aojun Zhou, Yingjie Cai, Si Liu, Jifeng Dai, Hongsheng Li

The codes and model checkpoints will be released very soon!

Model ZOO

We provide a set of results:

Models with the ResNet-50 backbone

Name	Backbone	query	epochs	AP
Deformable-DETR	R50	300	12	43.7
Deformable-DETR	R50	300	36	46.8
Deformable-DETR + tricks	R50	300	12	47.0
Deformable-DETR + tricks	R50	300	36	49.0
Deformable-DETR + tricks + DataAug	R50	300	48	50.0
Deformable-DETR + tricks + FeatAug-FC	R50	300	12	48.7
Deformable-DETR + tricks + FeatAug-FC	R50	300	24	49.9
H-Deformable-DETR + tricks + FeatAug-Flip	R50	300	12	49.4
H-Deformable-DETR + tricks + FeatAug-Flip	R50	300	24	50.4

Models with Swin Transformer backbones

Name	Backbone	query	epochs	AP
Deformable-DETR	Swin Tiny	300	12	45.3, 46.0
Deformable-DETR	Swin Tiny	300	36	49.0, 49.6
Deformable-DETR + tricks	Swin Tiny	300	12	49.3
Deformable-DETR + tricks	Swin Tiny	300	36	51.8
Deformable-DETR + tricks + DataAug	Swin Tiny	300	48	53.3
Deformable-DETR + tricks + FeatAug-FC	Swin Tiny	300	12	50.9
Deformable-DETR + tricks + FeatAug-FC	Swin Tiny	300	24	52.7
Deformable-DETR	Swin Large	300	12	51.0
Deformable-DETR	Swin Large	300	36	53.7
Deformable-DETR + tricks	Swin Large	300	12	54.5
Deformable-DETR + tricks	Swin Large	300	36	56.3
Deformable-DETR + tricks + DataAug	Swin Large	300	48	57.0
Deformable-DETR + tricks + FeatAug-FC	Swin Large	300	12	55.8
Deformable-DETR + tricks + FeatAug-FC	Swin Large	300	24	57.1
Deformable-DETR + tricks + FeatAug-FC [topk=300]	Swin Large	900	24	57.6
H-Deformable-DETR + tricks + FeatAug-Flip	Swin Large	300	12	56.4
H-Deformable-DETR + tricks + FeatAug-Flip	Swin Large	300	24	57.6
H-Deformable-DETR + tricks + FeatAug-Flip [topk=300]	Swin Large	900	24	58.3