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

演習

ソフトマスクを表示する

前の演習では、Mask R-CNN モデルがセグメント化した最も可能性の高い2つの物体はいずれも猫であることを学びました。ここでは、元画像の上にこの2匹の猫のマスクを重ねて表示し、精度を視覚的に確認します。そのために、2つのマスクを反復処理し、各マスクについて、まず元の画像を描画し、その上に半透明のマスクを重ねて表示します。

指示

100 XP
  • prediction からマスクとラベルを抽出し、それぞれ masks と labels に代入します。
  • for ループ内で、i 番目のマスクを画像上に表示します。プロット関数には mask[i, 0] を渡し、カラーマップは "jet"、透過率は 0.5 に設定します。