Affichage des masques souples
Dans l'exercice précédent, vous avez appris que les deux objets les plus susceptibles d'avoir été segmentés par le modèle Mask R-CNN sont tous deux des chats. À présent, vous allez afficher les masques de ces deux chats superposés à l'image d'origine afin de vérifier visuellement leur exactitude. Cela nécessitera d'itérer sur les deux masques et, pour chacun d'eux, de tracer l'image originale suivie d'un masque semi-transparent par-dessus.
Cet exercice fait partie du cours
Deep learning pour les images avec PyTorch
Instructions
- Extrayez les masques et les étiquettes de l'
prediction, en les assignant respectivement àmasksetlabels. - À l'intérieur de la boucle for, affichez le masque d'
ie sur l'image en passant l'mask[i, 0]à la fonction de tracé, en utilisant la carte de couleurs"jet"et en définissant le paramètre de transparence sur0.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()