การวาด Bounding Box

เมื่อเตรียม tensor ของภาพและ bounding box พร้อมแล้ว ขั้นตอนต่อไปคือวาดกรอบลงบนภาพเพื่อตรวจสอบความถูกต้องด้วยภาพ

torch, torchvision, torchvision.transforms ได้รับการนำเข้าแล้ว และภาพถูกแปลงเป็น tensor ในชื่อ image_tensor พร้อมทั้งกำหนดพิกัดให้กับตัวแปร x_min, y_min, x_max, y_max แล้ว

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

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

ดูคอร์ส

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

นำเข้า draw_bounding_boxes จาก torchvision.utils
กำหนด bounding box bbox เป็น list ที่ประกอบด้วย x_min, y_min, x_max และ y_max
ส่ง image_tensor และ bbox_tensor ไปยัง draw_bounding_boxes เพื่อวาดกรอบลงบนภาพ แล้วกำหนดผลลัพธ์ให้กับ img_bbox
แปลง tensor img_bbox เป็นภาพ

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

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

# Import draw_bounding_boxes
____

# Define the bounding box coordinates
bbox = ____
bbox_tensor = torch.tensor(bbox).unsqueeze(0)

# Implement draw_bounding_boxes
img_bbox = ____(____, ____, width=3, colors="red")

# Tranform tensors to image
transform = transforms.Compose([
    ____
])
plt.imshow(transform(img_bbox))
plt.show()

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