加载Mask RCNN的数据

Question

我想在我自己的数据集上训练Mask-RCNN。我已经有了树叶的分割图像(地面实况)，看起来像下面的图像：

​

​

如何加载训练Mask RCNN的数据集？

Answer 1

由于Mask RCNN在COCO数据集上进行了预训练，因此您需要使用这些图像对其进行训练。为此，您必须标记它们并对其进行训练。由于涉及到掩码，因此使用诸如VGG annotator之类的工具来进行必要的注释和标记，它将根据您的类生成一个json文件。稍后，根据您的需求，您必须为您的类运行.py文件，训练它，然后生成它以进行测试。

Answer 2

您必须将其转换为TFrecords，以便掩码RCNN模型能够读取图像及其注释。请参考这篇媒体文章'https://medium.com/@vijendra1125/custom-mask-rcnn-using-tensorflow-object-detection-api-101149ce0765‘

Answer 3

您可以使用coco注解( here is an example )来注释您的数据集，然后像使用coco dataset一样运行它。

您还可以检查以下代码：https://github.com/matterport/Mask_RCNN/blob/master/samples/shapes/train_shapes.ipynb