如何在facebook / deepmask上训练我自己的数据集?

时间:2016-11-16 20:54:44

标签: facebook image-processing deep-learning torch

我目前正在尝试将测试图像添加到脸谱/深层面具,以便我可以训练这些图像。目前,deepmask程序使用CoCo图像和注释,但我想知道是否可以添加我自己的图像。唯一的原因是我只想训练一种类型的图像,在这种情况下是葡萄。我查看了注释文件,但它充满了图像的规格。我问这个问题的唯一原因是因为我的特定问题没有在线教程/答案。

以下是深层掩码的github:https://github.com/facebookresearch/deepmask

此外,要下载CoCo图像的注释文件,请在终端中使用此命令:wget http://msvocds.blob.core.windows.net/annotations-1-0-3/instances_train-val2014.zip

感谢大家的时间!

斯蒂芬

1 个答案:

答案 0 :(得分:0)

我的经验是一样的:你应该把你的面具转换成可可风格的格式。它是json,可以在这里找到:coco dataset。 我唯一的评论是coco使用" costum运行长度编码(RLE)",这是无处解释的。因此,很容易改为使用掩模的多边形编码。 因此,请执行:

  1. 获取面具
  2. 将其编码为多边形
  3. 将其写入json文件
  4. 希望这会有所帮助: - )