ConvertImageDtype

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

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

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

Note

The ConvertImageDtype 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.

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.