1.首先安装PIL库
pip install pillow
2.导入PIL库,并打开图像
from PIL import Image
img_path = './lena.jpg'
img1 = Image.open(img_path)
print(img1.format)
print(img1)
3.PIL.Image与numpy.array之间的互相转换
img = numpy.array(im)
img = Image.fromarray(img.astype('uint8')).convert('RGB')
4.torchvision.transforms.ToPILImage
对于一个Tensor的转化过程是:
将张量的每个元素乘上255
将张量的数据类型有FloatTensor转化成Uint8
将张量转化成numpy的ndarray类型
对ndarray对象做permute (1, 2, 0)的操作
利用Image下的fromarray函数,将ndarray对象转化成PILImage形式
输出PILImage
参考文献: