Analisando os resultados do modelo

Depois de pegar a previsão de um modelo pré-treinado, você pode analisar tudo direitinho. O “ prediction ” está à sua disposição.

Você vai ver que é uma lista com um item (você fez uma previsão para só uma imagem) com um dicionário com quatro chaves. “ labels ” são números inteiros que indicam as classes previstas, e “ scores ” são as probabilidades correspondentes das classes. Você pode acessar a lista de nomes de classes correspondentes a labels em class_names.

Quais são as categorias previstas e as pontuações associadas às duas máscaras mais bem classificadas?