Een voorgetraind model laden

Je bouwt een applicatie om afbeeldingen van sociale media te labelen. Deze taak vraagt om hoge nauwkeurigheid en snelheid. Je gaat een voorgetraind ResNet18-model gebruiken om afbeeldingsklassen te voorspellen.

Deze oefening maakt deel uit van de cursus

Deep Learning voor afbeeldingen met PyTorch

Cursus bekijken

Oefeninstructies

Importeer resnet18 en ResNet18_Weights uit torchvision.models.
Maak het model aan met resnet18(), en zet de weights-parameter op weights.
Zet model in de evaluatiemodus.
Initialiseer de inputtransforms en wijs ze toe aan transform.

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

# 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 = ____

Code bewerken en uitvoeren