1. Nauka
    2. /
  2. Kursy
    3. /
  3. Głębokie uczenie dla obrazów z PyTorch
Connected

ćwiczenie

Nakładanie masek instancji

Świetna robota z wygenerowaniem maski semantycznej w poprzednim ćwiczeniu! Teraz możesz nadpisać ją maskami instancji w miejscach, gdzie model segmentacji instancji wykrył obiekty.

Skorzystasz z wstępnie wytrenowanego modelu MaskRCNN dostępnego w przestrzeni roboczej, aby wygenerować maski segmentacji instancji. Następnie przejdziesz przez te maski w pętli i dla każdej z nich nałożysz obszary, w których obiekt został wykryty z dużą pewnością, na wierzchnią warstwę maski semantycznej.

torch jest już zaimportowany.

Instrukcje

100 XP
  • Zainicjalizuj panoptic_mask, klonując semantic_mask.
  • Zdefiniuj pętlę for iterującą po maskach instancji, nadając zmiennej iteracyjnej nazwę mask.
  • Dla każdej maski instancji – w miejscach, gdzie jej wartość jest większa niż 0.5 – nadpisz maskę panoptyczną bieżącą wartością instance_id.