Weiche Masken anzeigen

In der vorherigen Übung hast du gelernt, dass die zwei wahrscheinlichsten Objekte, die das Mask-R-CNN-Modell segmentiert hat, beide Katzen sind. Jetzt wirst du die Masken für diese beiden Katzen über dem Originalbild anzeigen, um ihre Genauigkeit visuell zu überprüfen. Dazu iterierst du über die beiden Masken und zeichnest für jede zuerst das Originalbild und dann eine halbtransparente Maske darüber.

Diese Übung ist Teil des Kurses

Deep Learning für Bilder mit PyTorch

Kurs anzeigen

Anleitung zur Übung

Extrahiere Masken und Labels aus der prediction und weise sie masks bzw. labels zu.
Zeige innerhalb der For-Schleife die i-te Maske über dem Bild an, indem du mask[i, 0] an die Plot-Funktion übergibst, die Farbskala "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()

Code bearbeiten und ausführen