python - 如何使用 keras imagedatagenerator 为unet加载图像和掩码

Question

我需要加载unet模型的图像和掩码，我不想使用cv2或任何libaraires来加载图像。我想使用简单的方法来加载这些图像。我的数据集结构

data/train
    1.png
    2.png

data/mask
    1_mask.png

在谷歌搜索时，我找到了图像和掩码代码的 keras 文档。但它需要将图像和掩码存储在数组中。或者任何其他方法来加载图像和掩码。

以及如果出现以下情况如何工作。

/train:
      /dir_name/
                /images
                /masks
      /dir_name2/
                /images
                /masks
      /dir_name3/
                /images
                /masks
      /dir_name4/
                /images
                /masks

Answer 1

查看:

对于第一种类型的数据集

点击here !使用 keras imagedatagenrator 代码和数据集结构。

对于第二种类型:点击here