Код в статье использует класс ImageDataGenerator
и функцию flow_from_directory()
для загрузки и дополнения изображений.Проверяя документацию Keras , вы можете найти информацию об одном из аргументов, называемом classes
:
Необязательный список подкаталогов классов (например, ['dogs', 'cats']),По умолчанию: нет.Если не указан, список классов будет автоматически выведен из имен / структуры подкаталога в каталоге, где каждый подкаталог будет обрабатываться как отдельный класс (а порядок классов, который будет сопоставлен с индексами меток, будет буквенно-цифровым).Словарь, содержащий сопоставление имен классов и индексов классов, можно получить с помощью атрибута class_indices.
На основании этой информации я подозреваю, что «кошка» относится к 0, тогда как «собака» относится к 1Чтобы убедиться, что вы можете проверить словарь классов:
print(train_datagen.class_indices)