Carregando um modelo pré-treinado

Você está criando um aplicativo para rotular imagens de redes sociais. Essa tarefa exige alta precisão e rapidez. Você vai usar um modelo ResNet18 pré-treinado para inferir as classes das imagens.

Este exercício faz parte do curso

Deep Learning para Imagens com PyTorch

Ver curso

Instruções do exercício

Importe resnet18 e ResNet18_Weights de torchvision.models.
Instancie o model usando resnet18(), definindo o parâmetro de pesos como weights.
Coloque o model em modo de avaliação.
Inicialize as transformações de entrada e atribua-as a transform.

Exercício interativo prático

Experimente este exercício completando este código de exemplo.

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

Editar e executar o código