Carregando um modelo pré-treinado

Você tá criando um aplicativo pra rotular imagens das redes sociais. Essa tarefa precisa de muita precisão e rapidez. Você vai usar um modelo ResNet18 pré-treinado pra inferir classes de imagens.

Este exercício faz parte do curso

Aprendizado profundo para imagens com PyTorch

Ver curso

Instruções do exercício

Importa resnet18 e ResNet18_Weights de torchvision.models.
Instanciar o model usando resnet18(), definindo o parâmetro weights como weights.
Coloca o model no modo de avaliação.
Inicialize as transformações de entrada e atribua-as a um 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