Rysowanie ramki ograniczającej

Mając już przygotowane tensory obrazu i ramki ograniczającej, możesz narysować ramkę na obrazie i wizualnie sprawdzić jej poprawność.

torch, torchvision i torchvision.transforms zostały zaimportowane. Obraz został już przekształcony do postaci tensorów jako image_tensor. Współrzędne zostały przypisane do zmiennych: x_min, y_min, x_max, y_max.

To ćwiczenie jest częścią kursu

Głębokie uczenie dla obrazów z PyTorch

Zobacz kurs

Instrukcje do ćwiczenia

Zaimportuj draw_bounding_boxes z torchvision.utils.
Zdefiniuj ramkę ograniczającą bbox jako listę złożoną z wartości x_min, y_min, x_max i y_max.
Przekaż image_tensor i bbox_tensor do funkcji draw_bounding_boxes, aby narysować ramkę na obrazie, i przypisz wynik do zmiennej img_bbox.
Przekonwertuj tensor img_bbox na obraz.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

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

Edytuj i uruchom kod