Segmentation d'une image à l'aide d'un masque

Une fois le masque binaire prêt, vous pouvez l'utiliser pour isoler l'objet, c'est-à-dire le chat, de l'image.

Pour ce faire, vous devrez charger l'image originale et la transformer en tenseur. Ensuite, vous allez créer le tenseur d'objet en masquant l'image originale. Enfin, vous afficherez le résultat.

transforms ont été importés depuis torchvision, et le fichier binary_mask que vous avez créé lors de l'exercice précédent est à votre disposition.

Cet exercice fait partie du cours

Deep learning pour les images avec PyTorch

Afficher le cours

Instructions

Créez l'image_tensor en appliquant la transformation ToTensor(), définie comme transform, à l'image brute.
Segmentez l'image en masquant l'image_tensor e avec l'binary_mask, puis attribuez le résultat à object_tensor.
Appliquez la transformation déjà définie to_pil_image à l'élément object_tensor afin de l'afficher.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# Load image and transform to tensor
image = Image.open("images/Egyptian_Mau_123.jpg")
transform = transforms.Compose([transforms.ToTensor()])
image_tensor = ____

# Segment object out of the image
object_tensor = ____

# Convert segmented object to image and display
to_pil_image = ____
object_image = to_pil_image(object_tensor)
plt.imshow(object_image)
plt.show()

Modifier et exécuter le code