Jeu de données d'images
Commençons par créer un jeu de données Torch d'images. Vous l'utiliserez pour explorer les données et, plus tard, pour l'alimenter à un modèle.
Les données d'entraînement pour la tâche de classification des nuages sont stockées selon la structure de répertoires suivante :
clouds_train
- cirriform clouds
- 539cd1c356e9c14749988a12fdf6c515.jpg
- ...
- clear sky
- cumulonimbus clouds
- cumulus clouds
- high cumuliform clouds
- stratiform clouds
- stratocumulus clouds
Il y a sept dossiers dans clouds_train, chacun représentant un type de nuage (ou un ciel dégagé). À l'intérieur de chacun de ces dossiers se trouvent les fichiers d'images correspondants.
Les importations suivantes ont déjà été faites pour vous :
from torchvision.datasets import ImageFolder
from torchvision import transforms
Cette activité fait partie du cours
Apprentissage profond intermédiaire avec PyTorch
Instructions de l’exercice
- Composez deux transformations : la première pour convertir l'image en tenseur, et la seconde pour redimensionner l'image à
128par128, et assignez-les àtrain_transforms. - Utilisez
ImageFolderpour définirdataset_train, en lui passant le chemin du répertoire des données ("clouds_train") et les transformations définies plus tôt.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Compose transformations
train_transforms = ____([
transforms.____,
transforms.____,
])
# Create Dataset using ImageFolder
dataset_train = ____(
____,
transform=____,
)