tensorflow - 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)

上面的代码将从“火车”的所有子目录中读取数据，这是我所不希望的。一种选择是创建一个新目录，并复制“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目录并为它们提供不同的类别标签。