Een bounding box tekenen

Nu je de tensors voor de afbeelding en de bounding box hebt voorbereid, kun je het kader over de afbeelding tekenen en visueel controleren hoe nauwkeurig het is.

torch, torchvision, torchvision.transforms zijn geïmporteerd. De afbeelding is al omgezet naar tensors als image_tensor. De coördinaten zijn toegewezen aan de variabelen: x_min, y_min, x_max, y_max.

Deze oefening maakt deel uit van de cursus

Deep Learning voor afbeeldingen met PyTorch

Cursus bekijken

Oefeninstructies

Importeer draw_bounding_boxes uit torchvision.utils.
Definieer de bounding box bbox als een lijst met x_min, y_min, x_max en y_max.
Geef image_tensor en bbox_tensor door aan draw_bounding_boxes om het kader over de afbeelding te tekenen en ken de output toe aan img_bbox.
Zet de img_bbox-tensor om naar een afbeelding.

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

# 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 bewerken en uitvoeren