Back to Mmdetection

Deformable DETR

configs/deformable_detr/README.md

3.3.04.4 KB
Original Source

Deformable DETR

Deformable DETR: Deformable Transformers for End-to-End Object Detection

<!-- [ALGORITHM] -->

Abstract

DETR has been recently proposed to eliminate the need for many hand-designed components in object detection while demonstrating good performance. However, it suffers from slow convergence and limited feature spatial resolution, due to the limitation of Transformer attention modules in processing image feature maps. To mitigate these issues, we proposed Deformable DETR, whose attention modules only attend to a small set of key sampling points around a reference. Deformable DETR can achieve better performance than DETR (especially on small objects) with 10 times less training epochs. Extensive experiments on the COCO benchmark demonstrate the effectiveness of our approach.

<div align=center> </div>

Results and Models

BackboneModelLr schdbox APConfigDownload
R-50Deformable DETR50e44.3configmodel | log
R-50+ iterative bounding box refinement50e46.2configmodel | log
R-50++ two-stage Deformable DETR50e47.0configmodel | log

NOTE

  1. All models are trained with batch size 32.
  2. The performance is unstable. Deformable DETR and iterative bounding box refinement may fluctuate about 0.3 mAP. two-stage Deformable DETR may fluctuate about 0.2 mAP.

Citation

We provide the config files for Deformable DETR: Deformable DETR: Deformable Transformers for End-to-End Object Detection.

latex
@inproceedings{
zhu2021deformable,
title={Deformable DETR: Deformable Transformers for End-to-End Object Detection},
author={Xizhou Zhu and Weijie Su and Lewei Lu and Bin Li and Xiaogang Wang and Jifeng Dai},
booktitle={International Conference on Learning Representations},
year={2021},
url={https://openreview.net/forum?id=gZ9hCDWe6ke}
}