1. Učit se
  2. /
  3. Kurzy
  4. /
  5. Deep Learning pro obrázky s PyTorchem

Connected

cvičení

Stavba U-Netu: definice vrstev

V tomto a následujícím cvičení sestavíš architekturu U-Net od základů. Začneme definicí metody __init__(), kde určíš jednotlivé vrstvy a bloky vrstev, které model použije.

Kodér a transponovaná konvoluce jsou již připravené. Zbývá doplnit konvoluční bloky dekodéru. Každému z nich musíš předat správný počet vstupních a výstupních kanálů s ohledem na skip connections.

První blok, dec1, dostane na vstup zřetězení výstupu upconv3 s výstupem enc3. Výstup dec1 by pak měl odpovídat výstupu enc3. Dokážeš doplnit všechny chybějící velikosti vstupů a výstupů?

Pokyny

100 XP
  • Definuj tři konvoluční bloky dekodéru tak, že každému předáš správný počet vstupních a výstupních kanálů.