ćwiczenie

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.

Instrukcje

100 XP

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.

.css-6su6fj{-webkit-flex-shrink:0;-ms-flex-negative:0;flex-shrink:0;}ćwiczenie

Instrukcje

ćwiczenie