Kreslení ohraničujícího rámečku

Máš připravené tensory obrázku i ohraničujícího rámečku – teď můžeš rámeček nakreslit přímo přes obrázek a vizuálně ověřit jeho přesnost.

torch, torchvision a torchvision.transforms jsou již naimportované. Obrázek byl převeden na tensor jako image_tensor. Souřadnice jsou uloženy v proměnných: x_min, y_min, x_max, y_max.

Toto cvičení je součástí kurzu

Deep Learning pro obrázky s PyTorchem

Zobrazit kurz

Pokyny k cvičení

Importuj draw_bounding_boxes z torchvision.utils.
Definuj ohraničující rámeček bbox jako seznam obsahující x_min, y_min, x_max a y_max.
Předej image_tensor a bbox_tensor funkci draw_bounding_boxes, aby se rámeček nakreslil přes obrázek, a výsledek ulož do img_bbox.
Převeď tensor img_bbox na obrázek.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

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

Upravit a spustit kód