Deformable DETR: Deformable Transformers for End-to-End Object Detection
About
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. Code is released at https://github.com/fundamentalvision/Deformable-DETR.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Object Detection | COCO 2017 (val) | AP51.4 | 2454 | |
| Object Detection | COCO (test-dev) | mAP56.6 | 1195 | |
| Instance Segmentation | COCO 2017 (val) | -- | 1144 | |
| Object Detection | MS COCO (test-dev) | mAP@.571.9 | 677 | |
| Object Detection | COCO (val) | mAP49.8 | 613 | |
| Object Detection | LVIS v1.0 (val) | APbbox32.5 | 518 | |
| Object Detection | COCO v2017 (test-dev) | mAP52.3 | 499 | |
| Oriented Object Detection | DOTA v1.0 (test) | SV72.53 | 378 | |
| Video Object Detection | ImageNet VID (val) | mAP (%)55.4 | 341 | |
| Object Detection | MS-COCO 2017 (val) | -- | 237 |