CommencerCommencer gratuitement

Jeu de données avec augmentations

Vous avez déjà construit le jeu de données d’images à partir de photos de nuages et le modèle convolutionnel pour classer différents types de nuages. Avant de l’entraîner, adaptons le jeu de données en ajoutant des augmentations susceptibles d’améliorer les performances de classification du modèle.

Le code pour configurer le Dataset et le DataLoader est déjà prêt pour vous et devrait vous être familier. Votre tâche consiste à définir la composition des transformations qui seront appliquées aux images d’entrée lors de leur chargement.

Remarquez qu’auparavant vous redimensionniez les images en 128 × 128 pour un affichage soigné, mais vous utiliserez maintenant des images plus petites pour accélérer l’entraînement. Comme vous le verrez, 64 × 64 sera suffisant pour que le modèle apprenne.

from torchvision import transforms a déjà été exécuté pour vous, alors allons-y !

Cet exercice fait partie du cours

Deep learning intermédiaire avec PyTorch

Afficher le cours

Instructions

  • Définissez train_transforms en composant cinq transformations : une symétrie horizontale aléatoire, une rotation aléatoire (d’un angle entre 0 et 45 degrés), un ajustement automatique du contraste aléatoire, la conversion en tenseur, puis un redimensionnement en 64 × 64 pixels.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

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

dataset_train = ImageFolder(
  "clouds_train",
  transform=train_transforms,
)
dataloader_train = DataLoader(
  dataset_train, shuffle=True, batch_size=16
)
Modifier et exécuter le code