Skip to content

Instantly share code, notes, and snippets.

@pranjalAI
Created September 8, 2020 13:33
Show Gist options
  • Save pranjalAI/11e8a519695104e33d1723d600003c1c to your computer and use it in GitHub Desktop.
Save pranjalAI/11e8a519695104e33d1723d600003c1c to your computer and use it in GitHub Desktop.
datagen = ImageDataGenerator(rescale=1.0/255.0)
train_path = 'Your Image Folder Path'
train_batches = datagen.flow_from_directory(train_path, target_size=(200,200), classes=classes_required, batch_size=batch_size_train)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment