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

cvičení

Vytvoření sémantických masek

Běžný způsob, jak provést panoptickou segmentaci, je zkombinovat výstupy sémantické a instanční segmentace. Podívej se na následující snímek newyorské ulice.

street image

Tvým úkolem je provést její panoptickou segmentaci tak, aby byl každý taxík identifikován jako samostatný objekt a zároveň se rozlišilo pozadí tvořené asfaltem od pozadí budov.

Začneš tím, že pomocí předtrénovaného U-Netu, dostupného jako UNet(), vytvoříš sémantickou masku. Ta by měla rozlišit oba typy pozadí (ale jednotlivé taxíky od sebe neodliší)!

Pokyny

100 XP
  • Vytvoř instanci modelu U-Net jako model.
  • Vygeneruj semantic_masks předáním vstupního obrazového tenzoru do modelu.
  • Vytvoř jednu sémantickou masku výběrem třídy s nejvyšší pravděpodobností pro každý pixel.