遮罩

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

用於分割和偵測遮罩的 torch.Tensor 子類別，其形狀為 [..., H, W]。

參數:

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

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

• device (torch.device, optional) – 期望的裝置。如果省略且 data 是一個 torch.Tensor，裝置將會從該 Tensor 取得。否則，遮罩將會在 CPU 上建立。

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

使用 Mask 的範例

transforms v2 入門

transforms v2 入門

TVTensors 常見問題

TVTensors 常見問題