ComeçarComece de graça

Analisando a saída do modelo

Depois de obter a predição de um modelo pré-treinado, você pode analisá-la em detalhes. prediction está disponível para você.

Você verá que é uma lista de comprimento um (você fez a predição para apenas uma imagem) com um dicionário com quatro chaves. labels são inteiros que indicam as classes previstas, e scores são as probabilidades correspondentes. Você pode acessar a lista de nomes de classes correspondentes a labels como class_names.

Quais são as categorias previstas e suas pontuações associadas para as duas melhores máscaras?

Este exercício faz parte do curso

Deep Learning para Imagens com PyTorch

Ver curso

Exercício interativo prático

Transforme a teoria em ação com um de nossos exercícios interativos

Começar o exercício