CommencezCommencez gratuitement

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

Voir le cours

Instructions de l’exercice

  • Composez deux transformations : la première pour convertir l'image en tenseur, et la seconde pour redimensionner l'image à 128 par 128, et assignez-les à train_transforms.
  • Utilisez ImageFolder pour définir dataset_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=____,
)
Modifier et exécuter le code