ConvertImageDtype

class torchvision.transforms.v2.ConvertImageDtype(dtype: dtype = torch.float32)

[DEPRECATED] Use v2.ToDtype(dtype, scale=True) instead.

Convert input image to the given dtype and scale the values accordingly.

Warning

Consider using ToDtype(dtype, scale=True) instead. See ToDtype.

This function does not support PIL Image.

Parameters:

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

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.

transform(inpt: Any, params: dict[str, Any]) → Any

Method to override for custom transforms.

See How to write your own v2 transforms