ToDtype

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

将输入转换为特定的数据类型，可选择为图像或视频缩放值。

注意

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

Parameters:

• dtype (torch.dtype 或 TVTensor 的字典 -> torch.dtype) – 要转换的目标数据类型。 如果传递了一个 torch.dtype，例如 torch.float32，则只有图像和视频会被转换为该数据类型：这是为了与 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 的示例：

开始使用 transforms v2

Getting started with transforms v2

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

Transforms v2: End-to-end object detection/segmentation example

如何使用CutMix和MixUp

How to use CutMix and MixUp