Rita en begränsningsbox

Nu när bild- och begränsningsboxens tensorer är förberedda kan du rita boxen ovanpå bilden och visuellt kontrollera dess noggrannhet.

torch, torchvision och torchvision.transforms har importerats. Bilden har redan omvandlats till tensorer som image_tensor. Koordinaterna har tilldelats variablerna: x_min, y_min, x_max, y_max.

Den här övningen är en del av kursen

Djupinlärning för bilder med PyTorch

Visa kurs

Övningsinstruktioner

Importera draw_bounding_boxes från torchvision.utils.
Definiera begränsningsboxen bbox som en lista bestående av x_min, y_min, x_max och y_max.
Skicka image_tensor och bbox_tensor till draw_bounding_boxes för att rita boxen ovanpå bilden och tilldela resultatet till img_bbox.
Konvertera tensorn img_bbox till en bild.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

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

Redigera och kör kod