Skip to content

Instantly share code, notes, and snippets.

@AmrutaKoshe

AmrutaKoshe/datagen.py

Created July 8, 2021 07:45
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 AmrutaKoshe/01580e342f3285fe6147a5a2620b2454 to your computer and use it in GitHub Desktop.
Save AmrutaKoshe/01580e342f3285fe6147a5a2620b2454 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
datagen.py
img_datagen = ImageDataGenerator(rescale=1./255,
vertical_flip=True,
horizontal_flip=True,
rotation_range=40,
width_shift_range=0.2,
height_shift_range=0.2,
zoom_range=0.1,
validation_split=0.2)
test_datagen = ImageDataGenerator(rescale=1./255)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment