1. Learn
    2. /
  2. कोर्स
    3. /
  3. 使用 PyTorch 进行图像深度学习
Connected

अभ्यास

使用预训练的 Mask R-CNN 进行分割

在本练习中，您将使用预训练的 Mask R-CNN 模型，对下面这张包含 2 只猫的图像执行实例分割。

two cats image

您将使用的模型已在 COCO 数据集 上进行了预训练。该数据集包含常见物体（包括动物）的图像。因此，模型无需微调就应该可以直接识别猫。

您的任务是加载模型和两只猫的图像，准备图像，并将其传入模型以获得预测结果。PIL 中的 Image、torch、torchvision 中的 transforms，以及 maskrcnn_resnet50_fpn 已为您导入。

निर्देश

100 XP
  • 使用 maskrcnn_resnet50_fpn() 加载带有 pretrained 的 Mask R-CNN 到 model。
  • 将两只猫的图像转换为张量并进行 unsqueeze。
  • 将图像传入模型进行推理，并将输出赋给 prediction。