Tensor与PIL Image之间的转换

Tensor、PIL Image以及Numpy之间的转换

Tensor ->PIL Image

from torchvision.transforms import ToPILImage()
improt torch as t

image = t.randn((64,64)) #随机定义一个tensor

img = ToPILImage()(image) #tensor转为PIL Image

img.show() #显示图片

Numpy ->PIL Image

from PIL import Image
import numpy

n = numpy.random.random((2,3)) #随机定义一个数组

img = Image.fromarray(n)

img.show()