pytorch显示图像

在PyTorch中，可以使用matplotlib库的imshow()函数来显示图像。具体步骤如下：
1. 读取图像数据，可以使用PIL库或OpenCV库等。
2. 将图像数据转换为PyTorch的Tensor格式。
3. 将Tensor格式的图像数据转换为numpy数组格式。
4. 使用imshow()函数显示图像。

下面是一个示例代码：

import torch
import numpy as np
import matplotlib.pyplot as plt

# 读取图像数据
img = Image.open('image.jpg')

# 将图像数据转换为PyTorch的Tensor格式
img_tensor = transforms.ToTensor()(img)

# 将Tensor格式的图像数据转换为numpy数组格式
img_numpy = img_tensor.numpy()

# 使用imshow()函数显示图像
plt.imshow(np.transpose(img_numpy, (1, 2, 0)))
plt.show()

其中，transforms.ToTensor()函数可以将PIL图像数据转换为PyTorch的Tensor格式。np.transpose()函数可以将Tensor格式的图像数据转换为numpy数组格式，并将通道维度放到最后。最后使用plt.imshow()函数显示图像。