การแสดง soft mask

ในแบบฝึกหัดก่อนหน้า คุณได้เรียนรู้ว่าวัตถุ 2 อันดับแรกที่โมเดล Mask R-CNN แบ่งส่วนไว้นั้นเป็นแมวทั้งคู่ คราวนี้จะมาแสดง mask ของแมวทั้งสองซ้อนทับบนภาพต้นฉบับ เพื่อตรวจสอบความถูกต้องด้วยสายตา โดยจะวนซ้ำผ่าน mask ทั้งสอง แล้วพล็อตภาพต้นฉบับพร้อมกับ mask แบบกึ่งโปร่งใสซ้อนอยู่ด้านบน

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

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

ดูคอร์ส

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

ดึง mask และ label จาก prediction แล้วกำหนดให้กับตัวแปร masks และ labels ตามลำดับ
ภายใน for-loop ให้แสดง mask ลำดับที่ i ซ้อนทับบนภาพ โดยส่ง mask[i, 0] เข้าไปในฟังก์ชันพล็อต พร้อมกำหนด color map เป็น "jet" และตั้งค่าพารามิเตอร์ความโปร่งใสเป็น 0.5

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

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

# Extract masks and labels from prediction
masks = ____
labels = ____

# Plot image with two overlaid masks
for i in range(2):
    plt.imshow(image)
    # Overlay the i-th mask on top of the image
    plt.imshow(____, ____, ____)
    plt.title(f"Object: {class_names[labels[i]]}")
    plt.show()

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