convert_image_dtype

torchvision.transforms.functional.convert_image_dtype(image: torch.Tensor, dtype: torch.dtype = torch.float32) → torch.Tensor

Convert a tensor image to the given dtype and scale the values accordingly This function does not support PIL Image.

Parameters

• image (torch.Tensor) – Image to be converted

• dtype (torch.dpython:type) – Desired data type of the output

Returns

Converted image

Return type

Tensor

Note

When converting from a smaller to a larger integer dtype the maximum values are not mapped exactly. If converted back and forth, this mismatch has no effect.

Raises

RuntimeError – When trying to cast torch.float32 to torch.int32 or torch.int64 as well as for trying to cast torch.float64 to torch.int64. These conversions might lead to overflow errors since the floating point dtype cannot store consecutive integers over the whole range of the integer dtype.

Examples using convert_image_dtype:

Repurposing masks into bounding boxes

Repurposing masks into bounding boxes