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
exercicio interativo prático
Transforme teoria em prática com um dos nossos exercicio interativos
Iniciar exercicio