Eine Begrenzungsbox zeichnen

Nachdem du das Bild und die Begrenzungsrahmen-Tensoren vorbereitet hast, kannst du jetzt den Rahmen über das Bild zeichnen und die Genauigkeit anschauen.

torch``torchvision undtorchvision.transforms wurden importiert. Das Bild wurde schon in Tensoren umgewandelt, wie du hier sehen kannst: image_tensor. Die Koordinaten wurden den Variablen zugewiesen: x_min, y_min, x_max, y_max.

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 Begrenzungsrahmen bbox als Liste, die aus x_min, y_min, x_max und y_max besteht.
Gib „ image_tensor “ und „ bbox_tensor “ in „ draw_bounding_boxes “ ein, um das Feld über dem Bild zu zeichnen, und schick die Ausgabe an „ img_bbox “.
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