convert_image_dtype

torchvision.transforms.functional.convert_image_dtype(image: Tensor, dtype: dtype = torch.float32) → Tensor

將張量影像轉換為指定的 dtype 並相應地縮放值。此函式不支援 PIL Image。

引數:

• image (torch.Tensor) – 要轉換的影像

• dtype (torch.dpython:type) – 輸出所需的日期型別

返回:

轉換後的影像

返回型別:

張量

注意

從較小的整數 dtype 轉換為較大的整數 dtype 時，最大值**不會**被精確對映。如果來回轉換，此不匹配將沒有影響。

丟擲:

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

使用 convert_image_dtype 的示例

將掩碼重新用作邊界框

將掩碼重新用作邊界框