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 调整值范围。

参数:

mode (PIL.Image 模式) –

输入数据的颜色空间和像素深度（可选）。如果 mode 为 None（默认），则对输入数据进行一些假设

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

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

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

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

transform(inpt: Union[Tensor, Image, ndarray], params: Dict[str, Any]) → Image

用于覆盖自定义转换的方法。

请参阅 如何编写您自己的 v2 转换