快捷方式

ToDtype

class torchvision.transforms.v2.ToDtype(dtype: Union[dtype, Dict[Union[Type, str], Optional[dtype]]], scale: bool = False)[源代码]

将输入转换为特定 dtype,可选地对图像或视频的值进行缩放。

注意

ToDtype(dtype, scale=True)ConvertImageDtype(dtype) 的推荐替代方案。

参数:
  • dtype (torch.dtypeTVTensortorch.dtype 的字典) – 要转换到的 dtype。如果传递了 torch.dtype,例如 torch.float32,则只有图像和视频将转换为该 dtype:这是为了与 ConvertImageDtype 兼容。可以传递一个字典来指定每个 tv_tensor 的转换,例如 dtype={tv_tensors.Image: torch.float32, tv_tensors.Mask: torch.int64, "others":None}。 “others” 键可以用作任何其他 tv_tensor 类型的通用键,而 None 表示不进行转换。

  • scale (布尔值, 可选) – 是否对图像或视频的值进行缩放。请参阅 Dtype 和预期值范围。默认值:False

使用 ToDtype 的示例

transforms v2 入门

transforms v2 入门

Transforms v2:端到端目标检测/分割示例

Transforms v2:端到端目标检测/分割示例

如何使用 CutMix 和 MixUp

如何使用 CutMix 和 MixUp

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源