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

演習

U-Net を構築する: forward メソッド

エンコーダ層とデコーダ層は定義済みなので、ここでは U-Net の forward() メソッドを実装します。入力はすでにエンコーダに通されていますが、最後のデコーダブロックを定義する必要があります。

デコーダの目的は、特徴マップをアップサンプリングして、出力の高さと幅を U-Net の入力画像と同じにすることです。これにより、ピクセル単位のセマンティックマスクを得られます。

指示

100 XP
  • スキップ接続を作るために torch.cat() を使い、最後のデコーダブロックを定義してください。