Predicción de cuadros delimitadores
Has entrenado un modelo de reconocimiento de objetos. Ahora, quieres utilizar para generar cuadros delimitadores y clasificaciones para una imagen de prueba.
El modelo está disponible como « model » y ya se encuentra en modo de evaluación. También está disponible el paquete « test_image », y se han importado « torch » y « torch.nn » como « nn ».
Este ejercicio forma parte del curso
Aprendizaje profundo para imágenes con PyTorch
Instrucciones del ejercicio
- Obtén el resultado del modelo para
test_imagey asígnalo aoutput. - Extraiga los cuadros delimitadores previstos de la salida y asígnenlos a un
boxes. - Extraiga las puntuaciones previstas de la salida y asígnalas a un
scores.
Ejercicio interactivo práctico
Prueba este ejercicio completando 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)