keras ImageDataGenerator Целевые данные

Question

Согласно этой странице мы можем построить генератор, используя ImageDataGenerator класс (и flow_from_directory метод), который мы можем передать model.fit_generatorметод в керасе; как это:

train_datagen = ImageDataGenerator(rescale=1./255)

train_generator = train_datagen.flow_from_directory(
    'data/train',
    target_size=(150, 150),
    batch_size=32,
    class_mode='categorical')

model.fit_generator(
    train_generator,
    steps_per_epoch=2000,
    epochs=50)

Но в flow_from_directory Метод это только получить изображения для обучения из каталога, а не целевые метки, потому что, в то время как в нормальном model.fit метод передачи целевых данных в качестве параметра, model.fit_generator принимать только генератор тренировочных образов; так где же взять ожидаемый результат?

Answer 1

flow_from_dictionary говорит, что "он должен содержать один подкаталог на класс", это потому, что имя подкаталога будет меткой класса, поэтому, если бы вы имели изображения собак и кошек в соответствующих помеченных каталогах, ярлыки были бы собаками и кошки, это, скорее всего, то, что вы ищете, я помню, как использовал эту функцию, я не могу ничего сказать, если она изменилась