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 preserving the value range.
- 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, themode
is assumed to beRGBA
. - If the input has 3 channels, themode
is assumed to beRGB
. - If the input has 2 channels, themode
is assumed to beLA
. - If the input has 1 channel, themode
is determined by the data type (i.eint
,float
,short
).
Examples using
ToPILImage
: