Segmentando uma imagem com uma máscara
Com a máscara binária pronta, você pode usá-la para separar o objeto, que é o gato, da imagem.
Para fazer isso, você vai precisar carregar a imagem original e transformá-la em um tensor. Depois, você vai criar o tensor do objeto mascarando a imagem original. Por fim, você vai ver o resultado.
transforms de torchvision foram importados, e o arquivo binary_mask que você criou no exercício anterior está disponível para você.
Este exercício faz parte do curso
Aprendizado profundo para imagens com PyTorch
Instruções do exercício
- Crie um
image_tensor, aplicando a transformaçãoToTensor(), definida comotransform, à imagem bruta. - Segmente a imagem mascarando o
image_tensorcom obinary_maske colocando o resultado emobject_tensor. - Aplique a transformação já definida
to_pil_imageaoobject_tensorpara 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()