Eine Bounding Box zeichnen

Nachdem du die Bild- und Bounding-Box-Tensoren vorbereitet hast, kannst du die Box jetzt auf das Bild zeichnen und ihre Genauigkeit visuell prüfen.

torch, torchvision, torchvision.transforms wurden importiert. Das Bild wurde bereits in Tensoren umgewandelt und als image_tensor gespeichert. Die Koordinaten wurden den Variablen x_min, y_min, x_max, y_max zugewiesen.

Diese Übung ist Teil des Kurses

Deep Learning für Bilder mit PyTorch

Kurs anzeigen

Anleitung zur Übung

Importiere draw_bounding_boxes aus torchvision.utils.
Definiere die Bounding Box bbox als Liste aus x_min, y_min, x_max und y_max.
Übergib image_tensor und bbox_tensor an draw_bounding_boxes, um die Box auf das Bild zu zeichnen, und weise die Ausgabe img_bbox zu.
Wandle den Tensor img_bbox in ein Bild um.

Interaktive Übung

Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.

# Import draw_bounding_boxes
____

# Define the bounding box coordinates
bbox = ____
bbox_tensor = torch.tensor(bbox).unsqueeze(0)

# Implement draw_bounding_boxes
img_bbox = ____(____, ____, width=3, colors="red")

# Tranform tensors to image
transform = transforms.Compose([
    ____
])
plt.imshow(transform(img_bbox))
plt.show()

Code bearbeiten und ausführen