Skip to content

Instantly share code, notes, and snippets.

@tanmayymishra

tanmayymishra/preprocessing.py

Created September 21, 2021 19:33
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 tanmayymishra/34d1b5ce29fa1b1529069488c7d7f510 to your computer and use it in GitHub Desktop.
Save tanmayymishra/34d1b5ce29fa1b1529069488c7d7f510 to your computer and use it in GitHub Desktop.
Download ZIP
tyre
Raw
preprocessing.py
train_dir= "./tire-dataset/"
train_datagen= tf.keras.preprocessing.image.ImageDataGenerator(rescale= 1./255, validation_split=0.2)
train_generator= train_datagen.flow_from_directory(train_dir, target_size= (100,100), color_mode= 'grayscale', batch_size= 20, class_mode= 'categorical', subset= 'training')
val_generator= train_datagen.flow_from_directory(train_dir, target_size= (100,100), color_mode= 'grayscale', batch_size= 20, class_mode= 'categorical', subset= 'validation')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment