DINO:通过改进的去噪锚框实现端到端目标检测的 DETR

DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection

张磊 Lei Zhang · IDEA Research / HKUST · 2022-03-07 · arXiv:2203.03605 ↗ · 被引 2849

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

摘要 · Abstract

我们提出了 DINO(改进去噪锚框的 DETR),一种最先进的端到端目标检测器。DINO 通过使用对比方式进行去噪训练、混合查询选择方法进行锚点初始化以及两次前向预测方案进行边界框预测,在性能和效率上优于之前的 DETR 类模型。在 COCO 数据集上,使用 ResNet-50 骨干网络和多尺度特征,DINO 在 12 个 epoch 中达到 49.4 AP,在 24 个 epoch 中达到 51.3 AP,相比之前最好的 DETR 类模型 DN-DETR,分别显著提升了+6.0 AP 和+2.7 AP。DINO 在模型大小和数据规模上具有良好的可扩展性。无需额外技巧,在使用 SwinL 骨干网络在 Objects365 数据集上预训练后,DINO 在 COCO val2017(63.2 AP)和 test-dev(63.3 AP)上均取得了最佳结果。与排行榜上的其他模型相比,DINO 在取得更好结果的同时显著减小了模型大小和预训练数据规模。我们的代码将在 https://github.com/IDEACVR/DINO 上开源。

We present DINO (\textbf{D}ETR with \textbf{I}mproved de\textbf{N}oising anch\textbf{O}r boxes), a state-of-the-art end-to-end object detector. % in this paper. DINO improves over previous DETR-like models in performance and efficiency by using a contrastive way for denoising training, a mixed query selection method for anchor initialization, and a look forward twice scheme for box prediction. DINO achieves $49.4$AP in $12$ epochs and $51.3$AP in $24$ epochs on COCO with a ResNet-50 backbone and multi-scale features, yielding a significant improvement of $\textbf{+6.0}$\textbf{AP} and $\textbf{+2.7}$\textbf{AP}, respectively, compared to DN-DETR, the previous best DETR-like model. DINO scales well in both model size and data size. Without bells and whistles, after pre-training on the Objects365 dataset with a SwinL backbone, DINO obtains the best results on both COCO \texttt{val2017} ($\textbf{63.2}$\textbf{AP}) and \texttt{test-dev} (\textbf{$\textbf{63.3}$AP}). Compared to other models on the leaderboard, DINO significantly reduces its model size and pre-training data size while achieving better results. Our code will be available at \url{https://github.com/IDEACVR/DINO}.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 16)

阅读逐段中英对照全文 →