pytorch载入和显示图片

代码实现如下：

import numpy as np
from torchvision import transforms
import cv2

#transforms.ToTensor()
transform1=transforms.Compose([transforms.ToTensor()])
#读入图像
img=cv2.imread('cat.jpg')
img1=transform1(img) #归一化到【0.0,1.0】
print ("img1=",img1)   #输出的是像素值，应该是Tensor

#转化为numpy.ndarray并显示
img_1=img1.numpy()*255
img_1=img_1.astype('uint8')
img_1=np.transpose(img_1,(1,2,0)) #将通道数移到最后
cv2.imshow('img_1',img_1)
cv2.waitKey()

用ＰＩＬ显示图片：

from PIL import Image
import matplotlib.pyplot as plt
img=Image.open('./dog.jpg')
plt.figure("dog")
plt.imshow(img)
plt.show()