Predicția casetelor de delimitare

Ai antrenat un model de recunoaștere a obiectelor. Acum vrei să îl folosești pentru a genera casete de delimitare și clasificări pentru o imagine de test.

Modelul este disponibil sub numele model și este deja în modul de evaluare. test_image este de asemenea disponibilă, iar torch și torch.nn au fost importate ca nn.

Acest exercițiu face parte din cursul

Deep Learning pentru imagini cu PyTorch

Vezi cursul

Instrucțiuni pentru exercițiu

Obține rezultatul modelului pentru test_image și atribuie-l variabilei output.
Extrage casetele de delimitare prezise din rezultat și atribuie-le variabilei boxes.
Extrage scorurile prezise din rezultat și atribuie-le variabilei scores.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

# Get model's prediction
with torch.no_grad():
    output = ____

# Extract boxes from the output
boxes = ____

# Extract scores from the output
scores = ____

print(boxes, scores)

Editează și rulează codul