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

如何在facebook/deepmask上训练自己的数据集?,facebook,image-processing,deep-learning,torch,Facebook,Image Processing,Deep Learning,Torch,我目前正在尝试将测试图像添加到facebook/deepmask,以便可以训练这些图像。目前,deepmask程序使用CoCo图像和注释,但我想知道我是否可以添加自己的图像。唯一的原因是我只想训练一种类型的图像,在本例中是葡萄。我查看了注释文件,但其中充满了图像的规格。我问这个问题的唯一原因是因为网上没有关于我具体问题的教程/答案 下面是github到deepmask的链接: 此外,要下载CoCo图像的注释文件,请在终端中使用以下命令:wgethttp://msvocds.blob.core.w

我目前正在尝试将测试图像添加到facebook/deepmask,以便可以训练这些图像。目前,deepmask程序使用CoCo图像和注释,但我想知道我是否可以添加自己的图像。唯一的原因是我只想训练一种类型的图像,在本例中是葡萄。我查看了注释文件,但其中充满了图像的规格。我问这个问题的唯一原因是因为网上没有关于我具体问题的教程/答案

下面是github到deepmask的链接:

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

谢谢大家抽出时间


斯蒂芬

我的经验是一样的:你确实应该把你的面具换成可可式的格式。它是json,可在此处找到:。 我唯一的评论是coco使用了“肋行程编码(RLE)”,这一点没有解释。因此,很容易使用遮罩的多边形编码。 因此,请:

  • 戴上面具
  • 将其编码为多边形
  • 将其写入json文件
  • 希望这有帮助:-)