如何加载具有多个掩码的数据集（图像分割）

Question

我正在尝试使用qubvel/segmentation_models 对 Flickr 徽标数据集进行语义分割

我可以按照使用 CamVid 数据集的 qubvel 教程进行操作。但就我而言，每张图片都可以有多个蒙版。

我是这个话题的新手，我该如何处理多个面具问题？

Answer 1

简单地说，您可以将每个图像的所有可用蒙版合并到一个图像中，从而为每个图像构建一个蒙版。然后，您可以使用新数据集训练任何分割模型（如 UNet）。