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.

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

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.

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.

# Get model's prediction
with torch.no_grad():
    output = ____

# Extract boxes from the output
boxes = ____

# Extract scores from the output
scores = ____

print(boxes, scores)

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