Created
August 3, 2017 13:06
-
-
Save abehmiel/6167cf71863bdf736df84376b18ce39e to your computer and use it in GitHub Desktop.
configure random transformations and normalization operations to be done on your image data during training
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from keras.preprocessing.image import ImageDataGenerator, array_to_img, img_to_array, load_img | |
datagen = ImageDataGenerator( | |
rotation_range=40, | |
width_shift_range=0.2, | |
height_shift_range=0.2, | |
shear_range=0.2, | |
zoom_range=0.2, | |
horizontal_flip=True, | |
fill_mode='nearest') | |
img = load_img('data/train/cats/cat.0.jpg') # this is a PIL image | |
x = img_to_array(img) # this is a Numpy array with shape (3, 150, 150) | |
x = x.reshape((1,) + x.shape) # this is a Numpy array with shape (1, 3, 150, 150) | |
# the .flow() command below generates batches of randomly transformed images | |
# and saves the results to the `preview/` directory | |
i = 0 | |
for batch in datagen.flow(x, batch_size=1, | |
save_to_dir='preview', save_prefix='cat', save_format='jpeg'): | |
i += 1 | |
if i > 20: | |
break # otherwise the generator would loop indefinitely |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment