Keras Image Обработка этикеток

Question

Я следую за прохождением: https://blog.keras.io/building-powerful-image-classification-models-using-very-little-data.html

Я не понимаю, как ImageDataGenerator создает данные метки.

Мои каталоги изображений настроены как: .../train/cats и .../train/dogs

Мой ImageDataGenerator создан:

train_datagen = ImageDataGenerator(rescale=1./255, rotation_range=40, width_shift_range=0.2, height_shift_range=0.2, shear_range=0.2, zoom_range=0.2, horizontal_flip=True,)

train_generator = train_datagen.flow_from_directory(train_dir, target_size=(150,150), batch_size=32, class_mode='binary')

Я проверяю данные с помощью:

for data_batch, labels_batch in train_generator: print("labels: ", labels_batch)

Кажется, что ярлыки генерируются просто отлично. У меня вопрос, как ImageDataGenerator создает метки, когда я только кормлю его .jpg файлами? Я хотел бы

Answer 1

Из имен подкаталогов.

https://keras.io/preprocessing/image/#flow_from_directory

"классы: необязательный список подкаталогов классов (например, ['dogs', 'cats']). По умолчанию: нет. Если нетпри условии, что список классов будет автоматически выведен из имен / структуры подкаталога в каталоге, где каждый подкаталог будет обрабатываться как отдельный класс (и порядок классов, который будет сопоставлен с индексами меток, будет буквенно-цифровым).Словарь, содержащий сопоставление имен классов и индексов классов, можно получить с помощью атрибута class_indices. "