1. Learn
    2. /
  2. Cursuri
    3. /
  3. Deep Learning pentru imagini cu PyTorch
Connected

exercițiu

Desenarea unei cutii de delimitare

Acum că ai pregătit tensorii pentru imagine și cutia de delimitare, poți desena cutia peste imagine și îi poți verifica vizual acuratețea.

torch, torchvision și torchvision.transforms au fost deja importate. Imaginea a fost transformată în tensori sub forma image_tensor. Coordonatele au fost atribuite variabilelor: x_min, y_min, x_max, y_max.

Instrucțiuni

100 XP
  • Importă draw_bounding_boxes din torchvision.utils.
  • Definește cutia de delimitare bbox ca listă formată din x_min, y_min, x_max și y_max.
  • Transmite image_tensor și bbox_tensor funcției draw_bounding_boxes pentru a desena cutia peste imagine și atribuie rezultatul variabilei img_bbox.
  • Convertește tensorul img_bbox în imagine.