開始使用免費開始

建立二元遮罩

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

cat image

在這題與下一題中,你會使用對應的遮罩,把貓從影像中分割出來。首先,你需要載入遮罩並將其二值化。

PILImagetorchvisiontransforms,以及 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())
編輯並執行程式碼