ImageDataGenerator 目录

问题描述

我正在 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')

解决方法

检查instructions here。

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

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