1. Learn
  2. /
  3. Courses
  4. /
  5. Deep Learning for Images with PyTorch

Connected

Exercise

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?

Instructions

50 XP

Possible answers