CommencerCommencer 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 1 (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. 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 ?

Cet exercice fait partie du cours

Deep Learning pour l’image avec PyTorch

Afficher le cours

Exercice interactif pratique

Passez de la théorie à la pratique avec l’un de nos exercices interactifs

Commencer l’exercice