Skip to content

Instantly share code, notes, and snippets.

@maskaravivek

maskaravivek/dataset_batches.py

Created June 18, 2020 02:24
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 maskaravivek/e836e464c05ec59e8c930b43e42b1f48 to your computer and use it in GitHub Desktop.
Save maskaravivek/e836e464c05ec59e8c930b43e42b1f48 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
dataset_batches.py
train, test = datasets['train'], datasets['validation']
train_batch = train.map(
lambda image, label: (tf.image.resize(image, (448, 448)), label)).shuffle(100).batch(batch_size).repeat()
validation_batch = test.map(
lambda image, label: (tf.image.resize(image, (448, 448)), label)
).shuffle(100).batch(batch_size).repeat()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment