Skip to content

Instantly share code, notes, and snippets.

@AyishaR

AyishaR/currency_load_dataset.py

Created March 6, 2021 18:03
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 AyishaR/c6f2d74f5150d2587128868078ce1a97 to your computer and use it in GitHub Desktop.
Save AyishaR/c6f2d74f5150d2587128868078ce1a97 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
currency_load_dataset.py
# Loading the dataset
path = 'currency/'
input_shape = (256, 256, 3) # default input shape while loading the images
batch = 64
# The train and test datasets
print("Train dataset")
train_ds = preprocessing.image_dataset_from_directory(path+'Train', batch_size=batch, label_mode='categorical')
print("Test dataset")
test_ds = preprocessing.image_dataset_from_directory(path+'Test', batch_size=batch, label_mode='categorical')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment