Skapa binära masker

Bilder för segmenteringsuppgifter annoteras vanligtvis med masker på pixelnivå. Ta den här bilden av en egyptisk mau-katt som exempel.

cat image

I den här och nästa övning använder du den tillhörande masken för att segmentera ut katten ur bilden. Börja med att läsa in masken och binarisera den.

Image från PIL, transforms från torchvision och torch har redan importerats åt dig.

Den här övningen är en del av kursen

Djupinlärning för bilder med PyTorch

Visa kurs

Övningsinstruktioner

Läs in maskbilden som finns lagrad i annotations/Egyptian_Mau_123.png och tilldela den till mask.
Skapa en binary_mask från mask_tensor där varje pixel som är lika med 1/255 tilldelas tensorvärdet 1.0, och övriga pixlar tilldelas tensorvärdet 0.0.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

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

Redigera och kör kod