论文阅读笔记五十五：DenseBox:UnifyingLandmarkLocalizationwithEndtoEndObjectDetection（CVPR2015）...

论文原址&＃xff1a;https://arxiv.org/abs/1509.04874

github:https://github.com/CaptainEven/DenseBox

摘要

       本文先提出了一个问题:如何将全卷积网络应用到目标检测中去&＃xff1f;本文提出DenseBox&＃xff0c;一个集成的FCN 框架可以直接在图像的位置上预测出目标物的边框及类别。本文两方面贡献:(1)FCN可以用于检测不同的目标(2)在多任务学习过程中结合landmark定位可以进一步提高对目标的检测的准确性。

介绍

       本文只关注一个问题&＃xff0c;即如何将FCN应用到目标检测当中去&＃xff1f;本文提出DenseBox不需要生成proposal&＃xff0c;在训练过程中也可以达到最优。与现存的基于滑动窗的FCN的检测框架相类似&＃xff0c;DenseBox更偏重于小目标及较为模糊目标的检测。本文通过对DenseBox训练&＃xff0c;使用hard negative mining技术来提升检测性能。为了进一步提高&＃xff0c;后面多任务联合学习结合了landmark定位来进一步提升性能。

DenseBox for Detection

       DenseBox的整体流程如下图所示&＃xff0c;单一的卷积网路同时输出不同的预测框及类别分数。DenseBox中的所有目标检测模块都为全卷积网络结构&＃xff0c;除了NMS处理部分&＃xff0c;因此&＃xff0c;proposal的生成是没有必要的。测试时&＃xff0c;输入图片大小为(mxnx3)&＃xff0c;输出为&＃xff08;m/4 x n/4 *5&＃xff09;&＃xff0c;将目标边界框左上角及右下角的点的坐标域定义为及&＃xff0c;则第i个位置的像素的输出feature map用一个5维向量描述其中s^代表为目标的分数&＃xff0c;剩余几个变量代表输出边界框于真是边框之间的距离。最后对带有边框及类别分数的框进行NMS处理。

Ground Truth的生成

       没有必要将整张图片送人网络中训练&＃xff0c;这样会造成不必要的计算资源的消耗。一种有效的方式是裁剪较大块包含目标物及足够背景信息的patches进行训练。本文进行单一尺度训练&＃xff0c;多尺度测试。

       该网络的训练方式形似分割的类型&＃xff0c;训练时&＃xff0c;图片被裁剪并调整到240x240的大小&＃xff0c;输出的ground truth 是一个经过4倍下采样后大小为60x60x5的map&＃xff0c;在ground truth map的第一个通道中&＃xff0c;positive labeld区域设置一个半径为rc的圆&＃xff0c;该圆位于边界框的中心。半径rc与边界框的大小是成比例的。其缩放系数&＃xff0c;针对输出坐标域中边框大小设置为0.3。如下图所示&＃xff0c;剩余 的四个通道代表最相近(与ground truth比较)边框中左上角及右下角像素点的距离。如果patch中存在多个目标&＃xff0c;将那些位于patch中心(0.8至1.25)区域的目标标记为正样本&＃xff0c;像素的第一个通道&＃xff0c;代表类别的分数&＃xff0c;在ground truth中是目标像素值为1&＃xff0c;否则为0.本文的方法中每个像素就可以产生一个边框&＃xff0c;因此&＃xff0c;可以将每个像素当作一个训练样本。

模型设计

       网络结构如下图所示&＃xff0c;基于VGG19进行的改进&＃xff0c;整个网络包含16层卷积&＃xff0c;前12层由VGG19初始化&＃xff0c;输出conv4_4后接4个1x1的卷积&＃xff0c;前两个卷积产生1-channel map用于类别分数&＃xff0c;后两个产生4-channel map用于预测相对位置。最后一个1x1的卷积担当这全连接层的作用。

多层次特征融合

       Part-Level特征关注于目标物的局部细节&＃xff0c;进而借此可以区分外形的不同&＃xff0c;然而object-level或者high-level特征通常通过一个较大的感受野来对目标进行识别。感受野的范围越大&＃xff0c;其上下文信息越丰富&＃xff0c;越有利于提高准确性能。本文将conv3_4与conv4_4进行了拼接处理。conv3_4感受野的大小为48x48&＃xff0c;与本文的训练目标尺寸大小相似&＃xff0c;而conv4_4的感受野更大&＃xff0c;为118x118&＃xff0c;可以有效的利用全局的文本及上下文信息用于检测。而conv4_4的feature map是conv3_4的一半&＃xff0c;因此首先将其进行上采样至相同的分辨率。

多任务学习

       本文使用基于ImageNet预训练的VGG19网络来初始化DenseBox&＃xff0c;实际只初始化了前12层卷积&＃xff0c;其余的卷积层使用新的卷积替换&＃xff0c;并使用xavier进行初始化。

        与Fast R-CNN相似&＃xff0c;本文也存在两个子分支输出&＃xff0c;第一个输出为目标的类别分数(输出map中的每一个像素)&＃xff0c;ground truth label为y* 属于{0,1},分类的损失定义如下&＃xff0c;使用L2损失&＃xff08;Hinge 损失及交叉熵损失&＃xff09;&＃xff0c;L2损失效果已经很好。

        第二个分支输出边界框的回归损失&＃xff0c;定义如下&＃xff0c;用于最小化目标偏移及预测偏移之间的L2损失。

平衡采样

       在学习过程中&＃xff0c;负样本的采样挑选是至关重要的部分。如果简单的将mini-batch中的所有负样本都进行处理&＃xff0c;将会使模型倾向于负样本。另外&＃xff0c;检测器遇到正负样本之间的样本时会发生塌陷。本文使用一个二值mask输出像素&＃xff0c;用于决定是否为训练样本。

忽略灰度区域

       将正负区域之间的区域定义为忽略区域&＃xff0c;该区域的样本既不是正样本也不是负样本&＃xff0c;其损失权重也应设置为0。在输出坐标空间中&＃xff0c;对于每一个非正标记的像素&＃xff0c;只要其半径范围2内存在任意一个带正标记的像素&＃xff0c;则ignore flag fign设置为1。

Hard Negative Mining

       通过寻找预测较差的样本来提高学习效率。经过negative mining后&＃xff0c;分错的负样本更容易被挑出来&＃xff0c;对于这些样本基于随机梯度下降的学习可以产生噪音小而且更为鲁棒的预测。negative mining可以通过在线提升来提高性能。在前向传递过程中&＃xff0c;按分类损失将像素按降序排列&＃xff0c;top1%的为hard-negative。在实验中&＃xff0c;将positive于negative的比例设置在1:1.在negative samples中&＃xff0c;一半来自于hard-negative&＃xff0c;剩余的从非hard-negative中随机采样。为了方便&＃xff0c;将被挑选的像素设置标记fsel&＃61;1.

Loss with Mask

       为每个样本,定义mask ,函数如下

       本文的整体损失为&＃xff0c;回归损失只对正样本起作用&＃xff0c;本文将目标d*除以标准目标的高(ground truth map中为50/4)进行正则化&＃xff0c;设置

细节增强

       将特定尺寸中心处包含目标中心的输入patch称为"positive patch"&＃xff0c;这些patches在正样本的周围只包含负样本。本文将输入图片中随机进行裁剪&＃xff0c;并resize到相同的大小送入网络中&＃xff0c;这类patch称为"random patches"&＃xff0c;"positive patch"与"random patch"的比例为1:1&＃xff0c;另外&＃xff0c;为了提升网络的鲁棒性&＃xff0c;对每个patch进行了jitter操作&＃xff0c;left-right flip,translation shift(25个像素)&＃xff0c;尺寸变形([0.8,1.25])

       基于mini-batch的SGD训练&＃xff0c;batchsize&＃61;10&＃xff0c;loss及梯度通过贡献的像素个数进行正则化。global learning rate开始为0.001&＃xff0c;每100K迭代&＃xff0c;学习率衰减10倍。momentum设置为0.9&＃xff0c;权重衰减为0.0005

Refine with Landmark Localization

        在DenseBox中由于是全卷积网络&＃xff0c;因此&＃xff0c;基于landmark定位可以通过简单添加一些层来进行实现。通过融合landmark heatmaps及目标score maps来对检测结果进行增强。如下图所示&＃xff0c;增加了一个分支用于landmark定位&＃xff0c;假设存在N个landmarks&＃xff0c;landmark 定位分支将会输出N个响应maps&＃xff0c;其中&＃xff0c;每个像素值代表该位置为landmark的置信分数。该任务的ground truth maps与检测的十分相似&＃xff0c;对于一个landmark 实例&＃xff0c;landmark k的第i个实例&＃xff0c;其对应的ground truth 是位于输出坐标空间中第k个响应 map上的positive 标记的区域。半径rl应当较小从而避免准确率的损失。与分类任务相似&＃xff0c;landmark 定位损失也是定义为预测值与真实值的L2损失。同样使用negative mining及ignore region。

        将分类score map及landmark localization maps作为输入的增强分支的最终输出用于增强检测结果。通过一些高层次空间模型来了解landmark confidence及边界框分数的方法来进一步提高检测性能。本文使用带ReLU的卷积来近似空间模型&＃xff0c;增加了增强分支后的整体损失函数如下所示。

Experiment

References

       [1] Y. Bengio, A. Courville, and P. Vincent. Representation learning: A review and new perspectives. Pattern Analysis and Machine Intelligence, IEEE Transactions on, 35(8):1798–1828, 2013.

       [2] G. Bertasius, J. Shi, and L. Torresani. Deepedge: A multi-scale bifurcated deep network for top-down contour detection. arXiv preprint arXiv:1412.1123, 2014.

       [3] X. Chen, K. Kundu, Y. Zhu, A. Berneshawi, H. Ma, S. Fidler, and R. Urtasun. 3d object proposals for accurate object class detection. In NIPS, 2015.

       [4] R. G. Cinbis, J. Verbeek, and C. Schmid. Segmentation driven object detection with fisher vectors. In Computer Vision (ICCV), 2013 IEEE International Conference on, pages 2968– 2975. IEEE, 2013.