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
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
)