Mask¶
- class torchvision.tv_tensors.Mask(data: Any, *, dtype: Optional[dtype] = None, device: Optional[Union[device, str, int]] = None, requires_grad: Optional[bool] = None)[source]¶
用于分割和检测掩码的
torch.Tensor
子类,形状为[..., H, W]
。- 参数:
data (类似 tensor 的, PIL.Image.Image) – 任何可以转换为 tensor 的数据,使用
torch.as_tensor()
以及 PIL 图像。dtype (torch.dtype, 可选) – 期望的数据类型。如果省略,将从
data
推断。device (torch.device, 可选) – 期望的设备。如果省略且
data
是torch.Tensor
,则设备从中获取。否则,掩码在 CPU 上构建。requires_grad (bool, 可选) – 是否自动求导应该记录操作。如果省略且
data
是torch.Tensor
,则值从中获取。否则,默认为False
。
使用
Mask
的示例