ComenzarEmpieza gratis

Aumento de datos en PyTorch

Incluyamos el aumento de datos en tu conjunto de datos e inspeccionemos visualmente algunas imágenes para asegurarnos de que se aplican las transformaciones deseadas.

En primer lugar, añadirás las transformaciones de aumento a train_transforms. Utilicemos un giro horizontal aleatorio y una rotación con un ángulo aleatorio entre 0 y 45 grados. El código que sigue para crear el conjunto de datos y el DataLoader es exactamente el mismo que antes. Por último, remodelarás la imagen y la mostrarás para ver si las nuevas transformaciones de aumento son visibles.

Se han convocado para ti todas las importaciones que necesitas:

from torchvision.datasets import ImageFolder
from torch.utils.data import DataLoader
from torchvision import transforms
import matplotlib.pyplot as plt

Es hora de aumentar algunas fotos de nubes.

Este ejercicio forma parte del curso

Aprendizaje profundo intermedio con PyTorch

Ver curso

Instrucciones del ejercicio

  • Añade dos transformaciones más a train_transforms para realizar un volteo horizontal aleatorio y luego una rotación con un ángulo aleatorio entre 0 y 45 grados.
  • Remodela el tensor image del DataLoader para que sea adecuado para su visualización.
  • Muestra la imagen.

Ejercicio interactivo práctico

Prueba este ejercicio completando el código de muestra.

train_transforms = transforms.Compose([
    # Add horizontal flip and rotation
    ____,
    ____,
    transforms.ToTensor(),
    transforms.Resize((128, 128)),
])

dataset_train = ImageFolder(
  "clouds_train",
  transform=train_transforms,
)

dataloader_train = DataLoader(
  dataset_train, shuffle=True, batch_size=1
)

image, label = next(iter(dataloader_train))
# Reshape the image tensor
image = image.____.____(____, ____, ____) 
# Display the image
____
plt.show()
Editar y ejecutar código