ImageDataGenerator 未定义

Question

我是 Keras 和 DL 的新手。 我从 keras 文档中得到了这段代码。想在我的系统上运行它。

train_datagen = ImageDataGenerator(
    rescale=1./255,
    shear_range=0.2,
    zoom_range=0.2,
    horizontal_flip=True)

test_datagen = ImageDataGenerator(rescale=1./255)

 train_generator = train_datagen.flow_from_directory('C:\\Users\\kashy\\Desktop\\Data\\train',target_size=(200,200),batch_size=batch_size,class_mode='binary')

 validation_generator = test_datagen.flow_from_directory('C:\\Users\\kashy\\Desktop\\Data\\validation',target_size=(200,200),batch_size=batch_size,class_mode='binary')

这是我正在处理的代码片段..

当我尝试在我的系统上运行它时...... 我收到以下错误

Traceback (most recent call last):
 File "C:\Users\kashy\Desktop\trial1.py", line 29, in <module>
train_datagen = ImageDataGenerator(
NameError: name 'ImageDataGenerator' is not defined

有人可以帮助我吗??

谢谢！！！

Answer 1

ImageDataGenerator

是 Keras 中定义的实用函数。如果您还没有导入它，请使用

导入它

from keras.preprocessing.image import ImageDataGenerator

Answer 2

如果导入后遇到错误，错误信息为：

ImportError: cannot import name 'ImageDataGenerator' from 'keras.preprocessing.image'

---

你也可以尝试这行代码：

from tensorflow.keras.preprocessing.image import ImageDataGenerator.