Visualizzare le soft mask
Nel precedente esercizio hai visto che i due oggetti più probabili segmentati dal modello Mask R-CNN sono entrambi gatti. Ora visualizzerai le maschere di questi due gatti sovrapposte all’immagine originale per verificarne visivamente l’accuratezza. Dovrai iterare sulle due maschere e, per ciascuna, tracciare prima l’immagine originale e poi applicarvi sopra una maschera semitrasparente.
Questo esercizio fa parte del corso
Deep Learning per Immagini con PyTorch
Istruzioni dell'esercizio
- Estrai maschere ed etichette da
prediction, assegnandole rispettivamente amaskselabels. - All’interno del ciclo for, visualizza la maschera
i-esima sull’immagine passandomask[i, 0]alla funzione di plotting, usando la mappa di colori"jet"e impostando il parametro di trasparenza a0.5.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# Extract masks and labels from prediction
masks = ____
labels = ____
# Plot image with two overlaid masks
for i in range(2):
plt.imshow(image)
# Overlay the i-th mask on top of the image
plt.imshow(____, ____, ____)
plt.title(f"Object: {class_names[labels[i]]}")
plt.show()