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?
This exercise is part of the course
Deep Learning for Images with PyTorch
Hands-on interactive exercise
Turn theory into action with one of our interactive exercises
