CommencezCommencez gratuitement

Analyser la sortie du modèle

Après avoir obtenu la prédiction d'un modèle préentraîné, vous pouvez maintenant l'analyser en détail. prediction est à votre disposition.

Vous verrez qu'il s'agit d'une liste de longueur un (vous avez fait une prédiction pour une seule image) contenant un dictionnaire avec quatre clés. labels sont des entiers indiquant les classes prédites, et scores sont les probabilités correspondantes pour ces classes. Vous pouvez accéder à la liste des noms de classes correspondant à labels via class_names.

Quelles sont les catégories prédites et leurs scores associés pour les deux meilleurs masques?

Cette activité fait partie du cours

Deep Learning pour les images avec PyTorch

Voir le cours

Exercice interactif pratique

Passez de la théorie à l’action grâce à l’un de nos exercices interactifs

Commencer l’exercice