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

cvičení

Načtení předtrénovaného modelu

Vytváříš aplikaci pro označování obrázků ze sociálních sítí. Tento úkol vyžaduje vysokou přesnost i rychlost. K určování tříd obrázků použiješ předtrénovaný model ResNet18.

Pokyny

100 XP
  • Importuj resnet18 a ResNet18_Weights z torchvision.models.
  • Vytvoř instanci model pomocí resnet18() a nastav parametr weights na weights.
  • Přepni model do režimu vyhodnocování.
  • Inicializuj vstupní transformace a přiřaď je do proměnné transform.