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 轉換