Tensor与PIL Image之间的转换

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/nekoli/article/details/89918058

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

 

展开阅读全文

没有更多推荐了,返回首页