ConvertImageDtype

class torchvision.transforms.v2.ConvertImageDtype(dtype: dtype = torch.float32)

[已棄用] 請改用 v2.ToDtype(dtype, scale=True)。

將輸入影像轉換為指定的 dtype 並相應地縮放值。

警告

建議改用 ToDtype(dtype, scale=True)。請參閱 ToDtype。

此函式不支援 PIL Image。

引數:

dtype (torch.dpython:type) – 期望的輸出資料型別

注意

當從較小整數 dtype 轉換為較大整數 dtype 時，最大值不會精確對映。如果來回轉換，這種不匹配沒有影響。

引發:

RuntimeError – 嘗試將 torch.float32 轉換為 torch.int32 或 torch.int64，以及嘗試將 torch.float64 轉換為 torch.int64 時會引發此錯誤。這些轉換可能導致溢位錯誤，因為浮點 dtype 無法在其整個範圍內儲存整數 dtype 的連續整數。

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

用於自定義變換的方法（需要覆蓋此方法）。

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