Criando máscaras binárias

As imagens para tarefas de segmentação geralmente são anotadas com máscaras em nível de pixel. Olha só essa foto de um gato Mau egípcio.

imagem de gato

Neste exercício e no próximo, você vai usar a máscara certa para separar o gato da imagem. Primeiro, você precisa carregar a máscara e binarizá-la.

Image de PIL, transforms de torchvision e torch já foram importados pra você.

Este exercício faz parte do curso

Aprendizado profundo para imagens com PyTorch

Ver curso

Instruções do exercício

Carregue a imagem da máscara que tá em annotations/Egyptian_Mau_123.png e coloque ela em mask.
Crie um objeto “ binary_mask ” a partir de “ mask_tensor ”, onde cada pixel igual a “ 1/255 ” recebe um valor tensor de “ 1.0 ”, e os pixels restantes recebem um valor tensor de “ 0.0 ”.

Exercício interativo prático

Experimente este exercício completando este código de exemplo.

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

Editar e executar o código