1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Deep Learning cho Ảnh với PyTorch
Connected

Bài tập

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.

Hướng dẫn

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