我是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:UserskashyDesktoptrial1.py", line 29, in <module>
train_datagen = ImageDataGenerator(
NameError: name 'ImageDataGenerator' is not defined
有人可以帮助我吗?
谢谢!!!
>ImageDataGenerator
是在 Keras 中定义的实用程序函数。如果您尚未导入它,请使用
from keras.preprocessing.image import ImageDataGenerator