Creare maschere binarie

Le immagini per i compiti di segmentazione sono di solito annotate con maschere a livello di pixel. Considera questa immagine di un gatto Egyptian Mau.

cat image

In questo e nel prossimo esercizio userai la maschera corrispondente per segmentare il gatto dall'immagine. Per prima cosa, dovrai caricare la maschera e binarizzarla.

Image da PIL, transforms da torchvision e torch sono già stati importati per te.

Questo esercizio fa parte del corso

Deep Learning per Immagini con PyTorch

Visualizza il corso

Istruzioni dell'esercizio

Carica l'immagine della maschera salvata in annotations/Egyptian_Mau_123.png e assegnala a mask.
Crea una binary_mask da mask_tensor in cui a ogni pixel uguale a 1/255 venga assegnato un valore tensore di 1.0, e ai pixel rimanenti un valore tensore di 0.0.

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

# Load mask image
mask = ____

# Transform mask to tensor
transform = transforms.Compose([transforms.ToTensor()])
mask_tensor = transform(mask)

# Create binary mask
binary_mask = ____(
    ____, 
    ____,
    ____,
)

# Print unique mask values
print(binary_mask.unique())

Modifica ed esegui il codice