Vorhersage von Begrenzungsrahmen
Du hast ein Modell für die Objekterkennung trainiert. Jetzt willst du damit Begrenzungsrahmen und Klassifizierungen für ein Testbild erstellen.
Das Modell ist als „ model ” verfügbar und kann schon getestet werden. Der „ test_image ” ist auch verfügbar, und „ torch ” und „ torch.nn ” wurden als „ nn ” importiert.
Diese Übung ist Teil des Kurses
Deep Learning für Bilder mit PyTorch
Anleitung zur Übung
- Hol die Ausgabe des Modells für „
test_image“ und schick sie an „output“. - Extrahier die vorhergesagten Begrenzungsrahmen aus der Ausgabe und ordne sie „
boxes“ zu. - Hol die vorhergesagten Werte aus der Ausgabe raus und speicher sie in „
scores“.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
# Get model's prediction
with torch.no_grad():
output = ____
# Extract boxes from the output
boxes = ____
# Extract scores from the output
scores = ____
print(boxes, scores)