如何使用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代码和数据集结构。