Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from keras.preprocessing.image import ImageDataGenerator
- train_datagen = ImageDataGenerator(rescale = 1./255,
- shear_range = 0.2,
- zoom_range = 0.2,
- horizontal_flip = True)
- test_datagen = ImageDataGenerator(rescale = 1./255)
- training_set = train_datagen.flow_from_directory('dataset/train',
- target_size = (64, 64),
- batch_size = 32,
- class_mode = 'binary')
- test_set = test_datagen.flow_from_directory('dataset/validation',
- target_size = (64, 64),
- batch_size = 32,
- class_mode = 'binary')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement