Afficher des masques « soft »
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 deux chats. Vous allez maintenant afficher les masques de ces deux chats superposés à l’image d’origine pour en vérifier visuellement la justesse. Pour cela, il faudra itérer sur les deux masques et, pour chacun, tracer l’image d’origine puis un masque semi-transparent par-dessus.
Cet exercice fait partie du cours
Deep Learning pour l’image avec PyTorch
Instructions
- Extrayez les masques et les étiquettes depuis
prediction, et affectez-les respectivement àmasksetlabels. - À l’intérieur de la boucle for, affichez le iᵉ masque sur l’image en passant
mask[i, 0]à la fonction de tracé, 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 cet exemple de code.
# 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()