Afișarea măștilor soft

În exercițiul anterior, ai aflat că primele două obiecte identificate de modelul Mask R-CNN sunt ambele pisici. Acum vei afișa măștile corespunzătoare acestor două pisici suprapuse peste imaginea originală, pentru a verifica vizual acuratețea lor. Vei itera prin cele două măști și, pentru fiecare, vei reprezenta mai întâi imaginea originală, apoi masca semi-transparentă deasupra ei.

Acest exercițiu face parte din cursul

Deep Learning pentru imagini cu PyTorch

Vezi cursul

Instrucțiuni pentru exercițiu

Extrage măștile și etichetele din prediction și atribuie-le variabilelor masks, respectiv labels.
În interiorul buclei for, afișează masca cu indicele i peste imagine, transmițând mask[i, 0] funcției de reprezentare grafică, folosind schema de culori "jet" și setând parametrul de transparență la 0.5.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

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

Editează și rulează codul