Weiche Masken anzeigen
In der letzten Übung hast du gelernt, dass die beiden wahrscheinlichsten Objekte, die das Mask R-CNN-Modell segmentiert hat, beide Katzen sind. Jetzt zeigst du die Masken für diese beiden Katzen über dem Originalbild, um zu sehen, ob alles stimmt. Dazu musst du die beiden Masken durchlaufen und für jede Maske das Originalbild zeichnen und dann eine halbtransparente Maske darüber legen.
Diese Übung ist Teil des Kurses
Deep Learning für Bilder mit PyTorch
Anleitung zur Übung
- Masken und Beschriftungen aus „
prediction“ extrahieren und sie „masks“ bzw. „labels“ zuweisen. - Innerhalb der for-Schleife zeigst du die Maske „
i“ über dem Bild an, indem du „mask[i, 0]“ an die Plotting-Funktion übergibst, die Farbkarte „"jet"“ verwendest und den Transparenzparameter auf „0.5“ setzt.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
# 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()