Creación de máscaras binarias

Las imágenes para tareas de segmentación suelen anotarse con máscaras a nivel de píxel. Observa esta imagen de un gato Egyptian Mau.

cat image

En este y el siguiente ejercicio, usarás la máscara correspondiente para segmentar el gato de la imagen. Primero, necesitarás cargar la máscara y binarizarla.

Image de PIL, transforms de torchvision y torch ya se han importado por ti.

Este ejercicio forma parte del curso

Deep Learning para imágenes con PyTorch

Ver curso

Instrucciones del ejercicio

Carga la imagen de la máscara almacenada en annotations/Egyptian_Mau_123.png y asígnala a mask.
Crea una binary_mask a partir de mask_tensor donde cada píxel igual a 1/255 reciba un valor tensor de 1.0, y el resto de píxeles reciban un valor tensor de 0.0.

Ejercicio interactivo práctico

Prueba este ejercicio y completa el código de muestra.

# 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 y ejecutar código