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, optional) – 是否縮放圖像或影片的值。 請參閱 Dtype 和預期值範圍。 預設值：False。

使用 ToDtype 的範例

Transforms v2 入門

Transforms v2 入門

Transforms v2：端對端物件偵測/分割範例

Transforms v2：端對端物件偵測/分割範例

如何使用 CutMix 和 MixUp

如何使用 CutMix 和 MixUp

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

用於覆寫自訂轉換的方法。

請參閱 如何編寫自己的 v2 轉換