Keras flow_from_directory() 只从选定的子目录中读取

Question

我的目录结构如下。

train
|- dog --> contains image files of dogs
|- cat --> contains image files of cats
|- elephant --> contains image files of elephants

我想训练一个 CNN 来识别动物，但仅限于猫和狗，而不是大象。

我想使用keras ImageDataGenerator 类来扩充数据和flow_from_directory() 方法来读取图像文件。

train_datagen = ImageDataGenerator(rescale=1./255)
train_generator = train_datagen.flow_from_directory(directory='train', class_mode='categorical', target_size=(64,64), batch_size=16, shuffle=True)

上面的代码将从'train'的所有子目录中读取数据，这是我不想要的。一种选择是创建一个新目录并复制“dog”和“cat”子目录以及其中的文件。但是有没有办法从flow_from_directory()方法本身来控制呢？

Answer 1

假设我以正确的方式理解您的问题，这应该对您有所帮助：

train_generator = train_datagen.flow_from_directory(directory='train', class_mode='categorical', target_size=(64,64), batch_size=16, shuffle=True, classes=["dog", "cat"])

这将只读取目录dog 和cat 中的图像，省略elephant 目录并为它们提供不同的分类标签。