4 个月前

基于对比学习的弱监督目标检测中的目标发现

基于对比学习的弱监督目标检测中的目标发现

摘要

弱监督目标检测(WSOD)是一项仅使用图像级注释训练的模型来检测图像中目标的任务。当前最先进的模型受益于自监督实例级监督,但由于弱监督不包含数量或位置信息,最常见的“argmax”标记方法通常会忽略许多目标实例。为了解决这一问题,我们提出了一种新的多实例标记方法,称为目标发现。此外,我们引入了一种在没有实例级信息可供采样的情况下使用的新型对比损失函数,称为弱监督对比损失(WSCL)。WSCL通过利用同一类别中的嵌入向量的一致特征来构建可靠的目标发现相似度阈值。因此,我们在MS-COCO 2014和2017以及PASCAL VOC 2012数据集上取得了最新的最佳结果,并在PASCAL VOC 2007数据集上获得了具有竞争力的结果。

代码仓库

jinhseo/od-wscl
官方
pytorch

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

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