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

Bài tập

Tính NMS

Sau khi trích xuất các hộp giới hạn (bounding box) và điểm số dự đoán từ mô hình nhận dạng đối tượng, nhiệm vụ tiếp theo của bạn là đảm bảo chỉ giữ lại các hộp dự đoán chính xác nhất và không chồng lấp bằng kỹ thuật non-max suppression.

boxes và scores bạn đã tạo ở bài trước có sẵn trong không gian làm việc của bạn, và torch cùng torchvision đã được import.

Hướng dẫn

100 XP
  • Import nms từ torchvision.ops.
  • Đặt ngưỡng IoU bằng 0.5.
  • Áp dụng non-max suppression bằng cách truyền boxes, confidence_scores và iou_threshold vào hàm tương ứng.
  • Dùng các chỉ số đầu ra để lọc các hộp dự đoán.