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

演習

U-Net を構築する：レイヤーの定義

この演習と次の演習では、U-Net アーキテクチャを最初から構築します。ここでは、__init__() メソッドを定義し、モデルで使用するレイヤーやレイヤーブロックを用意します。

エンコーダと転置畳み込みはすでに用意されています。残っているのはデコーダ側の畳み込みブロックです。スキップ接続を考慮しながら、各ブロックに適切な入力チャネル数と出力チャネル数を渡してください。

最初のブロック dec1 は、upconv3 の出力と enc3 の出力を連結したものを入力に取ります。さらに、dec1 の出力は enc3 の出力と同じになるべきです。不足している入力サイズと出力サイズをすべて埋められますか？

指示

100 XP
  • デコーダ内の 3 つの畳み込みブロックについて、各ブロックに適切な入力チャネル数と出力チャネル数を渡して定義してください。