Affichage de masques souples
Dans l'exercice précédent, vous avez vu que les deux objets les plus probables segmentés par le modèle Mask R-CNN sont des chats. Vous allez maintenant afficher les masques de ces deux chats superposés à l'image originale pour vérifier visuellement leur exactitude. Pour ce faire, vous devez itérer sur les deux masques et, pour chacun, tracer l'image originale, puis y superposer un masque semi-transparent.
Cette activité fait partie du cours
Deep Learning pour les images avec PyTorch
Instructions de l’exercice
- Extrayez les masques et les étiquettes de
prediction, et affectez-les àmasksetlabels, respectivement. - À l'intérieur de la boucle for, affichez le masque
i-ième sur l'image en passantmask[i, 0]à la fonction de traçage, en utilisant la palette de couleurs"jet"et en réglant le paramètre de transparence à0.5.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# 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()