1. Обучение
    2. /
  2. Курса
    3. /
  3. Apprentissage profond intermédiaire avec PyTorch
Connected

упражнение

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

Инструкции

100 XP
  • 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.