论文笔记——CVPR 2017 Annotating Object Instances with a Polygon-RNN

2017 年 6 月 23 日 极市平台

↑ 点击蓝字关注极市平台识别先机创造未来

1.简介

　文章作者基于深度学习提出一种半自动目标事例标注（semi-automatic annotation of object instances）的算法。大多数前人是将目标分割看作是像素级别的标注问题（pixel-labeling）问题，但是文章作者是将其看做是一个多边形预测的任务（polygon prediction）。两者区别如下（本人自己注释，上图为像素级分割，下图是多边形标注）：

　　当前大多数语义图像分割算法是基于深度学习的方式，但是深度学习的效果很大程度上依赖于大量的训练数据，这就造成圈内人士需要花费很大的人力物力和时间去手动标注大规模训练数据集。这篇文章的目标正是为了加快标注精度很高的真值（ground truth）。
　　那这篇文章为何称为半自动目标事例标注呢？这是因为以下两点：
　　 ①、这篇文章算法首先需要给定一个bounding box真值，然后使用一个RNN（Recurrent Neural Network），文中称为Polygon-RNN在这个目标框中画出目标一个多边形圈住的轮廓。因为相比较手动标注目标轮廓，bounding box标注只要两下鼠标点击即可，容易很多（见上方右图）。
　　 ②、算法标注轮廓过程，人为可干预从而产生更精确的标注结果。这块细节下文再仔细介绍过程。

2.polygon-RNN：

2.1介绍

我们再来好好总结一下整个过程，作者是想创建一个有效的标注工具（annotation tool），从而以多边形形式标注目标事例。当给定bounding box中的图像块（image patch），文章算法基于RNN可以预测一个封闭的多边形来圈出目标的轮廓。多边形设计方法就是先找到一个起点，然后以顺时针方式连续生成多边形的其他顶点，顺序连接所有顶点即形成这个圈出目标轮廓的多边形。

　　模型是一个RNN，每一次迭代预测一个多边形顶点。RNN每一次迭代的输入it包含以下三个方面。第一是图片的CNN特征表示；第二是前两个RNN迭代输出的顶点yt−1和yt−2，依一个特殊方向形成多边形；第三是起点，帮助RNN决定何时封闭多边形。整个网络框架如下图：

2.2CNN形成图片特征表示

文章使用一个VGG-16结构表示图片特征，首先移除全连接层和最后的max-pooling层pool5，然后通过上采样和max-pooling统一跳跃连接VGG不同层的尺寸，形成一个28*28*512串联特征。最后是一个卷积层结合ReLu处理整个串联特征从而形成最终图片特征表示28*28*128，如上图绿色部分。

2.3RNN预测顶点

RNN网络可以在迭代过程中，通过线性和非线性方程携带复杂的历史信息，正是从这点考虑作者希望通过RNN来依次预测出多边形的顶点。文中RNN使用的是一个ConvolutionalLSTM框架，详细来说，作者设计了一个核为3*3和16通道的两层ConvLSTM框架，然后在每一步迭代就输出一个顶点 yt。当给定输入图像表示 xt，一个ConvLSTM单层的隐层 ht计算如下：