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

演習

マスクを使った画像のセグメンテーション

バイナリマスクの準備ができたら、それを使って画像から対象（ここでは猫）を切り出せます。

そのために、まず元の画像を読み込み、テンソルに変換します。次に、元の画像にマスクをかけてオブジェクトのテンソルを作成します。最後に、結果を表示します。

torchvision の transforms はインポート済みで、前の演習で作成した binary_mask も利用できます。

指示

100 XP
  • transform として定義された ToTensor() 変換を生の画像に適用し、image_tensor を作成します。
  • image_tensor に binary_mask を適用してマスキングし、結果を object_tensor に代入して画像をセグメント化します。
  • すでに定義済みの to_pil_image 変換を object_tensor に適用し、表示できるようにします。