Когда я впервые использовал Keras для CNN, изображения были разделены в каталоге обучения, например: dataset/training_set/cats и dataset/training_set/dogs, и то же самое относится к каталогу набора тестирования. Вывод ImageDataGenerator().flow_from_directory() дал мне эту строку: Найдено 8000 изображений, принадлежащих 2 классам.
Так называются изображения в папке:
- id_1_label_1.png
- id_2_label_2.png
И вот картинка, чтобы было понятнее:
Обучаемые данные в моем каталоге
Метки являются частью имя файла, как показано выше.
Я попробовал запустить следующий код для чтения изображений:
Код: Выделить всё
train_datagen = ImageDataGenerator(rescale = 1./255,
shear_range = 0.2,
zoom_range = 0.2,
horizontal_flip = True)
training_set = train_datagen.flow_from_directory('dataset/train',
target_size = (64, 64),
batch_size = 32,
class_mode = 'categorical')
Как загрузить данные с помощью Keras для CNN, чтобы ImageDataGenerator понял, сколько у нас классов?
Подробнее здесь: https://stackoverflow.com/questions/786 ... ot-labeled