建立二元遮罩
分割任務的影像通常會以像素層級的遮罩來標註。請看這張埃及貓(Egyptian Mau)的照片。

在這題與下一題中,你會使用對應的遮罩,把貓從影像中分割出來。首先,你需要載入遮罩並將其二值化。
PIL 的 Image、torchvision 的 transforms,以及 torch 都已為你匯入。
本練習屬於課程
使用 PyTorch 進行影像深度學習
練習說明
- 載入儲存在
annotations/Egyptian_Mau_123.png的遮罩影像,並指定給mask。 - 從
mask_tensor建立binary_mask:將每個等於1/255的像素指定為張量值1.0,其餘像素指定為張量值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())