Image

class torchvision.tv_tensors.Image(data: Any, *, dtype: Optional[dtype] = None, device: Optional[Union[device, str, int]] = None, requires_grad: Optional[bool] = None)

適用於形狀為 [..., C, H, W] 的圖像的 torch.Tensor 子類別。

請注意

在 transforms 中，Image 實例與純 torch.Tensor 大致可以互換。詳情請參閱這個說明。

參數:

• data (tensor-like, PIL.Image.Image) – 任何可以使用 torch.as_tensor() 轉換為 tensor 的資料，以及 PIL 圖片。

• dtype (torch.dpython:type, optional) – 想要的資料類型。如果省略，將從 data 推斷。

• device (torch.device, optional) – 想要的裝置。如果省略且 data 是一個 torch.Tensor，則裝置將從它那裡取得。否則，影像將在 CPU 上構建。

• requires_grad (bool, optional) – 是否 autograd 應該記錄操作。如果省略且 data 是一個 torch.Tensor，則數值將從它那裡取得。否則，預設為 False。

使用 Image 的範例

transforms v2 入門

transforms v2 入門

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

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

TVTensors 常見問題

TVTensors 常見問題