Predikce ohraničujících rámečků
Natrénoval/a jsi model pro rozpoznávání objektů. Teď ho chceš použít k vygenerování ohraničujících rámečků a klasifikací pro testovací snímek.
Model je dostupný jako model a je už v režimu vyhodnocování. K dispozici máš také test_image a importovaný torch a torch.nn jako nn.
Toto cvičení je součástí kurzu
Deep Learning pro obrázky s PyTorchem
Pokyny k cvičení
- Získej výstup modelu pro
test_imagea přiřaď ho do proměnnéoutput. - Extrahuj předpovězené ohraničující rámečky z výstupu a přiřaď je do proměnné
boxes. - Extrahuj předpovězená skóre z výstupu a přiřaď je do proměnné
scores.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
# Get model's prediction
with torch.no_grad():
output = ____
# Extract boxes from the output
boxes = ____
# Extract scores from the output
scores = ____
print(boxes, scores)