Crearea măștilor binare

Imaginile utilizate în sarcinile de segmentare sunt de obicei adnotate cu măști la nivel de pixel. Ia în considerare această imagine cu o pisică Egyptian Mau.

cat image

În acest exercițiu și în cel următor, vei folosi masca corespunzătoare pentru a segmenta pisica din imagine. Mai întâi, va trebui să încarci masca și să o binarizezi.

Image din PIL, transforms din torchvision și torch au fost deja importate pentru tine.

Acest exercițiu face parte din cursul

Deep Learning pentru imagini cu PyTorch

Vezi cursul

Instrucțiuni pentru exercițiu

Încarcă imaginea măștii stocată în annotations/Egyptian_Mau_123.png și atribuie-o variabilei mask.
Creează un binary_mask din mask_tensor, în care fiecărui pixel egal cu 1/255 i se atribuie valoarea tensor 1.0, iar restul pixelilor primesc valoarea tensor 0.0.

Exercițiu interactiv practic

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

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

Editează și rulează codul