Ensemble de données avec augmentations
Vous avez déjà construit l'ensemble de données d'images de nuages et le modèle convolutif pour classer les différents types de nuages. Avant de l'entraîner, adaptons le jeu de données en y ajoutant les augmentations susceptibles d'améliorer les performances du modèle en matière de classification des nuages.
Le code de configuration du Dataset et du DataLoader est déjà préparé pour vous et devrait vous sembler familier. Votre tâche consiste à définir la composition des transformations qui seront appliquées aux images d'entrée lors de leur chargement.
Notez qu'auparavant, vous redimensionniez les images à 128 par 128 pour les afficher correctement, mais que vous utiliserez désormais des images plus petites pour accélérer la formation. Comme vous le verrez plus loin, une taille de 64 par 64 est suffisante pour permettre au modèle d'apprendre.
from torchvision import transforms
a déjà été exécutée pour vous, alors commençons !
Cet exercice fait partie du cours
Deep learning intermédiaire avec PyTorch
Instructions
- Définissez
train_transforms
en combinant cinq transformations : un retournement horizontal aléatoire, une rotation aléatoire (de 0 à 45 degrés), un ajustement automatique aléatoire du contraste, l'analyse en tenseur et le redimensionnement à 64 x 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
)