Загрузка предобученной модели

Вы разрабатываете приложение для разметки изображений из социальных сетей. Задача требует высокой точности и скорости. Для определения классов изображений вы будете использовать предобученную модель ResNet18.

Это упражнение является частью курса

Глубокое обучение для работы с изображениями на PyTorch

Посмотреть курс

Инструкции к упражнению

Импортируйте resnet18 и ResNet18_Weights из torchvision.models.
Создайте экземпляр model с помощью resnet18(), передав в параметр весов значение weights.
Переведите model в режим оценки.
Инициализируйте входные преобразования и присвойте их переменной transform.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

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

Редактировать и запускать код