ToPILImage¶
- class torchvision.transforms.ToPILImage(mode=None)[source]¶
Convert a tensor or an ndarray to PIL Image
This transform does not support torchscript.
Converts a torch.*Tensor of shape C x H x W or a numpy ndarray of shape H x W x C to a PIL Image while adjusting the value range depending on the
mode
.- Parameters:
mode (PIL.Image mode) –
color space and pixel depth of input data (optional). If
mode
isNone
(default) there are some assumptions made about the input data:If the input has 4 channels, the
mode
is assumed to beRGBA
.If the input has 3 channels, the
mode
is assumed to beRGB
.If the input has 2 channels, the
mode
is assumed to beLA
.If the input has 1 channel, the
mode
is determined by the data type (i.eint
,float
,short
).