IniziaInizia gratis

Insieme di immagini (dataset)

Iniziamo creando un Dataset di Torch per le immagini. Lo userai per esplorare i dati e, più avanti, per fornirli a un modello.

I dati di training per il task di classificazione delle nuvole sono salvati nella seguente struttura di directory:

clouds_train
  - cirriform clouds
    - 539cd1c356e9c14749988a12fdf6c515.jpg
    - ...
  - clear sky
  - cumulonimbus clouds
  - cumulus clouds
  - high cumuliform clouds
  - stratiform clouds
  - stratocumulus clouds

Ci sono sette cartelle dentro clouds_train, ognuna rappresenta un tipo di nuvola (o un cielo sereno). All'interno di ciascuna cartella si trovano i relativi file immagine.

I seguenti import sono già stati eseguiti per te:

from torchvision.datasets import ImageFolder
from torchvision import transforms

Questo esercizio fa parte del corso

Deep Learning intermedio con PyTorch

Visualizza il corso

Istruzioni dell'esercizio

  • Componi due trasformazioni: la prima per convertire l'immagine in un tensore, e la seconda per ridimensionarla a 128 per 128, assegnandole a train_transforms.
  • Usa ImageFolder per definire dataset_train, passandogli il percorso della directory dei dati ("clouds_train") e le trasformazioni definite in precedenza.

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

# Compose transformations
train_transforms = ____([
    transforms.____,
    transforms.____,
])

# Create Dataset using ImageFolder
dataset_train = ____(
    ____,
    transform=____,
)
Modifica ed esegui il codice