การพยากรณ์ Bounding Box

คุณได้ฝึกโมเดลสำหรับการจดจำวัตถุเรียบร้อยแล้ว ขั้นตอนต่อไปคือนำโมเดลนี้ไปสร้าง bounding box และผลการจำแนกประเภทสำหรับภาพทดสอบ

โมเดลพร้อมใช้งานในตัวแปร model และอยู่ในโหมดประเมินผลแล้ว test_image ก็พร้อมใช้งานเช่นกัน และได้ import torch และ torch.nn ในชื่อ nn ไว้แล้ว

แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร

Deep Learning สำหรับภาพด้วย PyTorch

ดูคอร์ส

คำแนะนำการฝึกหัด

รับผลลัพธ์ของโมเดลสำหรับ test_image และกำหนดให้กับตัวแปร output
แยก bounding box ที่พยากรณ์ได้จาก output และกำหนดให้กับตัวแปร boxes
แยกค่า score ที่พยากรณ์ได้จาก output และกำหนดให้กับตัวแปร scores

แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ

ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์

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

# Extract boxes from the output
boxes = ____

# Extract scores from the output
scores = ____

print(boxes, scores)

แก้ไขและรันโค้ด