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

Connected

exercițiu

Budowanie U-Netu: metoda forward

Po zdefiniowaniu warstw enkodera i dekodera możesz teraz zaimplementować metodę forward() sieci U-Net. Dane wejściowe zostały już przepuszczone przez enkoder. Twoim zadaniem jest zdefiniowanie ostatniego bloku dekodera.

Celem dekodera jest próbkowanie w górę map cech tak, aby jego wyjście miało taką samą wysokość i szerokość jak obraz wejściowy sieci U-Net. Dzięki temu uzyskasz maski semantyczne na poziomie pikseli.

Instrucțiuni

100 XP
  • Zdefiniuj ostatni blok dekodera, używając torch.cat() do utworzenia połączenia pomijającego (skip connection).