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 (布尔值, 可选) – 是否对图像或视频的值进行缩放。请参阅 Dtype 和预期值范围。默认值:
False
。
使用
ToDtype
的示例transforms v2 入门Transforms v2:端到端目标检测/分割示例如何使用 CutMix 和 MixUp