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.

Toto cvičení je součástí kurzu

Deep Learning pro obrázky s PyTorchem

Zobrazit kurz

Pokyny k cvičení

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.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

# Import resnet18 model
from torchvision.models import ____

# Initialize model with default weights
weights = ResNet18_Weights.DEFAULT
model = ____

# Set model to evaluation mode
model.____

# Initialize the transforms
transform = ____

Upravit a spustit kód