Caricare un modello pre-addestrato

Stai creando un'applicazione per etichettare immagini provenienti dai social media. Questo compito richiede alta accuratezza e velocità. Userai un modello ResNet18 pre-addestrato per inferire le classi delle immagini.

Questo esercizio fa parte del corso

Deep Learning per Immagini con PyTorch

Visualizza il corso

Istruzioni dell'esercizio

Importa resnet18 e ResNet18_Weights da torchvision.models.
Istanzia il model usando resnet18(), impostando il parametro dei pesi a weights.
Imposta model in modalità di valutazione.
Inizializza le trasformazioni di input e assegnale a transform.

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

# 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 = ____

Modifica ed esegui il codice