Läsa in en förtränad modell

Du håller på att bygga ett program för att märka bilder från sociala medier. Uppgiften kräver hög precision och snabb bearbetning. Du ska använda en förtränad ResNet18-modell för att klassificera bilder.

Den här övningen är en del av kursen

Djupinlärning för bilder med PyTorch

Visa kurs

Övningsinstruktioner

Importera resnet18 och ResNet18_Weights från torchvision.models.
Instansiera model med resnet18() och sätt parametern weights till weights.
Ställ in model i utvärderingsläge.
Initiera inmatningstransformationerna och tilldela dem till transform.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

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

Redigera och kör kod