Vẽ một khung giới hạn

Sau khi đã chuẩn bị các tensor ảnh và khung giới hạn, bạn có thể vẽ khung lên trên ảnh và quan sát trực quan độ chính xác của nó.

torch, torchvision,torchvision.transforms đã được import. Ảnh đã được chuyển thành tensor là image_tensor. Các tọa độ đã được gán cho các biến: x_min, y_min, x_max, y_max.

Bài tập này là một phần của khóa học

Deep Learning cho Ảnh với PyTorch

Xem khóa học

Hướng dẫn bài tập

Import draw_bounding_boxes từ torchvision.utils.
Định nghĩa khung giới hạn bbox là một list gồm x_min, y_min, x_max, và y_max.
Truyền image_tensor và bbox_tensor vào draw_bounding_boxes để vẽ khung lên ảnh và gán đầu ra cho img_bbox.
Chuyển tensor img_bbox thành ảnh.

Bài tập tương tác thực hành trực tiếp

Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.

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

Chỉnh sửa và Chạy Mã