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

<Kurs>Deep Learning für Bilder mit PyTorch</Kurs>

Kurs ansehen

Übungsanweisungen

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 praktische Übung

Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.

# 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