2017-12-30 58 views
0

Keras 및 DL을 처음 사용합니다. 케 라스 문서에서이 코드를 받았습니다. 제 시스템에서 실행하고 싶었습니다. 난 내 시스템에서이를 실행하려고ImageDataGenerator가 정의되지 않았습니다.

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 

누군가 할 수있는 다음과 같은 오류가 발생했습니다 도와주세요 ??

감사합니다.

답변

0

ImageDataGenerator은 Keras 내에 정의 된 유틸리티 함수입니다. 가져 오기하지 않은 경우 가져 오기

from keras.preprocessing.image import ImageDataGenerator