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

道练习

使用掩码分割图像

现在已经有了二值掩码，您可以用它将目标（也就是猫）从图像中分割出来。

为此，您需要先加载原始图像并将其转换为张量。接着，通过对原始图像进行掩码操作来创建目标张量。最后，显示分割结果。

torchvision 中的 transforms 已导入，且您在上一个练习中创建的 binary_mask 已可用。

说明

100 XP
  • 通过将定义为 transform 的 ToTensor() 变换应用到原始图像，创建 image_tensor。
  • 使用 binary_mask 对 image_tensor 进行掩码以分割图像，并将结果赋给 object_tensor。
  • 对 object_tensor 应用已定义的 to_pil_image 变换以显示它。