Carga de un modelo preentrenado

Estás creando una aplicación para etiquetar imágenes de las redes sociales. Esta tarea requiere gran precisión y rapidez. Vas a utilizar un modelo ResNet18 preentrenado para inferir clases de imágenes.

Este ejercicio forma parte del curso

Aprendizaje profundo para imágenes con PyTorch

Ver curso

Instrucciones del ejercicio

Importa resnet18 y ResNet18_Weights de torchvision.models.
Instancia el model utilizando resnet18() y estableciendo el parámetro de pesos en weights.
Configure model en modo de evaluación.
Inicializa las transformaciones de entrada y asígnalas a un transform.

Ejercicio interactivo práctico

Prueba este ejercicio completando el código de muestra.

# 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 y ejecutar código