Créer des masques binaires

Les images utilisées 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 allez utiliser le masque correspondant pour extraire le chat de l’image. Vous devez d’abord charger le masque et le binariser.

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

Cet exercice fait partie du cours

Deep Learning pour l’image avec PyTorch

Afficher le cours

Instructions

Chargez l’image du masque stockée dans annotations/Egyptian_Mau_123.png et affectez-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 cet exemple de code.

# 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