开始使用免费开始使用

创建二值掩码

用于分割任务的图像通常带有像素级掩码。请看下面这张埃及猫的图像。

cat image

在本题和下一题中,您将使用对应的掩码把猫从图像中分割出来。首先,您需要加载掩码并将其二值化。

PIL 中的 Imagetorchvision 中的 transforms,以及 torch 已为您导入。

本练习是课程的一部分

使用 PyTorch 进行图像深度学习

查看课程

练习说明

  • 加载存储在 annotations/Egyptian_Mau_123.png 的掩码图像,并将其赋给 mask
  • mask_tensor 创建一个 binary_mask:将每个像素中等于 1/255 的位置赋值为张量 1.0,其余像素赋值为张量 0.0

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Load mask image
mask = ____

# Transform mask to tensor
transform = transforms.Compose([transforms.ToTensor()])
mask_tensor = transform(mask)

# Create binary mask
binary_mask = ____(
    ____, 
    ____,
    ____,
)

# Print unique mask values
print(binary_mask.unique())
编辑并运行代码