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 и предоставлять им отдельные категориальные метки.