Prédiction des boîtes englobantes
Vous avez formé un modèle de reconnaissance d'objets. À présent, vous souhaitez utiliser pour générer des cadres de délimitation et des classifications pour une image de test.
Le modèle est disponible à l'adresse model et est déjà en mode évaluation. L'test_image est également disponible, et torch et torch.nn ont été importés sous le nom nn.
Cet exercice fait partie du cours
Deep learning pour les images avec PyTorch
Instructions
- Obtenez la sortie du modèle pour l'
test_imageet attribuez-la àoutput. - Extrayez les cadres de délimitation prédits de la sortie et associez-les à l'
boxes. - Extrayez les scores prédits de la sortie et attribuez-les à l'
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)