Desenarea unei cutii de delimitare

Acum că ai pregătit tensorii pentru imagine și cutia de delimitare, poți desena cutia peste imagine și îi poți verifica vizual acuratețea.

torch, torchvision și torchvision.transforms au fost deja importate. Imaginea a fost transformată în tensori sub forma image_tensor. Coordonatele au fost atribuite variabilelor: x_min, y_min, x_max, y_max.

Acest exercițiu face parte din cursul

Deep Learning pentru imagini cu PyTorch

Vezi cursul

Instrucțiuni pentru exercițiu

Importă draw_bounding_boxes din torchvision.utils.
Definește cutia de delimitare bbox ca listă formată din x_min, y_min, x_max și y_max.
Transmite image_tensor și bbox_tensor funcției draw_bounding_boxes pentru a desena cutia peste imagine și atribuie rezultatul variabilei img_bbox.
Convertește tensorul img_bbox în imagine.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

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

Editează și rulează codul