Hanrui-Wang/plot_image_batch.py

Created July 17, 2019 00:40

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Select an option

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/Hanrui-Wang/3b7b0d812f0036dc491bc7c73aff96f2.js"></script>
Save Hanrui-Wang/3b7b0d812f0036dc491bc7c73aff96f2 to your computer and use it in GitHub Desktop.

Download ZIP

Plot image batch from pytorch tensor

Raw

plot_image_batch.py

	grid = utils.make_grid(images_batch)
	plt.imshow(grid.numpy().transpose((1, 2, 0)))

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment