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
Istruzioni dell'esercizio
- Componi due trasformazioni: la prima per convertire l'immagine in un tensore, e la seconda per ridimensionarla a
128per128, assegnandole atrain_transforms. - Usa
ImageFolderper definiredataset_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=____,
)