Bild mit einer Maske segmentieren
Wenn die Binärmaske fertig ist, kannst du sie benutzen, um das Objekt, also die Katze, aus dem Bild herauszuschneiden.
Dazu musst du das Originalbild laden und in einen Tensor umwandeln. Als Nächstes erstellst du den Objekttensor, indem du das Originalbild maskierst. Zum Schluss zeigst du das Ergebnis.
transforms aus „ torchvision “ wurden importiert, und die Datei „ binary_mask “, die du in der vorherigen Übung erstellt hast, steht dir zur Verfügung.
Diese Übung ist Teil des Kurses
Deep Learning für Bilder mit PyTorch
Anleitung zur Übung
- Erstell „
image_tensor“, indem du die Transformation „ToTensor()“, die als „transform“ definiert ist, auf das Rohbild anwendest. - Teile das Bild auf, indem du den Bereich „
image_tensor“ mit „binary_mask“ maskierst und das Ergebnis „object_tensor“ zuweist. - Wende die bereits definierte Transformation „
to_pil_image“ auf „object_tensor“ an, um sie anzuzeigen.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
# 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()