1. Learn
    2. /
  2. Courses
    3. /
  3. Deep Learning pour les images avec PyTorch
Connected

Exercise

Charger un modèle préentraîné

Vous développez une application pour étiqueter des images provenant des médias sociaux. Cette tâche exige une grande précision et rapidité. Vous allez utiliser un modèle ResNet18 préentraîné pour inférer les classes d'images.

Instructions

100 XP
  • Importez resnet18 et ResNet18_Weights de torchvision.models.
  • Instanciez le model en utilisant resnet18(), en définissant le paramètre weights à weights.
  • Placez model en mode évaluation.
  • Initialisez les transformations d'entrée et assignez-les à transform.