快捷方式

ToPILImage

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

将张量或 ndarray 转换为 PIL Image

此转换不支持 torchscript。

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

参数:

mode (PIL.Image 模式) –

输入数据的颜色空间和像素深度(可选)。如果 modeNone(默认),则会对输入数据进行一些假设。

  • 如果输入有 4 个通道,则假设 modeRGBA

  • 如果输入有 3 个通道,则假设 modeRGB

  • 如果输入有 2 个通道,则假设 modeLA

  • 如果输入有 1 个通道,则 mode 由数据类型确定(例如 intfloatshort)。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并解答问题

查看资源