Cargar un modelo preentrenado

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

Este ejercicio forma parte del curso

Deep Learning para imágenes con PyTorch

Ver curso

Instrucciones del ejercicio

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

Ejercicio interactivo práctico

Prueba este ejercicio y completa 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