Prediktion av begränsningsrutor

Du har tränat en objektigenkänningsmodell. Nu vill du använda den för att generera begränsningsrutor och klassificeringar för en testbild.

Modellen är tillgänglig som model och är redan i utvärderingsläge. test_image är också tillgänglig, och torch samt torch.nn som nn har importerats.

Den här övningen är en del av kursen

Djupinlärning för bilder med PyTorch

Visa kurs

Övningsinstruktioner

Hämta modellens utdata för test_image och tilldela resultatet till output.
Extrahera de predikterade begränsningsrutorna från utdata och tilldela dem till boxes.
Extrahera de predikterade konfidenspoängen från utdata och tilldela dem till scores.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

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

# Extract boxes from the output
boxes = ____

# Extract scores from the output
scores = ____

print(boxes, scores)

Redigera och kör kod