Skip to content

Instantly share code, notes, and snippets.

@aniruddha27
Created April 5, 2020 12:53
Show Gist options
  • Save aniruddha27/4d8485f235e2d174b5b32339008f63ba to your computer and use it in GitHub Desktop.
Save aniruddha27/4d8485f235e2d174b5b32339008f63ba to your computer and use it in GitHub Desktop.
# ImageDataGenerator brightness
datagen = ImageDataGenerator(brightness_range=[0.4,1.5])
# 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