Menggambar bounding box

Setelah menyiapkan tensor gambar dan bounding box, sekarang Anda dapat menggambar kotak di atas gambar dan memeriksa akurasinya secara visual.

torch, torchvision, torchvision.transforms telah diimpor. Gambar telah diubah menjadi tensor sebagai image_tensor. Koordinat telah ditetapkan ke variabel: x_min, y_min, x_max, y_max.

Latihan ini merupakan bagian dari kursus

Deep Learning untuk Gambar dengan PyTorch

Lihat Kursus

Instruksi latihan

Impor draw_bounding_boxes dari torchvision.utils.
Definisikan bounding box bbox sebagai list yang terdiri dari x_min, y_min, x_max, dan y_max.
Oper image_tensor dan bbox_tensor ke draw_bounding_boxes untuk menggambar kotak di atas gambar dan tetapkan keluarannya ke img_bbox.
Konversi tensor img_bbox menjadi gambar.

Latihan interaktif langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

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

Edit dan Jalankan Kode