ToImagePIL¶
- class torchvision.transforms.v2.ToImagePIL(mode: Optional[str] = None)[source]¶
[BETA] Convert a tensor or an ndarray to PIL Image - this does not scale values.
Warning
The ToImagePIL transform is in Beta stage, and while we do not expect major breaking changes, some APIs may still change according to user feedback. Please submit any feedback you may have in this issue: https://github.com/pytorch/vision/issues/6753, and you can also check out https://github.com/pytorch/vision/issues/7319 to learn more about the APIs that we suspect might involve future changes.
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
).