Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # KERAS
- train_datagen = ImageDataGenerator(
- shear_range=10,
- zoom_range=0.2,
- horizontal_flip=True,
- preprocessing_function=preprocess_input)
- train_generator = train_datagen.flow_from_directory(
- 'data/train',
- batch_size=32,
- class_mode='binary',
- target_size=(224,224))
- validation_datagen = ImageDataGenerator(
- preprocessing_function=preprocess_input)
- validation_generator = validation_datagen.flow_from_directory(
- 'data/validation',
- shuffle=False,
- class_mode='binary',
- target_size=(224,224))
Add Comment
Please, Sign In to add comment