训练样本制作--Annotating Object Instances with a Polygon-RNN

来源:互联网 发布:形容数据整齐的词语 编辑:程序博客网 时间:2024/06/11 03:07

Annotating Object Instances with a Polygon-RNN
CVPR2017
project page: http://www.cs.toronto.edu/polyrnn/

这里主要提出一个制作真值数据的工具。如下图所示:

这里写图片描述

这里我们使用VGG网络提取特征,使用一个 Convolutional LSTM 进行物体轮廓的提取。

这里写图片描述

我们的模型之所以是 RNN,是因为它每次只预测一个节点vertex。 每个步骤 RNN 的输入是: 图像块的CNN特征,前两个步骤预测的vertices 和第一个节点。 用前两个步骤的节点来预测下一个节点。第一个节点用于判断polygon 是否结束。 这里的CNN提取的特征倾向于 物体的边缘信息。RNN 用于跟踪这些边缘。

在RNN中我们可以随时加入人工标记信息。

这里写图片描述

这里写图片描述

0 0
原创粉丝点击