Insieme di dati con augmentations
Hai già costruito l’insieme di dati di immagini con foto di nuvole e il modello convoluzionale per classificare i diversi tipi di nuvole. Prima di addestrarlo, adattiamo l’insieme di dati aggiungendo le augmentations che possono migliorare le prestazioni del modello nella classificazione delle nuvole.
Il codice per impostare Dataset e DataLoader è già pronto per te e dovrebbe risultarti familiare. Il tuo compito è definire la composizione di trasformazioni da applicare alle immagini in ingresso mentre vengono caricate.
Nota che prima ridimensionavi le immagini a 128 per 128 per visualizzarle meglio, ma ora userai dimensioni più piccole per velocizzare l’addestramento. Come vedrai tra poco, 64 per 64 sarà sufficiente perché il modello possa imparare.
from torchvision import transforms è già stato eseguito per te, quindi mettiamoci al lavoro!
Questo esercizio fa parte del corso
Deep Learning intermedio con PyTorch
Istruzioni dell'esercizio
- Definisci
train_transformscomponendo insieme cinque trasformazioni: un ribaltamento orizzontale casuale, una rotazione casuale (con angolo da 0 a 45 gradi), una regolazione automatica del contrasto casuale, la conversione in tensore e il ridimensionamento a 64 per 64 pixel.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# Define transforms
train_transforms = transforms.Compose([
____,
____,
____,
____,
____,
])
dataset_train = ImageFolder(
"clouds_train",
transform=train_transforms,
)
dataloader_train = DataLoader(
dataset_train, shuffle=True, batch_size=16
)