ComenzarEmpieza gratis

Conjunto de datos con aumentos

Ya has construido el conjunto de datos de imágenes de nubes y el modelo convolucional para clasificar los distintos tipos de nubes. Antes de entrenarlo, vamos a adaptar el conjunto de datos mediante la adición de aumentos que podrían mejorar el rendimiento de clasificación de nubes del modelo.

El código para configurar el conjunto de datos y el DataLoader ya está preparado y debería resultarte familiar. Tu tarea consiste en definir la composición de las transformaciones que se aplicarán a las imágenes de entrada a medida que se carguen.

Ten en cuenta que antes redimensionabas las imágenes a 128 por 128 para que se vieran bien, pero ahora las utilizarás más pequeñas para acelerar el entrenamiento. Como verás más adelante, 64 por 64 será lo suficientemente grande para que el modelo aprenda.

from torchvision import transforms ya se ha ejecutado para ti, así que ¡manos a la obra!

Este ejercicio forma parte del curso

Aprendizaje profundo intermedio con PyTorch

Ver curso

Instrucciones del ejercicio

  • Define train_transforms al componer juntas cinco transformaciones: un volteo horizontal aleatorio, una rotación aleatoria (por ángulo de 0 a 45 grados), un ajuste automático aleatorio del contraste, un análisis sintáctico a tensor y un cambio de tamaño a 64 por 64 píxeles.

Ejercicio interactivo práctico

Prueba este ejercicio completando el código de muestra.

# Define transforms
train_transforms = transforms.Compose([
    ____,
    ____,
    ____,
    ____,
    ____,
])

dataset_train = ImageFolder(
  "clouds_train",
  transform=train_transforms,
)
dataloader_train = DataLoader(
  dataset_train, shuffle=True, batch_size=16
)
Editar y ejecutar código