运行语义分割
设计 U-Net 做得很好!现在为您提供了一个与您刚刚构建的模型非常相似的、已经预训练好的模型。该模型在大规模图像集上完成了训练,并在架构上做了少量增强,例如加入了批量归一化层。
您可以通过 UNet() 来实例化模型,这将加载预训练权重。您的任务是使用它为下面这张汽车图像生成分割掩码。

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()