ToPILImage

class torchvision.transforms.v2.ToPILImage(mode: Optional[str] = None)

[BETA] Convert a tensor or an ndarray to PIL Image

Note

The ToPILImage transform is in Beta stage, and while we do not expect disruptive breaking changes, some APIs may slightly change according to user feedback. Please submit any feedback you may have in this issue: https://github.com/pytorch/vision/issues/6753.

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).