Aan de slagGa gratis aan de slag

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?

Deze oefening maakt deel uit van de cursus

Deep Learning for Images with PyTorch

Cursus bekijken

Praktische interactieve oefening

Zet theorie om in actie met een van onze interactieve oefeningen.

Begin met trainen