Exibindo máscaras suaves
No exercício anterior, você aprendeu que os dois objetos mais prováveis que o modelo Mask R-CNN segmentou são gatos. Agora, você vai ver as máscaras desses dois gatos sobrepostas na imagem original pra conferir se estão certas. Isso vai precisar repetir as duas máscaras e, pra cada uma delas, desenhar a imagem original e, depois, uma máscara meio transparente em cima dela.
Este exercício faz parte do curso
Aprendizado profundo para imagens com PyTorch
Instruções do exercício
- Pega as máscaras e rótulos do arquivo “
prediction”, e coloca no “masks” e “labels”, respectivamente. - Dentro do loop for, mostra a máscara
i-th sobre a imagem passandomask[i, 0]para a função de plotagem, usando o mapa de cores"jet"e definindo o parâmetro de transparência como0.5.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Extract masks and labels from prediction
masks = ____
labels = ____
# Plot image with two overlaid masks
for i in range(2):
plt.imshow(image)
# Overlay the i-th mask on top of the image
plt.imshow(____, ____, ____)
plt.title(f"Object: {class_names[labels[i]]}")
plt.show()