快捷方式

convert_image_dtype

torchvision.transforms.functional.convert_image_dtype(image: Tensor, dtype: dtype = torch.float32) Tensor[源代码]

将张量图像转换为给定的 dtype 并相应地缩放值 此函数不支持 PIL 图像。

参数::
  • image (torch.Tensor) – 要转换的图像

  • dtype (torch.dpython:type) – 输出的所需数据类型

返回::

转换后的图像

返回类型::

Tensor

注意

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

引发::

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

使用 convert_image_dtype 的示例

将掩码重新用于边界框

将掩码重新用于边界框

文档

访问 PyTorch 的综合开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源