Analyzing model output
Having obtained the prediction from a pre-trained model, you can now analyze it in detail. prediction is available to you.
You will find it is a list of length one (you made prediction for just one image) with a dictionary with four keys. labels are integers indicating predicted classes, and scores are corresponding class probabilities. You can access the list of the class names corresponding to labels as class_names.
What are the predicted categories and their associated scores for the top two masks?
Questo esercizio fa parte del corso
Deep Learning for Images with PyTorch
Esercizio pratico interattivo
Passa dalla teoria alla pratica con uno dei nostri esercizi interattivi
Inizia esercizio