1. Nauka
    2. /
  2. Kursy
    3. /
  3. Głębokie uczenie dla obrazów z PyTorch
Connected

ćwiczenie

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.

Instrukcje

100 XP
  • 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.