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?
Latihan ini adalah bagian dari kursus
Deep Learning for Images with PyTorch
Latihan interaktif praktis
Ubah teori menjadi tindakan dengan salah satu latihan interaktif kami.
Mulai berolahraga