ToDtype

class torchvision.transforms.v2.ToDtype(dtype: Union[dtype, dict[Union[type, str], Optional[torch.dtype]], scale: bool = False)

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

注意

ToDtype(dtype, scale=True) 是 ConvertImageDtype(dtype) 的推荐替代品。

参数:

• dtype (torch.dtype 或 TVTensor 的字典 -> torch.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 (bool, optional) – 是否为图像或视频缩放值。请参阅 dtype 和预期值范围。默认值：False。

ToDtype 示例

如何使用 CutMix 和 MixUp

如何使用 CutMix 和 MixUp

变换 v2：端到端目标检测/分割示例

变换 v2：端到端目标检测/分割示例

变换 v2 入门

变换 v2 入门

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

用于覆盖自定义变换的方法。

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