Comece agoraComece grátis

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 exercicio faz parte do curso

Deep Learning para Imagens com PyTorch

Ver curso

exercicio interativo prático

Transforme teoria em prática com um dos nossos exercicio interativos

Iniciar exercicio