影像

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

torch.Tensor 影像的子類，形狀為 [..., C, H, W]。

注意

在 轉換 中，Image 例項與純粹的 torch.Tensor 大體上可以互換。有關更多詳細資訊，請參閱 此注意。

引數：

• data (類張量, PIL.Image.Image) – 任何可以使用 torch.as_tensor() 轉換為張量的資料，以及 PIL 影像。

• dtype (torch.dpython:type, 可選) – 期望的資料型別。如果省略，將從 data 推斷。

• device (torch.device, 可選) – 期望的裝置。如果省略且 data 是一個 torch.Tensor，則裝置從中獲取。否則，影像將在 CPU 上構造。

• requires_grad (bool, 可選) – Autograd 是否應記錄操作。如果省略且 data 是一個 torch.Tensor，則從中取值。否則，預設為 False。

使用 Image 的示例

轉換 v2：端到端物件檢測/分割示例

轉換 v2：端到端物件檢測/分割示例

轉換 v2 入門

轉換 v2 入門

TVTensors 常見問題

TVTensors 常見問題