Segmentación de imágenes con una máscara
Con la máscara binaria lista, puedes usarla para segmentar el objeto, es decir, el gato, de la imagen.
Para ello, deberás cargar la imagen original y transformarla en un tensor. A continuación, crearás el tensor de objetos enmascarando la imagen original. Por último, mostrarás el resultado.
transforms desde torchvision se han importado, y el archivo binary_mask que creaste en el ejercicio anterior está disponible.
Este ejercicio forma parte del curso
Aprendizaje profundo para imágenes con PyTorch
Instrucciones del ejercicio
- Crea un canal de color llamado «
image_tensor» aplicando la transformación «ToTensor()», definida como «transform», a la imagen sin procesar. - Segmenta la imagen enmascarando el
image_tensorcon elbinary_masky asignando el resultado aobject_tensor. - Aplica la transformación ya definida «
to_pil_image» a «object_tensor» para mostrarla.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# 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()