Prédiction des boîtes englobantes
Vous avez entraîné un modèle de reconnaissance d’objets. Vous souhaitez maintenant l’utiliser pour générer des boîtes englobantes et des classifications pour une image de test.
Le modèle est disponible sous le nom model et il est déjà en mode évaluation. L’test_image est également disponible, et torch ainsi que torch.nn sous nn ont été importés.
Cet exercice fait partie du cours
Deep Learning pour l’image avec PyTorch
Instructions
- Récupérez la sortie du modèle pour
test_imageet affectez-la àoutput. - Extrayez les boîtes englobantes prédites de la sortie et affectez-les à
boxes. - Extrayez les scores prédits de la sortie et affectez-les à
scores.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Get model's prediction
with torch.no_grad():
output = ____
# Extract boxes from the output
boxes = ____
# Extract scores from the output
scores = ____
print(boxes, scores)