HyperAIHyperAI

Command Palette

Search for a command to run...

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

Xizhou Zhu Weijie Su Lewei Lu Bin Li Xiaogang Wang Jifeng Dai

摘要

DETR 近期被提出,旨在消除目标检测中大量手工设计组件的需求,同时展现出良好的性能。然而,由于 Transformer 注意力模块在处理图像特征图时存在局限性,DETR 仍面临收敛速度慢和特征空间分辨率有限的问题。为缓解上述问题,我们提出了可变形 DETR(Deformable DETR),其注意力模块仅关注以参考点为中心的一小部分关键采样点。实验结果表明,与原始 DETR 相比,可变形 DETR 在相同训练轮次下性能更优,尤其在小目标检测方面表现显著提升,且仅需其 1/10 的训练轮次即可达到相近或更优效果。在 COCO 基准数据集上的大量实验充分验证了该方法的有效性。代码已开源,地址为:https://github.com/fundamentalvision/Deformable-DETR


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
可变形DETR:用于端到端目标检测的可变形Transformer | 论文 | HyperAI超神经