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

cvičení

Klasifikace obrázků pomocí ResNet

Model jsi sestavil/a z předtrénovaného ResNet18. Teď je čas otestovat ho na ukázkovém obrázku.

Na obrázek aplikuješ předzpracovací transformace a klasifikuješ ho. Budeš potřebovat vrstvu softmax() následovanou argmax(), protože ResNet18 byl natrénován na vícetrídní datové sadě.

Pro testování predikcí jsi vybral/a tento obrázek: A cup of espresso

Předzpracovací transformace je uložena jako preprocess. PIL obrázek je nahrán jako img.

Pokyny

100 XP
  • Aplikuj předzpracovací transformace na obrázek a uprav jeho tvar pomocí .unsqueeze(0), aby se přidala dávková dimenze.
  • Předej obrázek modelem, uprav tvar výstupu pomocí .squeeze(0) pro odstranění dávkové dimenze a přidej vrstvu softmax().
  • Použij argmax() pro výběr třídy s nejvyšší pravděpodobností.