1. Nauka
  2. /
  3. Kursy
  4. /
  5. Głębokie uczenie dla obrazów z PyTorch

Connected

ćwiczenie

Budowanie U-Netu: definicje warstw

W tym i kolejnym ćwiczeniu zbudujesz architekturę U-Net od podstaw. Zacznij od zdefiniowania metody __init__(), w której określisz warstwy i bloki warstw wykorzystywane przez model.

Koder oraz transpozycyjna splotowa zostały już dla ciebie zdefiniowane. Pozostało zdefiniowanie bloków splotowych dekodera. Musisz przekazać do każdego z nich odpowiednią liczbę kanałów wejściowych i wyjściowych, uwzględniając połączenia pomostowe (skip connections).

Pierwszy blok, dec1, przyjmuje jako wejście konkatenację wyjścia upconv3 z wyjściem enc3. Wyjście dec1 powinno z kolei odpowiadać wyjściu enc3. Czy potrafisz uzupełnić wszystkie brakujące rozmiary wejść i wyjść?

Instrukcje

100 XP
  • Zdefiniuj trzy bloki splotowe dekodera, przekazując do każdego z nich odpowiednią liczbę kanałów wejściowych i wyjściowych.