1. 学ぶ
    2. /
  2. コース
    3. /
  3. PyTorch で学ぶ画像向け Deep Learning
Connected

演習

事前学習済み Mask R-CNN でセグメンテーションする

この演習では、事前学習済みの Mask R-CNN モデルを使って、次の2匹の猫の画像に対してインスタンスセグメンテーションを行います。

two cats image

使用するモデルは COCO dataset で事前学習されており、動物を含む一般的な物体の画像が収録されています。そのため、このモデルは追加の微調整なしで、すぐに猫を認識できるはずです。

あなたのタスクは、モデルと猫の画像を読み込み、画像を前処理してモデルに渡し、予測結果を得ることです。PIL の Image、torch、torchvision の transforms、および maskrcnn_resnet50_fpn はインポート済みです。

指示

100 XP
  • maskrcnn_resnet50_fpn() を用いて、pretrained な Mask R-CNN を model に読み込みます。
  • 猫の画像をテンソルに変換し、unsqueeze します。
  • 画像をモデルに渡して推論を行い、出力を prediction に代入します。