CommencerCommencer gratuitement

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

Afficher le cours

Instructions

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