Skip to content

Instantly share code, notes, and snippets.

@aniruddha27

aniruddha27/rotation.py

Last active April 5, 2020 11:16
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save aniruddha27/5edfb9a54facce2a9e8fe8c6bf5a0451 to your computer and use it in GitHub Desktop.
Save aniruddha27/5edfb9a54facce2a9e8fe8c6bf5a0451 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
rotation.py
# ImageDataGenerator rotation
datagen = ImageDataGenerator(rotation_range=30, fill_mode='nearest')
# iterator
aug_iter = datagen.flow(img, batch_size=1)
# generate samples and plot
fig, ax = plt.subplots(nrows=1, ncols=3, figsize=(15,15))
# generate batch of images
for i in range(3):
# convert to unsigned integers
image = next(aug_iter)[0].astype('uint8')
# plot image
ax[i].imshow(image)
ax[i].axis('off')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment