1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Deep Learning pro obrázky s PyTorchem
Connected

cvičení

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.

Pokyny

100 XP
  • Získej výstup modelu pro test_image a 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.