ConvertImageDtype

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

[已弃用] 请改用 v2.ToDtype(dtype, scale=True)。

将输入图像转换为指定的 dtype 并相应地缩放值。

警告

建议改用 ToDtype(dtype, scale=True)。请参阅 ToDtype。

此函数不支持 PIL Image。

参数:

dtype (torch.dpython:type) – 期望的输出数据类型

注意

当从较小整数 dtype 转换为较大整数 dtype 时，最大值不会精确映射。如果来回转换，这种不匹配没有影响。

引发:

RuntimeError – 尝试将 torch.float32 转换为 torch.int32 或 torch.int64，以及尝试将 torch.float64 转换为 torch.int64 时会引发此错误。这些转换可能导致溢出错误，因为浮点 dtype 无法在其整个范围内存储整数 dtype 的连续整数。

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

用于自定义变换的方法（需要覆盖此方法）。

请参阅 如何编写自己的 v2 变换