如何使用我自己的数据集使用 Tensorflow 进行图像分割

数据挖掘 张量流 数据集 图像分割
2022-03-11 15:38:10

使用我自己创建的数据集使用 Tensorflow 进行图像分割时,我遇到了一个巨大的问题。我构建的数据集包含如下所示的图像:

数据集示例

我遇到的问题是:如何使用自己的数据集专门用于图像分割?我查看了有关如何创建数据集的文档,但所有示例要么仅使用单个类的对象检测,要么对整个图像进行分类。我想将每个像素分配给一个类(图像分割)并用它来训练我的模型。

我还找到了图像分割的示例,但它们都使用现有的数据集,例如 Cityscapes、ADE20k 等。我如何使用自己的图像和数据并将其转换为 tensorflow 数据集,以便我可以将其用于训练?

2个回答

仅仅拥有分段图像可能是不够的。用于分割的训练数据需要采用特定格式。查看用于图像分割的 coco 数据集。有时我们需要将数据集转换为该格式。我建议阅读一些有关如何在您自己的数据集上训练掩码 rcnn 模型的内容。网上有很多文章。

这是一个描述 COCO 数据集是什么的精彩视频,这是一个描述如何实际制作它的视频。有几种选择,但就努力而言,这似乎是合理的。