Predizione dei bounding box
Hai addestrato un modello di object recognition. Ora vuoi usarlo per generare bounding box e classificazioni per un'immagine di test.
Il modello è disponibile come model ed è già in modalità valutazione. Anche test_image è disponibile, e torch e torch.nn come nn sono stati importati.
Questo esercizio fa parte del corso
Deep Learning per Immagini con PyTorch
Istruzioni dell'esercizio
- Ottieni l'output del modello per
test_imagee assegnalo aoutput. - Estrai dall'output i bounding box previsti e assegnali a
boxes. - Estrai dall'output gli score previsti e assegnali a
scores.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# Get model's prediction
with torch.no_grad():
output = ____
# Extract boxes from the output
boxes = ____
# Extract scores from the output
scores = ____
print(boxes, scores)