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

cvičení

Spuštění sémantické segmentace

Skvělá práce při návrhu U-Netu! K dispozici máš již předtrénovaný model, který je velmi podobný tomu, který jsi právě sestavil/a. Tento model byl natrénován na velké sadě obrázků a obsahuje několik drobných rozšíření architektury, například vrstvy batch norm.

Model lze vytvořit jako UNet(), čímž se automaticky načtou předtrénované váhy. Tvým úkolem je použít ho k vytvoření segmentačních masek pro následující obrázek auta.

car image

Image z knihovny PIL je již naimportována.

Pokyny

100 XP
  • Vytvoř instanci UNet() a ulož ji do proměnné model.
  • Načti obrázek ze souboru car.jpg do proměnné image.
  • Vytvoř segmentační masky tak, že obrázek předáš modelu a na výstup aplikuješ squeeze(0).