tensorflow - ImageDataGenerator 目录

Question

我正在使用 Tensorflow 中的 te ImageDataGenerator 来标记图像。

我似乎做错了什么试图指向我的培训目录，有什么想法吗？

提前致谢

  from tensorflow.keras.preprocessing.image import ImageDataGenerator
    train_datagen =ImageDataGenerator(rescale=1./255)
    train_generator = train_datagen.flow_from_directory(
            'C:\Users\Ignacio\Documents\Python for Data Science\Tensorflow Exam\Horses or humans\train',
            target_size=(150, 150),
            batch_size=32,
            class_mode='binary')

Answer 1

在此处查看说明。

您的目录应具有以下结构：

main_directory/
...class_a/
......a_image_1.jpg
......a_image_2.jpg
...class_b/
......b_image_1.jpg
......b_image_2.jpg