Jeu de données avec augmentations
Vous avez déjà construit le jeu de données d'images à partir de photos de nuages ainsi que 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 qui pourraient 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 sembler familier. Votre tâche est de définir la composition des transformations qui seront appliquées aux images d'entrée au moment de leur chargement.
Notez qu'auparavant vous redimensionniez les images en 128 par 128 pour un affichage soigné, mais maintenant vous utiliserez des images plus petites pour accélérer l'entraînement. Comme vous le verrez, 64 par 64 sera suffisant pour que le modèle apprenne.
from torchvision import transforms a déjà été exécuté pour vous, allons-y !
Cette activité fait partie du cours
Apprentissage profond intermédiaire avec PyTorch
Instructions de l’exercice
- Définissez
train_transformsen composant cinq transformations : un retournement horizontal aléatoire, une rotation aléatoire (d'un angle de 0 à 45 degrés), un ajustement automatique du contraste aléatoire, la conversion en tenseur, et un redimensionnement à 64 par 64 pixels.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Define transforms
train_transforms = transforms.Compose([
____,
____,
____,
____,
____,
])
dataset_train = ImageFolder(
"clouds_train",
transform=train_transforms,
)
dataloader_train = DataLoader(
dataset_train, shuffle=True, batch_size=16
)