1. Learn
    2. /
  2. Cursuri
    3. /
  3. Deep Learning pentru imagini cu PyTorch
Connected

exercițiu

Rularea segmentării semantice

Bravo pentru că ai proiectat U-Net-ul! Vei găsi deja disponibil un model pre-antrenat, foarte asemănător cu cel pe care tocmai l-ai construit. Acest model a fost antrenat pe un set mare de imagini și include câteva completări minore ale arhitecturii, cum ar fi straturile de normalizare batch.

Poți instanția modelul ca UNet(), care îi va furniza greutățile pre-antrenate. Sarcina ta este să îl folosești pentru a genera măști de segmentare pentru următoarea imagine a unei mașini.

car image

Image din PIL a fost deja importat pentru tine.

Instrucțiuni

100 XP
  • Instanțiază UNet() într-o variabilă numită model.
  • Încarcă imaginea de la car.jpg într-o variabilă numită image.
  • Generează măști de segmentare transmițând imaginea către model și aplicând squeeze(0) rezultatului.