可变形 DETR:用于端到端目标检测的可变形 Transformer

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

代季峰 Jifeng Dai · SenseTime Research / USTC · 2020-10-08 · arXiv:2010.04159 ↗ · 被引 7789

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

最近提出的 DETR 消除了目标检测中许多手工设计的组件,并展示了良好的性能。然而,由于 Transformer 注意力模块在处理图像特征图时的局限性,DETR 存在收敛速度慢和特征空间分辨率有限的问题。为了解决这些问题,我们提出了可变形 DETR,其注意力模块仅关注参考点周围的一小组关键采样点。可变形 DETR 在训练周期减少 10 倍的情况下,性能优于 DETR(尤其在小目标上)。在 COCO 基准上的大量实验证明了我们方法的有效性。代码已发布在 https://github.com/fundamentalvision/Deformable-DETR。

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.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 9)

阅读逐段中英对照全文 →