ToPILImage — Torchvision 0.22 documentation (original) (raw)

class torchvision.transforms.v2.ToPILImage(mode: Optional[str] = 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 is None (default) there are some assumptions made about the input data:

• If the input has 4 channels, the mode is assumed to be RGBA.
• If the input has 3 channels, the mode is assumed to be RGB.
• If the input has 2 channels, the mode is assumed to be LA.
• If the input has 1 channel, the mode is determined by the data type (i.e int, float,short).

transform(inpt: Union[Tensor, Image, ndarray], params: Dict[str, Any]) → Image[source]¶

Method to override for custom transforms.

See How to write your own v2 transforms