深度学习和目标检测系列教程 7-凯发k8官方网
@author:runsen
faster r-cnn
由于fast r-cnn 过程中仍然存在一个瓶颈,即roi projection。众所周知,检测对象的第一步是在对象周围生成一组潜在的边界框。在 fast r-cnn 中,roi projection是使用selective search创建的,发现一个非常缓慢的过程是整个对象检测过程的瓶颈。
选择性搜索是一个缓慢且耗时的过程,会影响网络的性能。因此,任少清等人提出了一种对象检测算法,该算法消除了选择性搜索算法,让网络学习了区域提议区域提议。
faster r-cnn与 fast r-cnn 类似,图像作为输入,提供给提供卷积特征图的卷积网络。不是在特征图上使用选择性搜索算法来识别区域提议,而是使用单独的网络来预测roi projection。然后使用 roi 池化层对预测的区域建议进行重新整理,然后使用 roi 池化层对建议区域内的图像进行分类并预测边界框的偏移值。
从上图可以看出,faster r-cnn 比fast r-cnn要快得多。因此,它是目前先进的目标检测算法。
为什么不将这些相同的 cnn 结果重用于区域提议,而不是运行单独的选择性搜索算法
让我们看一个faster r-cnn的思想,首先,让我们了解一下具有不同纵横比和尺度的重叠roi的概念。
从图像中,我们看到很多物体相互重叠。在电视机里看到一辆汽车、一辆自行车、一个拿着电视机的人和一只狗。选择性搜索可以解决这个问题,但我们最终会得到大量的 roi。我们需要想一个能有效解决这个问题的想法。
如果我们在不同的对象周围绘制边界框,它看起来是这样的。
实现这一点有点复杂,但 anchor boxes 的想法使它变得简单。让我们了解这些锚框的含义。
通常,对象可以放入方形盒子中,也可以放入长方形和宽,或长方形和高。从广义上讲,它们可以说是大的、小的或中等的。因此,通过实验发现,可以使用 3 种不同比例和 3 种不同纵横比的框检测图像中的任何对象。
在feature map上使用不同大小的滑动窗口(如上图的9个所示)。这个概念被称为特征金字塔。这涉及在特征图顶部使用 9 种不同大小的滑动窗口。
rpn
- 区域提议网络 (rpn) 从将输入图像馈送到卷积神经网络开始。首先调整输入图像的大小,使其最短边为 600 像素,长边不超过 1000 像素。
- 网络的输出特征(由h x w表示)通常比输入图像小得多,这取决于网络模型的步幅。在faster r-cnn论文中使用的两个网络模型(vgg、zf-net),网络步长都是 16。这意味着网络输出特征中的两个连续像素对应于输入图像中相距 16 个像素的两个点。
下图显示了 在输入图像上,3 种不同纵横比和 3 种不同尺寸的 9 个可能的锚点放置,使用的锚点具有 128²、256²、512²的 3 个盒子区域比例和 1:1、1:2 和 2:1 的 3 个纵横比。
faster r-cnn架构
如下为整体框架结构,结合上述过程来看,主要是三部分,extrator进行特征提取、rpn 生成候选框、roihead对候选框进行分类并调整目标预测框的位置与大小。
faster r-cnn使用了预训练的vgg16作为backbone进行特征提取,实现方法是加载预训练模型,抽取并分离前面的卷积层和后面的全连接层,固定卷积层中部分层的权重,用作特征提取,而全连接层则给 roihead 用作分类和回归。
faster r-cnn 联合训练有 4 个损失:
- rpn分类(iou 大于 0.5 object foreground/ iou 介于 0.1 和 0.5background)
- rpn 回归(anchor → roi)
- fast rcnn 分类(对象类)。
- fast rcnn 回归(roi → bounding box)
总结,r-cnn 和 fast r-cnn 都使用基于 cpu 的区域提议算法,例如选择性搜索算法,每张图像大约需要 2 秒并在 cpu 计算上运行。faster r-cnn论文通过使用另一个卷积网络(rpn)来生成区域提议来解决这个问题。这不仅将每个图像的区域提议时间从 2 秒降低到 10 毫秒,而且还允许区域提议阶段与以下检测阶段共享层,从而导致特征表示的整体改进。
与50位技术专家面对面20年技术见证,附赠技术全景图总结
以上是凯发k8官方网为你收集整理的深度学习和目标检测系列教程 7-300:先进的目标检测faster r-cnn架构的全部内容,希望文章能够帮你解决所遇到的问题。
- 上一篇:
- 下一篇: