ROI Pooling &ROI Align[转发]

2020-02-10  本文已影响0人  数据智能谷

ROI Align

Mask RCNN中还有一个很重要的改进,就是ROIAlign。Faster R-CNN存在的问题是:特征图与原始图像是不对准的(mis-alignment),所以会影响检测精度。而Mask R-CNN提出了RoIAlign的方法来取代ROI pooling,RoIAlign可以保留大致的空间位置。

为了讲清楚ROI Align,这里先插入两个知识,双线性插值和ROI pooling。

1.双线性插值

在讲双线性插值之前,还得看最简单的线性插值。

线性插值

已知数据(x0, y0) 与(x1, y1) ,要计算区间内某一位置 x 在直线上的 y 值,如下图所示。


计算方法很简单,通过斜率相等就可以构建y和x之间的关系,如下:

双线性插值

双线性插值本质上就是在两个方向上做线性插值。

image

如图,假设我们想得到P点的插值,我们可以先在x方向上,对Q11和Q21 之间做线性插值得到R1 ,R2 同理可得。然后在y方向上对R1和R2 进行线性插值就可以得到最终的P。其实知道这个就已经理解了双线性插值的意思了,如果用公式表达则如下(注意 f 前面的系数看成权重就很好理解了)。

首先在 x 方向进行线性插值,得到

然后在 y 方向进行线性插值,得到

这样就得到所要的结果

2.ROIpooling

ROI pooling就不多解释了,直接通过一个例子来形象理解。假设现在我们有一个8x8大小的feature map,我们要在这个feature map上得到ROI,并且进行ROI pooling到2x2大小的输出。

假设ROI的bounding box为

。如图:

image

将它划分为2x2的网格,因为ROI的长宽除以2是不能整除的,所以会出现每个格子大小不一样的情况。

image

进行max pooling的最终2x2的输出为:

3. ROI Align

在Faster RCNN中,有两次整数化的过程:

  1. region proposal的xywh通常是小数,但是为了方便操作会把它整数化。
  2. 将整数化后的边界区域平均分割成 k x k 个单元,对每一个单元的边界进行整数化。

两次整数化的过程如下图所示:

image

事实上,经过上述两次整数化,此时的候选框已经和最开始回归出来的位置有一定的偏差,这个偏差会影响检测或者分割的准确度。在论文里,作者把它总结为“不匹配问题”(misalignment)。

为了解决这个问题,ROI Align方法取消整数化操作,保留了小数,使用以上介绍的双线性插值的方法获得坐标为浮点数的像素点上的图像数值。但在实际操作中,ROI Align并不是简单地补充出候选区域边界上的坐标点,然后进行池化,而是重新进行设计。

下面通过一个例子来讲解ROI Align操作。如下图所示,虚线部分表示feature map,实线表示ROI,这里将ROI切分成2x2的单元格。如果采样点数是4,那我们首先将每个单元格子均分成四个小方格(如红色线所示),每个小方格中心就是采样点。这些采样点的坐标通常是浮点数,所以需要对采样点像素进行双线性插值(如四个箭头所示),就可以得到该像素点的值了。然后对每个单元格内的四个采样点进行maxpooling,就可以得到最终的ROIAlign的结果。

image

需要说明的是,在相关实验中,作者发现将采样点设为4会获得最佳性能,甚至直接设为1在性能上也相差无几。事实上,ROI Align 在遍历取样点的数量上没有ROIPooling那么多,但却可以获得更好的性能,这主要归功于解决了misalignment的问题。

上一篇下一篇

猜你喜欢

热点阅读