为 Mask RCNN 加载数据

【问题标题】：Load data for Mask RCNN为 Mask RCNN 加载数据
【发布时间】：2020-07-25 11:37:46
【问题描述】：

我想在我自己的数据集上训练 Mask-RCNN。我已经有了叶子的分割图像（基本事实），如下图所示：

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

【问题讨论】：

标签： tensorflow computer-vision image-segmentation

【解决方案1】：

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

【讨论】：

【解决方案2】：

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

【讨论】：

【解决方案3】：

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

您也可以查看此代码：https://github.com/matterport/Mask_RCNN/blob/master/samples/shapes/train_shapes.ipynb

【讨论】：