Skip to content

Instantly share code, notes, and snippets.

@AMZzee

AMZzee/train_generator.py Secret

Created June 22, 2021 16:14
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 AMZzee/52bfa837d67c291253c25bc641663ad2 to your computer and use it in GitHub Desktop.
Save AMZzee/52bfa837d67c291253c25bc641663ad2 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
train_generator.py
train_datagen=ImageDataGenerator(preprocessing_function=preprocess_input)
train_generator=train_datagen.flow_from_directory('3D_Shapes_Dataset',
target_size=(IMAGE_SIZE,IMAGE_SIZE),
color_mode='rgb',
batch_size=32,
class_mode='categorical', shuffle=True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment