Ein vortrainiertes Modell laden

Du baust eine Anwendung, um Bilder aus sozialen Medien zu beschriften. Diese Aufgabe erfordert hohe Genauigkeit und Geschwindigkeit. Du wirst ein vortrainiertes ResNet18-Modell verwenden, um Bildklassen zu inferieren.

Diese Übung ist Teil des Kurses

<Kurs>Deep Learning für Bilder mit PyTorch</Kurs>

Kurs ansehen

Übungsanweisungen

Importiere resnet18 und ResNet18_Weights aus torchvision.models.
Erzeuge das model mit resnet18() und setze den Parameter weights auf weights.
Setze model in den Evaluierungsmodus.
Initialisiere die Eingabetransforms und weise sie transform zu.

Interaktive praktische Übung

Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.

# 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 bearbeiten und ausführen