如何将Torch Tensor转换为PIL图像

ToPILImage()转换将torch tensor转换为 PIL 图像。torchvision.transforms模块提供了许多重要的转换，可用于对图像数据执行不同类型的操作。 ToPILImage() 可接受形状为 [C, H, W] 的torch tensor，其中 C、H 和 W 分别为相应PIL图像的通道数、图像高度和宽度。

步骤

可以使用以下步骤将torch tensor转换为PIL图像 −

导入所需的库。在以下所有示例中，所需的Python库是 torch, Pillow 和 torchvision 。请确保您已经安装了它们。

import torch
import torchvision
import torchvision.transforms as T
from PIL import Image

定义一个形状为 [C, H, W] 的torch张量。

tensor = torch.rand(3,256,256)

定义一个转换来将torch张量转换为PIL图像。

transform = T.ToPILImage()

将上述定义的转换应用于输入的 torch 张量，将其转换为 PIL 图像。

img = transform(tensor)

展示转换后的PIL图像。

img.show()

例子

请看下面的例子 −

# import required libraries
import torch
import torchvision.transforms as T
from PIL import Image

# define a torch tensor
tensor = torch.rand(3,300,700)

# define a transform to convert a tensor to PIL image
transform = T.ToPILImage()

# convert the tensor to PIL image using above transform
img = transform(tensor)

# display the PIL image
img.show()