Skip to content

Instantly share code, notes, and snippets.

@saimadhu-polamuri

saimadhu-polamuri/dataaspirant-handle-overfitting-data-augmentation.py

Created August 24, 2020 05:47
Show Gist options
  • Save saimadhu-polamuri/d777b90af02d3631dcb6b554981c8902 to your computer and use it in GitHub Desktop.
Save saimadhu-polamuri/d777b90af02d3631dcb6b554981c8902 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
dataaspirant-handle-overfitting-data-augmentation.py
## Data augmentation code
from keras.preprocessing.image import ImageDataGenerator
aug = ImageDataGenerator(
rotation_range=20,
zoom_range=0.15,
width_shift_range=0.2,
height_shift_range=0.2,
shear_range=0.15,
horizontal_flip=True,
fill_mode="nearest")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment