Ensemble de données d'images
Commençons par créer un ensemble d'images Torch. Vous l'utiliserez pour explorer les données et, plus tard, pour les introduire dans un modèle.
Les données d'entraînement pour la tâche de classification des nuages sont stockées dans la structure de répertoire suivante :
clouds_train
- cirriform clouds
- 539cd1c356e9c14749988a12fdf6c515.jpg
- ...
- clear sky
- cumulonimbus clouds
- cumulus clouds
- high cumuliform clouds
- stratiform clouds
- stratocumulus clouds
Vous trouverez sept dossiers à l'intérieur de clouds_train
, chacun représentant un type de nuage (ou un ciel clair). Dans chacun de ces dossiers se trouvent les fichiers images correspondants.
Les importations suivantes ont déjà été effectuées pour vous :
from torchvision.datasets import ImageFolder
from torchvision import transforms
Cet exercice fait partie du cours
Apprentissage profond intermédiaire avec PyTorch
Instructions
- Composez deux transformations, la première pour analyser l'image en un tenseur, et la seconde pour redimensionner l'image à
128
par128
, en les assignant àtrain_transforms
. - Utilisez
ImageFolder
pour définirdataset_train
, en lui transmettant le chemin d'accès aux données ("clouds_train"
) et les transformations définies précédemment.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Compose transformations
train_transforms = ____([
transforms.____,
transforms.____,
])
# Create Dataset using ImageFolder
dataset_train = ____(
____,
transform=____,
)