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

Bài tập

Dự đoán bounding box

Bạn đã huấn luyện một mô hình nhận dạng đối tượng. Giờ bạn muốn dùng nó để tạo bounding box và phân loại cho một ảnh kiểm thử.

Mô hình có sẵn dưới tên model và đã ở chế độ đánh giá (evaluation). test_image cũng đã sẵn sàng, và torch cùng torch.nn với bí danh nn đã được import.

Hướng dẫn

100 XP
  • Lấy đầu ra của mô hình cho test_image và gán vào output.
  • Trích xuất các bounding box dự đoán từ đầu ra và gán vào boxes.
  • Trích xuất các điểm số (scores) dự đoán từ đầu ra và gán vào scores.