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

检查instructions here。

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

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