/FeatAug-DETR

Official repository of paper: "FeatAug-DETR: Enriching One-to-Many Matching for DETRs with Feature Augmentation"

Apache License 2.0Apache-2.0

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