เริ่มต้นใช้งานเริ่มต้นใช้งานได้ฟรี

การสร้าง Binary Mask

รูปภาพที่ใช้ในงาน Segmentation มักถูกระบุคำอธิบายด้วย Mask ระดับพิกเซล ลองพิจารณารูปภาพของแมว Egyptian Mau ด้านล่างนี้

cat image

ในแบบฝึกหัดนี้และแบบฝึกหัดถัดไป จะใช้ Mask ที่สอดคล้องกันเพื่อแยกแมวออกจากรูปภาพ ขั้นแรก ต้องโหลด Mask แล้วแปลงให้เป็น Binary

Image จาก PIL, transforms จาก torchvision และ torch ได้ถูก import ไว้ให้แล้ว

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

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

ดูคอร์ส

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

  • โหลดรูปภาพ Mask ที่เก็บอยู่ใน annotations/Egyptian_Mau_123.png แล้วกำหนดให้กับตัวแปร mask
  • สร้าง binary_mask จาก mask_tensor โดยให้พิกเซลที่มีค่าเท่ากับ 1/255 ถูกกำหนดค่า tensor เป็น 1.0 และพิกเซลที่เหลือถูกกำหนดค่า tensor เป็น 0.0

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

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

# Load mask image
mask = ____

# Transform mask to tensor
transform = transforms.Compose([transforms.ToTensor()])
mask_tensor = transform(mask)

# Create binary mask
binary_mask = ____(
    ____, 
    ____,
    ____,
)

# Print unique mask values
print(binary_mask.unique())
แก้ไขและรันโค้ด