HyperAIHyperAI

Command Palette

Search for a command to run...

Omni-DETR:基于Transformer的全监督目标检测

Pei Wang Zhaowei Cai Hao Yang Gurumurthy Swaminathan Nuno Vasconcelos Bernt Schiele Stefano Soatto

摘要

我们研究了全监督(omni-supervised)目标检测问题,该方法能够同时利用未标注数据、完全标注数据以及弱标注数据(如图像标签、目标数量、点标注等)进行目标检测。这一能力得益于一种统一的架构——Omni-DETR,其建立在近期学生-教师框架(student-teacher framework)以及端到端基于Transformer的目标检测方法的进展之上。在该统一架构下,通过基于二分图匹配的过滤机制,不同类型的弱标注可被有效利用,生成高精度的伪标签,供模型学习。实验结果表明,Omni-DETR在多个数据集和不同设置下均取得了当前最优的性能表现。我们还发现,弱标注有助于提升检测性能,而多种弱标注形式的混合使用,能够在标注成本与检测精度之间实现更优的权衡,优于传统的全标注方式。这些发现为构建包含混合标注的大规模目标检测数据集提供了有力支持。代码已开源,地址为:https://github.com/amazon-research/omni-detr


用 AI 构建 AI

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

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

HyperAI Newsletters

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