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

ćwiczenie

Segmentowanie obrazu za pomocą maski

Mając gotową maskę binarną, możesz jej użyć do wysegmentowania obiektu – w tym przypadku kota – z obrazu.

W tym celu wczytaj oryginalny obraz i przekształć go w tensor. Następnie utwórz tensor obiektu, nakładając maskę na oryginalny obraz. Na koniec wyświetl wynik.

Zaimportowano transforms z torchvision, a binary_mask utworzona w poprzednim ćwiczeniu jest już dostępna.

Instrukcje

100 XP
  • Utwórz image_tensor, stosując transformację ToTensor() zdefiniowaną jako transform do surowego obrazu.
  • Wysegmentuj obraz, nakładając binary_mask na image_tensor, a wynik przypisz do object_tensor.
  • Zastosuj już zdefiniowaną transformację to_pil_image do object_tensor, aby wyświetlić rezultat.