pytorch 张量tensor和图片相互转化

主要利用transforms第三方库来进行转换

import torch
from torchvision import transforms

张量tensor转换为图片

toPIL = transforms.ToPILImage() #这个函数可以将张量转为PIL图片，由小数转为0-255之间的像素值
img = torch.randn(3,128,64)  # 随机生成一个张量tensor
img_PIL = toPIL(img)  # 张量tensor转换为图片
img_PIL.save('')  # 保存图片；img_PIL.show()可以直接显示图片

图片转换为张量tensor

toTensor = transforms.ToTensor()
img_tensor = toTensor(img_PIL)

参考：/t20134297/article/details/109668638