如何为图像分割创建带有掩码的自定义图像数据集？（特别是对于 Tensorflow）

Posted 2023-03-12

【中文标题】如何为图像分割创建带有掩码的自定义图像数据集？（特别是对于 Tensorflow）

【英文标题】：How does one create a custom dataset of images with masks for image segmentation?(specifically for Tensorflow)

【发布时间】：2020-05-12 23:21:06

【问题描述】：

我找到的每个教程都涉及使用预制，但我正在尝试做的项目是在打牌时对图片进行图像分割。数据集将是我创建的数据集，但我几乎找不到有关创建数据集和所需图像蒙版的资源。任何帮助都会很棒！

【问题讨论】：

这个tool 可能会有所帮助。

这是一个相关的答案：***.com/a/58958450/1435741

【参考方案1】：

我使用带有图层的 Gimp (https://www.gimp.org/)。您可以使用几个有用的工具，例如“BucketFill”，快速为区域着色。然后，您只需将图层导出到新文件即可获得蒙版。 VGG 图像标注器也很有用 (https://www.robots.ox.ac.uk/~vgg/software/via/via-1.0.6.html)

对于 3D 图像，您可以使用 VTK 和 ITKsnap (http://www.itksnap.org/pmwiki/pmwiki.php) 进行体积识别、可视化和导出。 MIPAV (https://mipav.cit.nih.gov/) 也很有用。