1. Learn
  2. /
  3. 课程
  4. /
  5. 使用 PyTorch 进行图像深度学习

Connected

道练习

创建二值掩码

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

cat image

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

PIL 中的 Image、torchvision 中的 transforms,以及 torch 已为您导入。

说明

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