Segmentando a imagem com uma máscara

Com a máscara binária pronta, você pode usá-la para segmentar o objeto — no caso, o gato — da imagem.

Para isso, você vai carregar a imagem original e transformá-la em um tensor. Em seguida, vai criar o tensor do objeto aplicando a máscara à imagem original. Por fim, vai exibir o resultado.

transforms de torchvision já foram importados, e o binary_mask que você criou no exercício anterior está disponível.

Este exercício faz parte do curso

Deep Learning para Imagens com PyTorch

Ver curso

Instruções do exercício

Crie image_tensor aplicando a transformação ToTensor(), definida como transform, à imagem bruta.
Segmente a imagem mascarando o image_tensor com o binary_mask, atribuindo o resultado a object_tensor.
Aplique a transformação to_pil_image, já definida, ao object_tensor para exibi-lo.

Exercício interativo prático

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

# Load image and transform to tensor
image = Image.open("images/Egyptian_Mau_123.jpg")
transform = transforms.Compose([transforms.ToTensor()])
image_tensor = ____

# Segment object out of the image
object_tensor = ____

# Convert segmented object to image and display
to_pil_image = ____
object_image = to_pil_image(object_tensor)
plt.imshow(object_image)
plt.show()

Editar e executar o código