Chargement d'un modèle pré-entraîné
Vous développez une application destinée à étiqueter des images provenant des réseaux sociaux. Cette tâche exige une grande précision et une grande rapidité. Vous allez utiliser un modèle ResNet18 pré-entraîné pour inférer des classes d'images.
Cet exercice fait partie du cours
Deep learning pour les images avec PyTorch
Instructions
- Veuillez importer
resnet18etResNet18_Weightsdepuistorchvision.models. - Instancier l'
modelen utilisantresnet18(), en définissant le paramètre weights surweights. - Veuillez régler l'
modelen mode évaluation. - Initialisez les transformations d'entrée et assignez-les à l'
transform.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# 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 = ____