Get startedGet started for free

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

View Course

Hands-on interactive exercise

Turn theory into action with one of our interactive exercises

Start Exercise