tensor图像的显示

from PIL import Image
import matplotlib.pyplot as plt
import torch
import torchvision.transforms as transforms

ori_img = Image.open('./dog1.jpg')
# plt.imshow(ori_img)
# plt.show()
print(ori_img.size)
my_transforms = transforms.Compose([
    transforms.Resize((160,160)),
    transforms.ToTensor()
])
unloader = transforms.ToPILImage()

ori_img = my_transforms(ori_img)
print(ori_img.shape)

img_ = ori_img.cpu().clone()
img_ = img_.squeeze(0)
# 不能直接使用transforms.ToPILImage 方法来获取图像，如下面注释中的地方，具体情况我暂时也不知道
img_ = unloader(img_)
# img_ = transforms.ToPILImage(img_)

plt.imshow(img_)
plt.show()



def shou_tensor_img(tensor_img:torch.Tensor):
    to_pil = torchvision.transforms.ToPILImage()
    img = tensor_img.cpu().clone()
    img = to_pil(img)
    plt.imshow(img)
    plt.show()