Ein vortrainiertes Modell laden
Du entwickelst eine App, um Bilder aus den sozialen Medien zu beschriften. Diese Aufgabe braucht echt Genauigkeit und Schnelligkeit. Du wirst ein vortrainiertes ResNet18-Modell verwenden, um Bildklassen zu ermitteln.
Diese Übung ist Teil des Kurses
Deep Learning für Bilder mit PyTorch
Anleitung zur Übung
- Importiere „
resnet18“ und „ResNet18_Weights“ aus „torchvision.models“. - Instanziere „
model” mit „resnet18()” und setz den Parameter „weights” auf „weights”. - Stell „
model“ auf den Auswertungsmodus. - Initialisiere die Eingabetransformationen und ordne 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 = ____