Как получить правильную метку при прогнозировании в Keras с использованием структуры папок?

Question

Поскольку мой набор данных представляет собой набор изображений, я использую структуру папок для организации данных:

/train
    /class1
        img.jpg
        img.jpg
        ...
    /class2
        ...
/validation
    /class1
        ...
    /class2
        ...

Существует только два класса, поэтому я использовал режим двоичного класса, например, так:

train_generator = train_datagen.flow_from_directory(
    train_data_dir,
    target_size=(img_width, img_height),
    batch_size=batch_size,
    class_mode='binary')

validation_generator = test_datagen.flow_from_directory(
    validation_data_dir,
    target_size=(img_width, img_height),
    batch_size=batch_size,
    class_mode='binary')

Теперь, когда используется функция predict(), выходным значением, очевидно, является числовое значение от 0 до 1. Однако я не знаю, какая метка принадлежит какому значению (0 или 1).Как мне получить настоящий ярлык (class1 или class2)?

Answer 1

, чтобы получить имена классов, просто позвоните:

generator.classes