Afbeelding segmenteren met een masker

Met het binaire masker kun je het object — de kat — uit de afbeelding segmenteren.

Daarvoor laad je de originele afbeelding en zet je die om naar een tensor. Vervolgens maak je de object-tensor door de originele afbeelding te maskeren. Tot slot toon je het resultaat.

transforms uit torchvision zijn geïmporteerd, en de binary_mask die je in de vorige oefening hebt gemaakt is voor je beschikbaar.

Deze oefening maakt deel uit van de cursus

Deep Learning voor afbeeldingen met PyTorch

Cursus bekijken

Oefeninstructies

Maak image_tensor door de ToTensor()-transformatie, gedefinieerd als transform, toe te passen op de ruwe afbeelding.
Segmenteer de afbeelding door image_tensor te maskeren met de binary_mask en wijs het resultaat toe aan object_tensor.
Pas de al gedefinieerde to_pil_image-transformatie toe op object_tensor om deze weer te geven.

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

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

Code bewerken en uitvoeren