Binaire masks maken

Afbeeldingen voor segmentatietaken worden meestal geannoteerd met maskers op pixelniveau. Bekijk deze afbeelding van een Egyptian Mau-kat.

cat image

In deze en de volgende oefening gebruik je het bijbehorende masker om de kat uit de afbeelding te segmenteren. Eerst laad je het masker en maak je het binair.

Image uit PIL, transforms uit torchvision en torch zijn al voor je geïmporteerd.

Deze oefening maakt deel uit van de cursus

Deep Learning voor afbeeldingen met PyTorch

Cursus bekijken

Oefeninstructies

Laad de maskerafbeelding uit annotations/Egyptian_Mau_123.png en ken deze toe aan mask.
Maak een binary_mask van mask_tensor waarbij elke pixel die gelijk is aan 1/255 de tensorwaarde 1.0 krijgt en de overige pixels de tensorwaarde 0.0.

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

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

Code bewerken en uitvoeren