中文

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

计算机视觉与模式识别 2021-03-19 v4

摘要

DETR 最近被提出以消除目标检测中许多手工设计组件的需求,同时展现出良好的性能。然而,由于 Transformer 注意力模块在处理图像特征图方面的局限性,它存在收敛慢和特征空间分辨率有限的问题。为缓解这些问题,我们提出了 Deformable DETR,其注意力模块仅关注参考点周围的一小组关键采样点。Deformable DETR 可以用少 10 倍的训练轮数取得比 DETR 更好的性能(尤其是在小物体上)。在 COCO 基准上的大量实验证明了我们方法的有效性。代码发布于 https://github.com/fundamentalvision/Deformable-DETR。

关键词

引用

@article{arxiv.2010.04159,
  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},
  journal= {arXiv preprint arXiv:2010.04159},
  year   = {2021}
}

备注

ICLR 2021 Oral