实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

栏目: IT技术 · 发布时间: 4年前

内容简介:论文提出新的特征级超分辨方法用于提升检测网络的小物体检测性能,该方法适用于带ROI池化的目标检测算法。在VOC和COCO上的小物体检测最大有5~6%mAP提升,在Tsinghua-Tencent 100K上的小物体检测则最大有约9点F1提升**论文: Better to Follow, Follow to Be Better: Towards Precise Supervision of FeatureSuper-Resolution for Small Object Detection**

论文提出新的特征级超分辨方法用于提升检测网络的小物体检测性能,该方法适用于带ROI池化的目标检测算法。在VOC和COCO上的小物体检测最大有5~6%mAP提升,在Tsinghua-Tencent 100K上的小物体检测则最大有约9点F1提升

**论文: Better to Follow, Follow to Be Better: Towards Precise Supervision of Feature

Super-Resolution for Small Object Detection**

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

Introduction

目前检测算法的研究取得了很大的进步,但仍然面临小物体、遮挡物体和不完整物体的检测问题。论文专注于接近proposal-based检测算法的小物体检测问题,认为小物体检测问题的主要原因在于小物体在特征图上的信息太少

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

特征超分辨(feature-level super-resolution)是一个很不错的增强小物体特征信息的方法,该方法的关键在于构造有效的高分辨率特征直接指导generator的训练。有效的获取方法是通过缩放原图然后进行特征提取,取对应的区域的特征作为训练目标,这样能够有效地保留目标的上下文信息。论文发现高低分辨率特征的相对感受域并不是跟缩放比例一致的,而保持相对感受域大小对超分辨训练很重要,如图1所示,即要根据相对感受域变换对目标区域进行缩放

论文的主要贡献如下:

  • 考察现有的用于小物体检测的特征超分辨方法,总结其作用的关键点主要有两点:(i) 使用高分辨率的目标特征作为监督信号 (ii) 输入特征和目标特征在原图上的相对感受域大小需要匹配
  • 提出新的特征超分辨率方法,可以用于任何带ROI特征池化的proposal-based检测器。目标生成器与主干网络共享权重,使用空洞卷积,另外好提供了迭代式(iterative)的方法来产生超分辨特征
  • 从实验来看,该方法不仅有效增强小物体检测,对中大物体检测也有促进,在Tsinghua-Tencent 100K数据集达到SOTA

Mismatch of Relative Receptive Fields

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

为了说明感受域不对等的问题,论文举了图2的例子,其中需要注意的是$R_w$为特征图上一个点对应的感受域,即连续提取特征所涉及的区域,非stride

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

对于图2中的例子,ROI对应的特征的绝对感受域大小(ARF)如公式1

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

相对感受域大小(RRF)如公式2,即公式1除以原图大小

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

假设对输入进行0.5倍下采样,则原图和缩放后的图片的相对感受域比例如公式3,$c=R_W /D$是个常数。可以看到,若$w$增大,DRRF会趋向1,若$w$减小,DRRF会趋向2,即小物体在高低分辨率的图片中的感受域可以有接近两倍的差异。对于ResNet-50的Faster R-CNN,$R_W=291$,$D=16$,则$DRRF_{1/2}(4,1600)\approx 1.8$

Our Approach

论文的方法主要通过超分辨增强小目标的特征,基于两个关键点:(i) 超分辨生成器的直接监督训练 (ii) 使用空洞卷积来匹配高低分辨率特征图的感受域

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

超分辨模块主要为GAN模型,在基础的检测模型上添加了四个组件:

  • SR feature generator,作为GAN模块的生成器,生成高分辨率特征
  • SR feature discriminator,作为GAN模块的辨别器,辨别来自generator和SR target extractor的特征图
  • SR target extractor,提供准确率超分辨特征目标指导GAN训练
  • small predictor,替代原模型的预测器(large predictor),预测来自SR feature extractor的小目标特征图,小目标定义的阈值为Tsinghua-Tencent($32\times 32$)、VOC/COCO($96\times 96$)

Super-resolution Target Extractor

定义原图为$I^{1.0}$,下采样0.5倍的图为$I^{0.5}$,$F_i^{1.0}$为原图第$i$个ROI池化后的特征,上一节说到不能直接用$F_i^{1.0}$作为$F_i^{0.5}$的超分辨目标,需要获取更合适的高分辨特征目标$T_i^{1.0}$,与$F_i^{0.5}$有类似的感受域。因此,提出额外的CNN特征提取器(super-resolution target extractor)来生成$T_i^{1.0}$,结构如图3,为了生成一样的特征,与CNN主干共享权重

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

常规CNN中核大于1的卷积和池化操作都会增大感受域,SR target extractor需要匹配感受域的增大比例,调整每层的RRF。对于无参数的池化操作,可以直接增大核的大小,而对于卷积操作,增大卷积核则会导致权重跟原CNN不一致。因此,论文将对应主干的卷积核大于1的卷积改为扩张率为2空洞卷积,在保持权重一致的情况下增大感受域。另外对于步长不为1的卷积操作,直接使用仿照使用空洞卷积会导致像素被跳过,如图4b。因此,论文在空洞卷积后面加入池化层来解决匹配主干上步长不为1的卷积操作,如图4c

总而言之,SR target extractor包含空洞卷积操作和池化操作来匹配缩小图片上扩大的感受域,$T_i^{1.0}$比$F_i^{1.0}$更适合作为高分辨率目标,而且$T_i^{1.0}$包含更多对小物体检查有用的上下文信息

Super-resolution Feature Generator

特征超分辨模块为生成对抗网络(GAN),目的是将小目标的池化特征$F_i^{1.0}$转换为超分辨处理的特征$S_i^{1.0}$。为了获取高低像素特征对进行训练,将原图下采样0.5倍,获取第$i$层特征$F_i^{0.5}$和SR target extractor生成的目标特征$T_i^{1.0}$

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

由于$F_i^{0.5}$只包含了小物体的粗粒度且低频(理解为变化小)的信息,因此需要结合前面层(sub layer)的特征$F_{sub,i}^{0.5}$。如图5所示,超分辨特征生成器通过迭代式地精调特征$F_i^{0.5}$,最终生成与特征$T_i^{1.0}$相似的特征$S_i^{0.5}$

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

为了让生成器符合预期,增加公式2的特征级损失函数

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

对于超分辨特征辨别器,使用多层感知机(3层),目标是能够辨别$T_i^{1.0}$和$S_i^{0.5}$,而生成器是将$F_{i}^{1.0}$生成为难以与$T_i^{1.0}$区分开的$S_i^{0.5}$,两者结合起来最小化公式5和6。只有小目标才会过GAN模块,不同数据集不同的阈值,具体阈值见本章开头。为了确保生成的特征对最终的结果有用,在经过GAN模块后,将超分辨特征输入到small predictor,计算分类损失$\mathcal{L}_{cls}$和定位损失$\mathcal{L}_{loc}$对GAN进行fine-tuning

Training

首先训练基础检测网络,包含主干特征提取、RPN和large predictor。然后冻结主干特征提取和RPN进行GAN模块的训练,生成器通过加权$\mathcal{L}_{gen}$、$\mathcal{L}_{cont}$、$\mathcal{L}_{cls}$和$\mathcal{L}_{loc}$进行训练,而辨别器则使用$\mathcal{L}_{dis}$。同时,small predictor使用特征$S_i^{1.0}$进行训练。SR目标特征提取使用主干特征提取进行初始化,而small predictor使用large predictor进行初始化。当生成器和辨别器收敛后,冻结其它模块进行small predictor和large predictor的fine-tune,继续训练能够保证预测器能够适应其输入的尺寸

Inference

推理时,若ROI是大目标,则使用large predictor进行预测,若ROI是小目标,则使用SR特征生成器对其进行特征处理,然后在使用small predictoe进行预测

Experiments

Results on Tsinghua-Tencent 100K

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

这是个交通标志的数据集,若IoU大于0.5则认为正确,在不同的主干网络上添加论文提出的方法进行对比,结果如表1

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

与SOTA小目标检测算法进行对比

Results on PASCAL VOC and MS COCO

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

Comparison of Super-resolution Methods

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

对比不同超分辨方法的结果,表4的分别为不进行超分辨、去掉$\mathcal{L}_{cont}$、使用主干特征提取生成目标特征以及论文的完整方案

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

Qualitative Results

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019

Conclusion

论文提出新的特征级超分辨方法用于提升检测网络的小物体检测性能,该方法适用于带ROI池化的目标检测算法。在VOC和COCO上的小物体检测最大有5~6%mAP提升,在Tsinghua-Tencent 100K上的小物体检测则最大有约9点F1提升

如果本文对你有帮助,麻烦点个赞或在看呗~

实用,小物体检测的有监督特征级超分辨方法 | ICCV 2019


以上就是本文的全部内容,希望本文的内容对大家的学习或者工作能带来一定的帮助,也希望大家多多支持 码农网

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

写给大忙人看的Java SE 8

写给大忙人看的Java SE 8

【美】Cay S. Horstmann(凯.S.霍斯曼) 编 / 张若飞 / 电子工业出版社 / 2014-11 / 59.00元

《写给大忙人看的Java SE 8》向Java开发人员言简意赅地介绍了Java 8 的许多新特性(以及Java 7 中许多未被关注的特性),《写给大忙人看的Java SE 8》延续了《快学Scala》“不废话”的风格。 《写给大忙人看的Java SE 8》共分为9章。第1章讲述了lambda表达式的全部语法;第2章给出了流的完整概述;第3章给出了使用lambda表达式设计库的有效技巧;第4章......一起来看看 《写给大忙人看的Java SE 8》 这本书的介绍吧!

HTML 压缩/解压工具
HTML 压缩/解压工具

在线压缩/解压 HTML 代码

RGB CMYK 转换工具
RGB CMYK 转换工具

RGB CMYK 互转工具

HEX CMYK 转换工具
HEX CMYK 转换工具

HEX CMYK 互转工具