出发点是制定一种更加通用的目标检测问题,目的是借助于大量的image-caption数据来覆盖更多的object concept,使得object detection不再受限于带标注数据的少数类别,从而实现更加泛化的object detection,识别出更多novel的物体类别。

文章目录

  • 一、背景 & 动机
  • 二、Open-Vocabulary的简单引入
  • 三、Open-Vocabulary / zero-shot /weakly supervised之间的差异
  • 四、论文的核心
  • 五、论文流程
  • 六、模型结果对比

一、背景 & 动机

尽管深度神经网络在目标检测方面具有显著的准确性,但由于监管要求,它们的训练和拓展成本很高。特别是,学习更多的对象类别通常需要按比例增加更多的边界框注释。

例如Faster RCNN及其令人印象深刻的准确性,但训练这样的模型需要昂贵且耗时的人力监督,特别是,需要为每个ROI的对象类别手动标注至少数千个边界框。尽管之前很多机构已经完成了object detection上benchmark的建立,并且公开了这些有价值的数据集,例如Open Images和MSCOCO,这些数据集描述了一些有限的对象类别。但如果我们想将目标检测从600个类别扩展到60000个类别,那么我们需要100倍数据资源的标注,这使得把目标检测拓展到开放世界里变得遥不可及。

然而,人类通过自然监督学会毫不费力地识别和定位物体,即探索视觉世界和倾听他人描述情况。我们人类具有终生学习的能力,我们捕捉到视觉信息后,会将它们与口语联系起来,从而产生了丰富的视觉和语义词汇,这些词汇不仅可以用于检测物体,而且可以用来拓展模型的表达能力。尽管在对象周边绘制边界框不是人类自然学习的任务,但他们可以通过使用少量例子快速学习它,并将其快速泛化到所有类型的对象,而不需要每个对象类的示例,这就是Open Vocabulary Object Detection这一问题的motivation所在。

二、Open-Vocabulary的简单引入

Open-Vocabulary Object Detection (OVD)可以翻译为**“面向开放词汇下的目标检测”,**该任务和zero-shot object detection非常类似,核心思想都是在可见类(base class)的数据上进行训练,然后完成对不可见类(unseen/ target)数据的识别和检测。

弱监督和zero-shot学习技术已经被探索用于在较少监督的情况下将对象检测器扩展到更多类别,但它们还没有像监督模型那样成功和广泛应用。

在本文中,我们提出了一种新的目标检测问题的公式—Open-Vocabulary Object Detection,它比弱监督和zero-shot方法更通用、更实用、更有效。

我们提出了一种新的方法来训练对象检测器,使用限定对象类别的包围框注释,以及以显著较低的成本覆盖更多的图像—标题对。

我们表明,所提出的方法可以检测和定位在训练过程中没有提供边界框注释的对象,其精度明显高于zero-shot方法。

三、Open-Vocabulary / zero-shot /weakly supervised之间的差异


比较了一下三种setting,OVD和ZSD的区别应该就是在训练时,OVD可能用到target类的embedding信息,当然可能只说这些embedding信息包含在一堆caption中,谁也不知道里面有没有target信息,肯定不能给target类的bbox信息。而zero-shot完全没用到,weakly supervised就更直接了,直接利用子集来训练,从而强化泛化能力。

四、论文的核心

Open-Vocabulary工作的核心是利用image-caption数据来对视觉编码器进行pre-training。

由于caption中存在着丰富的用于图像区域等细粒度特征得描述单词和短语,能够覆盖更多的物体类别,因此经过大规模image-caption的预训练,Vision encoder便能够学习到更加泛化的视觉-语义对应空间。

因此训练好的vision encoder便可以用于替换faster rcnn中的encoder,提高检测模型的zero-shot检测能力。

五、论文流程


展开讨论预训练流程,整体的预训练流程有些类似于PixelBert,可参考如下:

  • 第一步:学习的视觉与文本的联系。通过训练一个现象变换层来把视觉空间的特征转换到文本空间,来充当一个V2L(vision to language)的模块,负责把视觉特征变换到文本空间去。输入的image-caption对首先各自经过各自模态的encoder,图像则是细分得到每个区域的特征,然后进一步经过V2L变换。之后,两个模态的特征concat起来之后送入多模态的transformer,得到的输出是视觉区域特征以及训练好的BERT,这样的模型具有泛化能力,而图像encoder是常用的resnet50。
  • 第二步:利用常规的目标检测框架:Faster RCNN,进行模型训练。为了保证延续性,backbone采用上一阶段中训练好的renet50,每个proposal的特征经过V2L变换之后又与类别标签的文本特征计算相似度来进行分类。事实上就是把回归问题转换成分类问题。
  • 第三步:把要检测的新类别加入文本的特征向量中做匹配。

一旦预训练结束后,trained vision encoder和trained V2L层,便可以替换至Faster RCNN框架中,通过在base数据集上进行finetune vision encoder,使其适配ROI区域特征,固定V2L层,保持其学习到的泛化的视觉-语义空间,即可进行target类别数据的检测。

总结来看,OVR-CNN通过在image-caption数据集上的预训练,学习到了丰富的文本词汇和图像区域表征,这样泛化的表征空间覆盖的物体类别,是远超过现阶段的带标注的目标检测数据集中物体的类别数。

整个模型框架跟Faster RCNN几乎完全一样,只是将最后的cls head换成了V2L,也就是换成了一个将visual feature投影到text embedding space的投影矩阵。

六、模型结果对比


可以发现,相较于原有的zero-shot的detection,模型的泛化性能显然是更强的。

根据上表,其实我们可以发现:

  • ZSD的检测效果差(map不高)

主要原因,我认为就是对于没有任何未知类的例子经过训练,OVD 应该是会有部分未知类通过image-caption dataset 训练可得知,因此从现有基类的特征其实很难推出新类。

  • WSD 定位效果不好

个人分析认为,他从没有注释的图片很难学习到特征,就很难像OVD那样通过image-caption那样,至少有图像和文本方向的特征,再通过基类的相关有注释框的图片学习,就能很好的定位。

  • mixed supervision

其实同样存在上面的缺陷,在基类上进行训练,然后使用弱监督学习转移到目标类,这些方法通常会在基类上降低性能相反,Visual grounding和Vision-language transformers 就是来帮助解决作者的设想,通过 Vision-language transformers 可以提取文本和图像的特征,Visual grounding 则就是根据这些特征进行定位。

同时我们也发现,利用这种用image-captioning pair模式训练出来的特征,其中每个类别的特征更加显著,如下图和zero-shot obejct detection的baseline的对比:

【开放域目标检测】一:Open-Vocabulary Object Detection Using Captions论文讲解相关推荐

  1. Open-Vocabulary Object Detection Using Captions论文讲解

    文章目录 一.论文前言 二.提出原因 三.论文的核心 四.论文讲解 4.1 论文流程 4.2 OVD与之前相关的setting 4.3 结果对比 一.论文前言 目标检测是人工智能最突出的应用之一,也是 ...

  2. 目标检测20年(Object Detection in 20 Years)

    Object Detection in 20 Years: A Survey Zhengxia Zou, Zhenwei Shi, Member, IEEE, Yuhong Guo, and Jiep ...

  3. CVPR 2021 Oral | OWOD:面向开放世界的目标检测 Towards Open World Object Detection

    Towards Open World Object Detection 论文:https://arxiv.org/abs/2103.02603 代码:https://github.com/Joseph ...

  4. 【3D目标检测】Accurate Monocular Object Detection via Color-Embedded 3D Reconstruction for AD

    目录 概述 细节 整体流程 第一阶段:3D数据生成 第二阶段:3D边界框估计 前景分割 颜色嵌入 3D 边界框估计 损失函数 概述 本文是基于图像的3D目标检测方法. 和伪点云原论文在流程上是相同的, ...

  5. 实时目标检测--Pelee: A Real-Time Object Detection System on Mobile Devices

    Pelee: A Real-Time Object Detection System on Mobile Devices ICLR 2018 Code: https://github.com/Robe ...

  6. 目标检测开源代码汇总 object detection algorithm codes

    无人机目标检测 SlimYOLOv3: Narrower, Faster and Better for Real-Time UAV Applications https://arxiv.org/abs ...

  7. 【3D目标检测】Monocular 3D Object Detection with Pseudo-LiDAR Point Cloud

    目录 概述 整体流程 伪点云vs点云 2D-3D边界框的一致性约束 概述 本文的输入数据仅仅是单目图像,在方法上是融合了伪点云(Pseudo-LiDAR)的深度信息表示方法与Frustum Point ...

  8. AI_综述:3D目标检测于RGB-D(Object detection in RGB-D images)

    转载自:https://zhuanlan.zhihu.com/p/34887948 基于深度学习的图像目标检测(上) 基于深度学习的图像目标检测(下) 大牛讲堂 | 基于DenesBox的目标检测在自 ...

  9. 目标检测开源代码汇总 object detection algorithm codes (转)

    注:原文:https://blog.csdn.net/zhangjunhit/article/details/78901952 博主「O天涯海阁O」 无人机目标检测 SlimYOLOv3: Narro ...

最新文章

  1. Oracle 导出CSV、导出大数据
  2. 黄聪:在C#中如何使用资源中的图片
  3. vi/vim 中批量在行插入或删除指定字符
  4. 基于GRNN神经网络的数字预失真系统的FPGA实现
  5. 一套模块化数据中心培训资料
  6. 文件字符串变量插入linux,Linux Shell脚本实现在文件指定的行插入字符串
  7. orm设置bool型 python_详解python的ORM中Pony用法
  8. 宝塔面板部署python项目
  9. Windows Server 2008 RC1标准版
  10. oracle 查询优化
  11. idea退出debug模式_一文搞懂如何在Intellij IDEA中使用Debug,超级详细
  12. Sublime Text 一键删除空白行的方法
  13. 使用无线网桥安装电梯监控图解
  14. Samba共享服务详解
  15. 2018最新微信小程序经典案例开发视频教程合集
  16. APISpace 迎国庆
  17. 贝尔数C语言,bzoj 3501 PA2008 Cliquers Strike Back——贝尔数
  18. PaddleNLP实战——LIC2021事件抽取任务基线(附代码)
  19. 关于定量控制污水流量计的安装、维护与定期比对及计量
  20. 基于docker安装的rabbitmq如何安装rabbitmq_delayed_message_exchange-3.8.0.ez插件

热门文章

  1. java项目-第97期基于springboot的企业ERP物流管理系统-springboot毕业设计
  2. mcbbs mysql数据库_[娱乐|RPG]WEpicInventory —— 史诗背包!无限拓展!高度自由!MYSQL支持![1.12.2]...
  3. 不得不知的 “2021 阿里巴巴春招(实习生)” 那些事儿
  4. nvidia-smi topo -m
  5. 运动学拟合(kfm)
  6. Scanner、Random和 ArrayList集合的基本使用规则
  7. python 阿里云短信服务接入流程
  8. linux根目录空间不足,追加空间到根目录下
  9. 河州花儿探析――河州花儿#8226;爱情#8226;女性美
  10. 矿业生死局:疫情或提前透支减半行情,2020将迎来行业拐点