Wczytywanie wstępnie wytrenowanego modelu

Budujesz aplikację do etykietowania obrazów z mediów społecznościowych. Zadanie to wymaga wysokiej dokładności i szybkości działania. Użyjesz wstępnie wytrenowanego modelu ResNet18, aby klasyfikować obrazy.

To ćwiczenie jest częścią kursu

Głębokie uczenie dla obrazów z PyTorch

Zobacz kurs

Instrukcje do ćwiczenia

Zaimportuj resnet18 i ResNet18_Weights z torchvision.models.
Utwórz instancję model, używając resnet18() i ustawiając parametr wag na weights.
Przełącz model w tryb ewaluacji.
Zainicjalizuj przekształcenia wejściowe i przypisz je do zmiennej transform.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

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

Edytuj i uruchom kod