开始使用免费开始使用

运行语义分割

设计 U-Net 做得很好!现在为您提供了一个与您刚刚构建的模型非常相似的、已经预训练好的模型。该模型在大规模图像集上完成了训练,并在架构上做了少量增强,例如加入了批量归一化层。

您可以通过 UNet() 来实例化模型,这将加载预训练权重。您的任务是使用它为下面这张汽车图像生成分割掩码。

car image

PIL 中的 Image 已为您导入。

本练习是课程的一部分

使用 PyTorch 进行图像深度学习

查看课程

练习说明

  • 在名为 model 的变量中实例化 UNet()
  • car.jpg 加载到名为 image 的变量中。
  • 将图像传入模型以生成分割掩码,并对输出调用 squeeze(0)

交互式实操练习

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

# Load model
model = ____
model.eval()

# Load and transform image
image = ____
transform = transforms.Compose([transforms.ToTensor()])
image_tensor = transform(image).unsqueeze(0)

# Predict segmentation mask
with torch.no_grad():
    prediction = ____

# Display mask
plt.imshow(prediction[1, :, :])
plt.show()
编辑并运行代码