Skip to content

Instantly share code, notes, and snippets.

@omarsar

omarsar/sample.py

Created December 29, 2019 16:06
Show Gist options
  • Save omarsar/c6849cb63bc615cad17ba0be92a0418d to your computer and use it in GitHub Desktop.
Save omarsar/c6849cb63bc615cad17ba0be92a0418d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
sample.py
## load the entire dataset
x, y = next(iter(train_dataset))
## print one example
dim = x.shape[1]
print("Dimension of image:", x.shape, "\n",
"Dimension of labels", y.shape)
plt.imshow(x[160].reshape(1, 3, 224, 224).squeeze().T.numpy())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment