Skip to content

Instantly share code, notes, and snippets.

@AmrutaKoshe

AmrutaKoshe/augment.py

Created July 13, 2021 07:57
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/b827efc03861f22376f7de1dc56f5008 to your computer and use it in GitHub Desktop.
Save AmrutaKoshe/b827efc03861f22376f7de1dc56f5008 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
augment.py
train_datagen = ImageDataGenerator(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)
val_datagen = ImageDataGenerator()
train_gen = train_datagen.flow(X_train, y_train, batch_size=batch_size)
val_gen = val_datagen.flow(X_val, y_val, batch_size=batch_size)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment