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 adalah bagian dari kursus

Deep Learning untuk Gambar dengan PyTorch

Lihat Kursus

Petunjuk 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 praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# 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