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

<Kurs>Deep Learning für Bilder mit PyTorch</Kurs>

Kurs ansehen

Übungsanweisungen

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 praktische Übung

Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.

# 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