Segmentace obrázku pomocí masky

Teď, když máš binární masku připravenou, můžeš ji použít k segmentaci objektu – v tomto případě kočky – z obrázku.

Nejdřív načteš původní obrázek a převedeš ho na tensor. Poté vytvoříš tensor objektu maskováním původního obrázku. Nakonec zobrazíš výsledek.

transforms z torchvision jsou již naimportované a binary_mask, kterou jsi vytvořil/a v předchozím cvičení, máš k dispozici.

Toto cvičení je součástí kurzu

Deep Learning pro obrázky s PyTorchem

Zobrazit kurz

Pokyny k cvičení

Vytvoř image_tensor aplikováním transformace ToTensor(), definované jako transform, na původní obrázek.
Segmentuj obrázek maskováním image_tensor pomocí binary_mask a výsledek ulož do object_tensor.
Aplikuj již definovanou transformaci to_pil_image na object_tensor, aby se obrázek zobrazil.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

# Load image and transform to tensor
image = Image.open("images/Egyptian_Mau_123.jpg")
transform = transforms.Compose([transforms.ToTensor()])
image_tensor = ____

# Segment object out of the image
object_tensor = ____

# Convert segmented object to image and display
to_pil_image = ____
object_image = to_pil_image(object_tensor)
plt.imshow(object_image)
plt.show()

Upravit a spustit kód