Zobrazení měkkých masek

V předchozím cvičení jsi zjistil/a, že dva nejpravděpodobnější objekty segmentované modelem Mask R-CNN jsou obě kočky. Teď tyto masky zobrazíš překryté přes původní obrázek, abys vizuálně ověřil/a jejich přesnost. Budeš procházet obě masky a pro každou z nich nejprve vykreslit původní obrázek a poté přes něj poloprůhlednou masku.

Toto cvičení je součástí kurzu

Deep Learning pro obrázky s PyTorchem

Zobrazit kurz

Pokyny k cvičení

Extrahuj masky a popisky z prediction a přiřaď je do proměnných masks a labels.
Uvnitř cyklu for zobraz i-tou masku přes obrázek tak, že předáš mask[i, 0] vykreslovací funkci, použiješ barevnou mapu "jet" a nastavíš parametr průhlednosti na 0.5.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

# 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()

Upravit a spustit kód