shgidi
5/7/2017 - 9:44 AM

keras_generator.py


#genertor
train_datagen = ImageDataGenerator(
        rescale=1./255,
        shear_range=0.2,
        zoom_range=0.2,
        horizontal_flip=True) #many other augmentations are avialbile

#flow
datagen.flow(x, batch_size=1, save_to_dir='preview', save_prefix='cat', save_format='jpeg')

#flow from directory
datagen.flow_from_directory( '/mnt/kaggle_crevical/train/',  target_size=(150, 150),batch_size=batch_size,class_mode='binary')