【计算机视觉——RCNN目标检测系列】二、边界框回归(Bounding-Box Regression)

微信公众号:AI那点小事 作者:戴璞微 知乎专栏:AI那点小事 CSDN博客:https://daipuweiai.blog.csdn.net/ 原文链接:https://zhuanlan.zhihu.com/p/76603583 https://daipuweiai.blog.csdn.net/article/details/96570669

前言

在上一篇博文:【计算机视觉——RCNN目标检测系列】一、选择性搜索详解中我们重点介绍了RCNN和Fast RCNN中一个重要的模块——选择性搜索算法,该算法主要用于获取图像中大量的候选目标框。为了之后更加顺利理解RCNN模型,在这篇博文中我们将主要介绍RCNN及其改进版本——Fast RCNN和Faster RCNN中一个重要模块——边界框回归(Bounding-Box Regression)


一、边界框回归简介

相比传统的图像分类,目标检测不仅要实现目标的分类,而且还要解决目标的定位问题,即获取目标在原始图像中的位置信息。在不管是最初版本的RCNN,还之后的改进版本——Fast RCNN和Faster RCNN都需要利用边界框回归来预测物体的目标检测框。因此掌握边界框回归(Bounding-Box Regression)是极其重要的,这是熟练使用RCNN系列模型的关键一步,也是代码实现中比较重要的一个模块。接下来,我们对边界框回归(Bounding-Box Regression)进行详细介绍。

首先我们对边界框回归的输入数据集进行说明。输入到边界框回归的数据集为

,其中

代表第

个带预测的候选目标检测框即region proposal。

个真实目标检测框即ground-truth。在RCNN和Fast RCNN中,

是利用选择性搜索算法进行获取;Faster RCNN中,

是利用RPN(Region Proposal Network,区域生成网络)获取。在

中,

代表候选目标框的中心点在原始图像中的

坐标,

代表候选目标框的中心点在原始图像中的

坐标,

代表候选目标框的长度,

代表候选目标框的宽度。

的四维特征的含义与

的一样。

那么边界框回归所要做的就是利用某种映射关系,使得候选目标框(region proposal)的映射目标框无限接近于真实目标框(ground-truth)。将上述原理利用数学符号表示如下:在给定一组候选目标框

,寻找到一个映射

,使得

。边界框回归过程图像表示如下图所示。在图1中红色框代表候选目标框,绿色框代表真实目标框,蓝色框代表边界框回归算法预测目标框,红色圆圈代表选候选目标框的中心点,绿色圆圈代表选真实目标框的中心点,蓝色圆圈代表选边界框回归算法预测目标框的中心点。


二、边界框回归细节

RCNN论文里指出,边界框回归是利用平移变换和尺度变换来实现映射 。平移变换的计算公式如下:

尺度变换的计算公式如下:

其中

(

代表

),接下来要做的就是求解这4个变换。在边界框回归中,我们利用了线性回归在RCNN论文代表这AlexNet第5个池化层得到的特征即将送入全连接层的输入特征的线型函数。在这里,我们将特征记作

,那么

。因此,我们可以利用最小二乘法或者梯度下降算法进行求解 ,RCNN论文中给出了 的求解表达式:

其中:

可以看出,上述模型就是一个Ridge回归模型。在RCNN中,边界框回归要设计4个不同的Ridge回归模型分别求


三、相关问题

3.1 为什么使用相对坐标差?

在式(4)中 ,那么为什么要将真实框的中心坐标与候选框的中心坐标的差值分别除以宽高呢?首先我们假设两张尺寸不同,但内容相同的图像,图像如下图所示。

那么我们假设经过CNN提取得到的特征分别为

。同时,我们假设

为第

个真实目标框的

坐标,

为第个候选目标框

坐标,边界框回归的映射关系 。那么我们可以得出:

由于CNN具有尺度不变性,因此

。那么理论上

。但是观察上图就可明显得出

,显然由于尺寸的变化,候选目标框和真实目标框坐标之间的偏移量也随着尺寸而成比例缩放,即这个比例值是恒定不变的。

因此,我们必须对

坐标的偏移量除以候选目标框的宽,

坐标的偏移量除以候选目标框的高。只有这样才能得到候选目标框与真实目标框之间坐标偏移量值的相对值。同时使用相对偏移量的好处可以自由选择输入图像的尺寸,使得模型灵活多变。也就说,对坐标偏移量除以宽高就是在做尺度归一化,即尺寸较大的目标框的坐标偏移量较大,尺寸较小的目标框的坐标偏移量较小。

3.2 为什么宽高比要取对数?

同时在式(4)中

,类比问题3.1,我们不禁要问为什么不直接使用宽高的比值作为目标进行学习,非得“多此一举”取对数?结合式(2)可以看出:

。也就说式(5)的后两个公式与式(2)可以视为等价。

3.3 为什么IoU较大时边界框回归可视为线性变换?

在这里我们需要回顾下在高等数学中有关等价无穷小的结论:

也就是说当

趋向于0时,我们可有

,即可将

近似看成线型变换。接下来我们将式(4)的后两个公式进行重写:

也就是说,当

趋向于0时,即

时,上式(6)可以近似将对数变换看成线性变换

时候选目标框和真实目标框非常接近,即IoU值较大。按照RCNN论文的说法,IoU大于0.6时,边界框回归可视为线型变换。

至此,边界框回归算法的讲解全部结束。在进行RCNN论文的相关介绍之前,我们必须对于RCNN论文中最后点小细节性概念进行讲解,例如交并比(IoU)和非极大抑制(NMS)。