pytorch载入和显示图片

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qxqsunshine/article/details/84887698

代码实现如下：

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()