使用掩码分割图像
现在已经有了二值掩码，您可以用它将目标（也就是猫）从图像中分割出来。
为此，您需要先加载原始图像并将其转换为张量。接着，通过对原始图像进行掩码操作来创建目标张量。最后，显示分割结果。
torchvision 中的 transforms 已导入，且您在上一个练习中创建的 binary_mask 已可用。
本练习是课程的一部分
使用 PyTorch 进行图像深度学习
练习说明
- 通过将定义为
transform的ToTensor()变换应用到原始图像，创建image_tensor。 - 使用
binary_mask对image_tensor进行掩码以分割图像，并将结果赋给object_tensor。 - 对
object_tensor应用已定义的to_pil_image变换以显示它。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Load image and transform to tensor
image = Image.open("images/Egyptian_Mau_123.jpg")
transform = transforms.Compose([transforms.ToTensor()])
image_tensor = ____
# Segment object out of the image
object_tensor = ____
# Convert segmented object to image and display
to_pil_image = ____
object_image = to_pil_image(object_tensor)
plt.imshow(object_image)
plt.show()