ToPILImage

class torchvision.transforms.v2.ToPILImage(mode: Optional[str] = None)

将张量或ndarray转换为PIL图像

此转换不支持torchscript。

将形状为 C x H x W 的 torch.*Tensor 或形状为 H x W x C 的 numpy ndarray 转换为 PIL 图像，同时根据 mode 调整值范围。

Parameters:

mode (PIL.Image mode) –

输入数据的色彩空间和像素深度（可选）。 如果 mode 是 None（默认值），则对输入数据做出以下假设：

• 如果输入有4个通道，则假定 mode 为 RGBA。

• 如果输入有3个通道，则假定 mode 为 RGB。

• 如果输入有2个通道，则假定 mode 为 LA。

• 如果输入有1个通道，则 mode 由数据类型决定（即 int, float, short）。