pytorch如何显示图片

在PyTorch中，可以使用matplotlib库来显示图片。以下是一种常见的显示图片的方法：

import matplotlib.pyplot as plt
import torchvision

# 加载图片
image = torchvision.datasets.ImageFolder('path_to_image_folder', transform=torchvision.transforms.ToTensor())

# 显示图片
plt.imshow(image.permute(1, 2, 0))
plt.axis('off')
plt.show()

上述代码中，首先使用`torchvision.datasets.ImageFolder`加载图片，`'path_to_image_folder'`是图片所在文件夹的路径。然后使用`torchvision.transforms.ToTensor()`将图片转换为张量。接下来，使用`plt.imshow()`函数显示图片，`image`表示第一张图片，`.permute(1, 2, 0)`用于调整张量的维度顺序。最后，使用`plt.axis('off')`去除坐标轴，并使用`plt.show()`显示图片。