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.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, 可选) – 是否对图像或视频的值进行缩放。参见 Dtype 和期望值范围。默认值：False。

使用 ToDtype 的示例

如何使用 CutMix 和 MixUp

如何使用 CutMix 和 MixUp

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

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

Transforms v2 入门

Transforms v2 入门

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

自定义转换时需要覆盖的方法。

参见 如何编写自己的 v2 转换