Segmentazione di un'immagine con una maschera

Con la maschera binaria pronta, puoi usarla per segmentare l'oggetto, cioè il gatto, dall'immagine.

Per farlo, dovrai caricare l'immagine originale e trasformarla in un tensore. Poi creerai il tensore dell'oggetto mascherando l'immagine originale. Infine, visualizzerai il risultato.

transforms di torchvision sono già stati importati e il binary_mask che hai creato nell'esercizio precedente è a tua disposizione.

Questo esercizio fa parte del corso

Deep Learning per Immagini con PyTorch

Visualizza il corso

Istruzioni dell'esercizio

Crea image_tensor applicando la trasformazione ToTensor(), definita come transform, all'immagine grezza.
Segnmenta l'immagine mascherando image_tensor con binary_mask, assegnando il risultato a object_tensor.
Applica la trasformazione to_pil_image già definita a object_tensor per visualizzarlo.

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

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

Modifica ed esegui il codice