convert_image_dtype¶
- torchvision.transforms.functional.convert_image_dtype(image: Tensor, dtype: dtype = torch.float32) Tensor [source]¶
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
totorch.int32
ortorch.int64
as well as for trying to casttorch.float64
totorch.int64
. These conversions might lead to overflow errors since the floating pointdtype
cannot store consecutive integers over the whole range of the integerdtype
.
Examples using
convert_image_dtype
:Repurposing masks into bounding boxes
Repurposing masks into bounding boxes