Predicción de bounding boxes
Has entrenado un modelo de reconocimiento de objetos. Ahora quieres usarlo para generar bounding boxes y clasificaciones para una imagen de prueba.
El modelo está disponible como model y ya está en modo de evaluación. La test_image también está disponible, y torch y torch.nn como nn ya se han importado.
Este ejercicio forma parte del curso
Deep Learning para imágenes con PyTorch
Instrucciones del ejercicio
- Obtén la salida del modelo para
test_imagey asígnala aoutput. - Extrae las bounding boxes predichas de la salida y asígnalas a
boxes. - Extrae las puntuaciones (scores) predichas de la salida y asígnalas a
scores.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# Get model's prediction
with torch.no_grad():
output = ____
# Extract boxes from the output
boxes = ____
# Extract scores from the output
scores = ____
print(boxes, scores)