ConvertDtype

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

[BETA] Convert input image or video to the given dtype and scale the values accordingly.

Warning

The ConvertDtype transform is in Beta stage, and while we do not expect major breaking changes, some APIs may still change according to user feedback. Please submit any feedback you may have in this issue: https://github.com/pytorch/vision/issues/6753, and you can also check out https://github.com/pytorch/vision/issues/7319 to learn more about the APIs that we suspect might involve future changes.

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.

Examples using ConvertDtype:

Transforms v2: End-to-end object detection example

Transforms v2: End-to-end object detection example