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

Deep Learning für Bilder mit PyTorch

Kurs anzeigen

Anleitung zur Übung

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 Übung

Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.

# 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