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
Instructions
- Définissez
train_transformsen 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
)