ToPILImage¶
- class torchvision.transforms.ToPILImage(mode=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 模式) –
输入数据的颜色空间和像素深度(可选)。如果
mode
为None
(默认值),则会对输入数据进行一些假设如果输入有 4 个通道,则假设
mode
为RGBA
。如果输入有 3 个通道,则假设
mode
为RGB
。如果输入有 2 个通道,则假设
mode
为LA
。如果输入有 1 个通道,则
mode
由数据类型确定(即int
、float
、short
)。