Membuat mask biner

Gambar untuk tugas segmentasi biasanya diberi anotasi dengan mask pada tingkat piksel. Pertimbangkan gambar kucing Egyptian Mau berikut.

cat image

Dalam latihan ini dan berikutnya, Anda akan menggunakan mask terkait untuk mengekstraksi kucing dari gambar. Pertama, Anda perlu memuat mask tersebut dan melakukan binarisasi.

Image dari PIL, transforms dari torchvision, dan torch sudah diimpor untuk Anda.

Latihan ini merupakan bagian dari kursus

Deep Learning untuk Gambar dengan PyTorch

Lihat Kursus

Instruksi latihan

Muat gambar mask yang disimpan di annotations/Egyptian_Mau_123.png dan tetapkan ke mask.
Buat binary_mask dari mask_tensor di mana setiap piksel yang sama dengan 1/255 diberi nilai tensor 1.0, dan piksel lainnya diberi nilai tensor 0.0.

Latihan interaktif langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

# 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())

Edit dan Jalankan Kode