Keras 如何对子目录使用\u directoey（）中的流\u？

我的目录看起来像：

-train->class-1->images
               ->masks
      ->class-2->images
               ->masks
data_generator = ImageDataGenerator()
train_generator = data_generator.flow_from_directory(r'C:\Users\user\project-system\New folder\train\\',target_size=(image_size,image_size),batch_size=batch_size,class_mode='categorical')

---

如何定义ImageDataGenerator（）.flow_from_directoey（）？我只想从class-1和class-2加载掩码。

根据Keras文档，

flow\u from\u directory

获取目录路径并生成成批的增强数据。每个类的目标目录应该包含一个子目录。每个子目录目录树中的任何PNG、JPG、BMP、PPM或TIF图像都将包含在生成器中

因此，如果只想加载遮罩，则应将文件夹设置为：

-train_masks ->class-1->masks
             ->class-2->masks

-train_image ->class-1->images
             ->class-2->images

---

我不能因为这个过程而改变，我正在跟随。所以我写了一个程序在一个目录中创建另一个class-1，在class-1文件夹下，我存储了来自mask的所有图像