Voorspelling van bounding boxes
Je hebt een objectherkenningsmodel getraind. Nu wil je het gebruiken om bounding boxes en classificaties voor een testafbeelding te genereren.
Het model is beschikbaar als model en staat al in evaluatiemodus. De test_image is ook beschikbaar, en torch en torch.nn als nn zijn geïmporteerd.
Deze oefening maakt deel uit van de cursus
Deep Learning voor afbeeldingen met PyTorch
Oefeninstructies
- Haal de output van het model op voor de
test_imageen ken deze toe aanoutput. - Haal de voorspelde bounding boxes uit de output en ken ze toe aan
boxes. - Haal de voorspelde scores uit de output en ken ze toe aan
scores.
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
# Get model's prediction
with torch.no_grad():
output = ____
# Extract boxes from the output
boxes = ____
# Extract scores from the output
scores = ____
print(boxes, scores)