Skip to content

Instantly share code, notes, and snippets.

@qmaruf
Created December 10, 2018 10:15
Show Gist options
  • Save qmaruf/51c20b4b1b98a2483d706fc5b5a5804e to your computer and use it in GitHub Desktop.
Save qmaruf/51c20b4b1b98a2483d706fc5b5a5804e to your computer and use it in GitHub Desktop.
def show_img_from_batch(batch):
img = transforms.ToPILImage()(batch[0])
plt.figure()
plt.imshow(img)
plt.show()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment