深度学习之边框回归(Bounding Box Regression)

最新推荐文章于 2024-10-05 18:40:59 发布

奔跑的大西吉

最新推荐文章于 2024-10-05 18:40:59 发布

阅读量3.7k

点赞数 8

分类专栏：深度学习文章标签：深度学习神经网络人工智能

原文链接：127.0.0.1

版权

深度学习专栏收录该内容

37 篇文章

订阅专栏

从rcnn， fast rcnn, faster rcnn, yolo, r-fcn, ssd，到cvpr的yolo9000。这些paper中损失函数都包含了边框回归，除了rcnn详细介绍了，其他的paper都是一笔带过，或者直接引用rcnn就把损失函数写出来了。前三条网上解释比较多，后面的两条我看了很多paper，才得出这些结论。

为什么要边框回归？
什么是边框回归？
边框回归怎么做的？
边框回归为什么宽高，坐标会设计这种形式？
为什么边框回归只能微调，在离Ground Truth近的时候才能生效？

为什么要边框回归？

这里引用王斌师兄的理解，如下图所示：
在这里插入图片描述

对于上图，绿色的框表示 Ground Truth, 红色的框为 Selective Search 提取的 Region Proposal。那么即便红色的框被分类器识别为飞机，但是由于红色的框定位不准(IoU<0.5)，那么这张图相当于没有正确的检测出飞机。如果我们能对红色的框进行微调，使得经过微调后的窗口跟 Ground Truth 更接近，这样岂不是定位会更准确。确实，Bounding-box regression 就是用来微调这个窗口的。

边框回归是什么？

对于窗口一般使用四维向量 $(x, y, w, h)$ 来表示，分别表示窗口的中心点坐标和宽高。对于下图, 红色的框 P 代表原始的 Proposal, 绿色的框 G 代表目标的 Ground Truth，我们的目标是寻找一种关系使得输入原始的窗口 P 经过映射得到一个跟真实窗口 G 更接近的回归窗口 $\hat{G}$ 。

边框回归的目的既是：给定 $P_{x},P_{y},P_{w},P_{h})$ 寻找一种映射 $f$ ，使得 $f(P_{x},P_{y},P_{w},P_{h}) = (\hat{G_x},\hat{G_y},\hat{G_w},\hat{G_h})$ 并且 $(\hat{G_x},\hat{G_y},\hat{G_w},\hat{G_h}) \approx (G_{x},G_{y},G_{w},G_{h})$

边框回归怎么做的？

RCNN论文：Region-based Convolution Networks for Accurate Object detection and Segmentation

作者在完成了的生成候选区域——CNN提取特征——SVM进行分类以后，为了进一步的提高定位效果，在文章的附录C中介绍了 Bounding-box Regression 的处理。Bounding-box Regression 训练的过程中，输入数据为N个训练对 ${(P^{i},G^{i})},i=1,2,...,N$ ，其中 $p^i=(p^i_x,p^i_y,p^i_w,p^i_h)$ 为proposal的位置，前两个坐标表示proposal的中心坐标，后面两个坐标分别表示proposal的width和height，而 $G^i=(G_x,G_y,G_w,G_h)$ 表示groundtruth的位置， regression的目标就是学会一种映射将P转换为G。

那么经过何种变换才能从上图中的窗口 P 变为窗口 $\hat{G}$ 呢？比较简单的思路就是: 平移+尺度放缩

作者设计了四种坐标映射方法，其中前两个表示对 proposal 中心坐标的尺度不变的平移变换，后面两个则是对 proposal 的 width 和 height 的对数空间的变换，

先做平移 $(\Delta x,\Delta y)$ ， $\Delta x = P_{w}d_{x}(P),\Delta y = P_{h}d_{y}(P)$ 这是R-CNN论文的：( $d_{ \ast }(P) = w_{ \ast }^{T}\Phi _{5}(P)=t_{ \ast }$ )

$\hat{G}_{x} = P_{w}d_{x}(P) + P_{x},\text{(1)}$

$\hat{G}_{y} = P_{h}d_{y}(P) + P_{y},\text{(2)}$

然后再做尺度缩放 $S_{w},S_{h})$ , $S_{w} = exp(d_{w}(P)),S_{h} = exp(d_{h}(P))$ , 对应论文中：

$\hat{G}_{w} = P_{w}exp(d_{w}(P)),\text{(3)}$

$\hat{G}_{h} = P_{h}exp(d_{h}(P)),\text{(4)}$

观察(1)-(4)我们发现，边框回归学习就是 $d_{x}(P),d_{y}(P),d_{w}(P),d_{h}(P)$ 这四个变换。下一步就是设计算法那得到这四个映射。

线性回归就是给定输入的特征向量 X, 学习一组参数 W, 使得经过线性回归后的值跟真实值 Y(Ground Truth)非常接近. 即 $\approx WX$ 。那么 Bounding-box 中我们的输入以及输出分别是什么呢？

Input:

$RegionProposal\rightarrow P = (P_{x},P_{y},P_{w},P_{h})$ ，这个是什么？输入就是这四个数值吗？不是，其实真正的输入是这个窗口对应的 CNN 特征，也就是 R-CNN 中的 Pool5 feature（特征向量）。 (注：训练阶段输入还包括 Ground Truth，也就是下边提到的 $t_{ \ast } = (t_{x},t_{y},t_{w},t_{h})$

Output:

outpue 为：需要进行的平移变换和尺度缩放 $d_{x}(P),d_{y}(P),d_{w}(P),d_{h}(P)$ ，或者说是 $\Delta x,\Delta y,S_{w},S_{h}$ 。我们的最终输出不应该是 Ground Truth 吗？是的，但是有了这四个变换我们就可以直接得到 Ground Truth。

这里有个问题需要注意，根据(1)~(4)我们可以知道， P 经过 $d_{x}(P),d_{y}(P),d_{w}(P),d_{h}(P)$ 得到的并不是真实值 G，而是预测值 $\hat{G}$ 。

在训练时这四个值 $\Delta x,\Delta y,S_{w},S_{h}$ 的真实值应该是经过 Ground Truth 和 Proposal 计算得到的真正需要的平移量 $t_{x},t_{y})$ 和尺度缩放 $t_{w},t_{h})$ 。这也就是 R-CNN 论文中的(6)~(9)：

$t_{x} = (G_{x}−P_{x})/ P_{w},(6)$

$t_{y} = (G_{y}−P_{y})/ P_{h},(7)$

$t_{w} = \log ⁡(G_{w}/ P_{w}),(8)$

$t_{h} = \log ⁡(G_{h}/ P_{h}),(9)$

目标函数

目标函数可以表示为 $d_{ \ast }(P) = w_{ \ast }^{T}\Phi _{5}(P)$ ， $\Phi _{5}(P)$ 是输入 Proposal 的特征向量， $w_{ \ast }$ 是要学习的参数（*表示 x,y,w,h，也就是每一个变换对应一个目标函数） , $d_{ \ast }(P)$ 是得到的预测值。我们要让预测值跟真实值 $t_{ \ast } = (t_{x},t_{y},t_{w},t_{h})$ 差距最小，得到损失函数为：

$\sum \limits_{i}^{N}(t_{ \ast }^{i}−\hat{w}_{ \ast }^{T}\phi _{5}(P^{i}))^2.$

函数优化目标为：

$W_{ \ast } = argmin_{w_{ \ast }} \sum \limits_{i}^{N}(t_{ \ast }^{i}−\hat{w}_{ \ast }^{T}\phi _{5}(P^{i}))^2 + \lambda ||\hat{w}_{ \ast }||^2.$

利用梯度下降法或者最小二乘法就可以得到 $w_{ \ast }$ 。

最终在进行实验时，lambda = 1000, 同时作者发现同一对中P和G相距过远时通过上面的变换是不能完成的，而相距过远实际上也基本不会是同一物体，因此作者在进行实验室，对于 pair(P,G) 的选择是选择离P较近的G进行配对，这里表示较近的方法是需要P和一个G的最大的IoU要大于0.6,否则则抛弃该P。

为什么宽高尺度会设计这种形式？

重点解释一下为什么设计的 $t_{x},t_{y}$ 为什么除以宽高，为什么 $t_{w},t_{h}$ 会有log形式！！！

首先CNN具有尺度不变性，以下图为例：
在这里插入图片描述

x,y 坐标除以宽高

上图的两个人具有不同的尺度，因为他都是人，我们得到的特征相同。假设我们得到的特征为 $\phi _{1},\phi _{2}$ ，那么一个完好的特征应该具备 $\phi _{1} = \phi$ 。ok，如果我们直接学习坐标差值，以x坐标为例， $x_{i},p_{i}$ 分别代表第i个框的x坐标，学习到的映射为 $f$ , $f(\phi _{1}) = x_{1}−p_{1}$ ，同理 $f(\phi _{2}) = x_{2}−p_{2}$ 。从上图显而易见， $x_{1}−p_{1} \neq x_{2}−p_{1}$ 。也就是说同一个x对应多个y，这明显不满足函数的定义。边框回归学习的是回归函数，然而你的目标却不满足函数定义，当然学习不到什么。

宽高坐标Log形式

我们想要得到一个放缩的尺度，也就是说这里限制尺度必须大于0。我们学习的 $t_{w},t_{h}$ 怎么保证满足大于0呢？直观的想法就是EXP函数，如公式(3), (4)所示，那么反过来推导就是Log函数的来源了。

为什么IoU较大，认为是线性变换？

当输入的 Proposal 与 Ground Truth 相差较小时(RCNN 设置的是 IoU>0.6)，可以认为这种变换是一种线性变换，那么我们就可以用线性回归来建模对窗口进行微调，否则会导致训练的回归模型不 work（当 Proposal跟 GT 离得较远，就是复杂的非线性问题了，此时用线性回归建模显然不合理）。这里我来解释：

Log函数明显不满足线性函数，但是为什么当Proposal 和Ground Truth相差较小的时候，就可以认为是一种线性变换呢？大家还记得这个公式不？参看高数1。

$lim_{x = 0}log(1 + x) = x$