Vorhersage von Bounding Boxes

Du hast ein Objekterkennungsmodell trainiert. Jetzt möchtest du es nutzen, um für ein Testbild Bounding Boxes und Klassifikationen zu erzeugen.

Das Modell ist als model verfügbar und befindet sich bereits im Evaluierungsmodus. Das test_image ist ebenfalls vorhanden, und torch sowie torch.nn als nn wurden importiert.

Diese Übung ist Teil des Kurses

Deep Learning für Bilder mit PyTorch

Kurs anzeigen

Anleitung zur Übung

Hole die Ausgabe des Modells für das test_image und weise sie output zu.
Extrahiere die vorhergesagten Bounding Boxes aus der Ausgabe und weise sie boxes zu.
Extrahiere die vorhergesagten Scores aus der Ausgabe und weise sie scores zu.

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)

Code bearbeiten und ausführen