IniziaInizia gratis

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

Visualizza il corso

Istruzioni dell'esercizio

  • Definisci train_transforms componendo 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
)
Modifica ed esegui il codice