Disegnare una bounding box

Dopo aver preparato i tensori dell'immagine e della bounding box, ora puoi disegnare il riquadro sull'immagine e verificarne visivamente l'accuratezza.

Sono stati importati torch, torchvision, torchvision.transforms. L'immagine è già stata trasformata in tensori come image_tensor. Le coordinate sono state assegnate alle variabili: x_min, y_min, x_max, y_max.

Questo esercizio fa parte del corso

Deep Learning per Immagini con PyTorch

Visualizza il corso

Istruzioni dell'esercizio

Importa draw_bounding_boxes da torchvision.utils.
Definisci la bounding box bbox come lista composta da x_min, y_min, x_max e y_max.
Passa image_tensor e bbox_tensor a draw_bounding_boxes per disegnare il riquadro sull'immagine e assegna l'output a img_bbox.
Converte il tensore img_bbox in immagine.

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

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

Modifica ed esegui il codice