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