Créer des masques binaires

Les images pour les tâches de segmentation sont généralement annotées avec des masques au niveau des pixels. Considérez cette image d'un chat Egyptian Mau.

cat image

Dans cet exercice et le suivant, vous utiliserez le masque correspondant pour isoler le chat de l'image. D'abord, vous devrez charger le masque et le binariser.

Image de PIL, transforms de torchvision et torch ont déjà été importés pour vous.

Cette activité fait partie du cours

Deep Learning pour les images avec PyTorch

Voir le cours

Instructions de l’exercice

Chargez l'image du masque stockée dans annotations/Egyptian_Mau_123.png et assignez-la à mask.
Créez un binary_mask à partir de mask_tensor où chaque pixel égal à 1/255 reçoit une valeur de tenseur 1.0, et les autres pixels reçoivent une valeur de tenseur 0.0.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

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

Modifier et exécuter le code