Segmentarea imaginii cu o mască

Cu masca binară pregătită, o poți folosi pentru a segmenta obiectul – adică pisica – din imagine.

Pentru aceasta, va trebui să încarci imaginea originală și să o transformi într-un tensor. Apoi, vei crea tensorul obiectului prin aplicarea măștii asupra imaginii originale. În final, vei afișa rezultatul.

transforms din torchvision a fost deja importat, iar binary_mask creată în exercițiul anterior îți este disponibilă.

Acest exercițiu face parte din cursul

Deep Learning pentru imagini cu PyTorch

Vezi cursul

Instrucțiuni pentru exercițiu

Creează image_tensor aplicând transformarea ToTensor(), definită ca transform, asupra imaginii brute.
Segmentează imaginea prin aplicarea binary_mask asupra image_tensor, atribuind rezultatul variabilei object_tensor.
Aplică transformarea to_pil_image, deja definită, asupra object_tensor pentru a o afișa.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

# 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()

Editează și rulează codul