本文介绍了如何在PyTorch中显示单个图像?的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！
问题描述
我想显示使用 ImageLoader 加载并存储在PyTorch Tensor 中的单个图像.当我尝试通过 plt.imshow(image)显示它时，我得到:
I want to display a single image loaded using ImageLoader and stored in a PyTorch Tensor. When I try to display it via plt.imshow(image) I get:
TypeError: Invalid dimensions for image data
张量的 .shape 是:
torch.Size([3, 244, 244])
如何将PyTorch张量显示为图像?
How do I display a PyTorch tensor as an image?
推荐答案
给出表示图像的 Tensor ，请使用 .permute() 将通道作为最后一个维度:
Given a Tensor representing the image, use .permute() to put the channels as the last dimension:
plt.imshow( tensor_image.permute(1, 2, 0) )
注意: permute 不复制或分配内存，而 from_numpy()也不一样.
Note: permute does not copy or allocate memory, and from_numpy() doesn't either.
这篇关于如何在PyTorch中显示单个图像?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！