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

cvičení

Tvorba U-Netu: metoda forward

Teď, když máš definované vrstvy enkodéru i dekodéru, můžeš implementovat metodu forward() U-Netu. Vstupy již byly zpracovány enkodérem. Zbývá ti ale definovat poslední blok dekodéru.

Cílem dekodéru je provést upsampling příznakových map tak, aby výstup měl stejnou výšku a šířku jako vstupní obrázek U-Netu. Díky tomu získáš sémantické masky na úrovni jednotlivých pixelů.

Pokyny

100 XP
  • Definuj poslední blok dekodéru a pomocí torch.cat() vytvoř skip connection.