Skip to content

Instantly share code, notes, and snippets.

@NMZivkovic

NMZivkovic/data_loader.py

Last active November 18, 2019 09:15
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 NMZivkovic/1bbe36648de5eb2f346f8180b361f301 to your computer and use it in GitHub Desktop.
Save NMZivkovic/1bbe36648de5eb2f346f8180b361f301 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
data_loader.py
data_loader = DataLoader(IMG_SIZE, BATCH_SIZE)
plt.figure(figsize=(10, 8))
i = 0
for img, label in data_loader.get_random_raw_images(20):
plt.subplot(4, 5, i+1)
plt.imshow(img)
plt.title("{} - {}".format(data_loader.get_label_name(label), img.shape))
plt.xticks([])
plt.yticks([])
i += 1
plt.tight_layout()
plt.show()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment