Kom igångKom igång gratis

Datamängd med augmenteringar

Du har redan byggt bilddatamängden med molnbilder och den konvolutionella modellen för att klassificera olika molntyper. Innan du tränar modellen ska vi anpassa datamängden genom att lägga till augmenteringar som kan förbättra modellens prestanda vid molnklassificering.

Koden för att sätta upp Dataset och DataLoader är redan förberedd och bör se bekant ut. Din uppgift är att definiera kompositionen av transformationer som tillämpas på indatabilderna när de laddas.

Observera att du tidigare ändrade storleken på bilderna till 128 × 128 för att visa dem snyggt, men nu använder du mindre bilder för att snabba upp träningen. Som du kommer att se senare räcker 64 × 64 för att modellen ska kunna lära sig.

from torchvision import transforms har redan körts åt dig – nu kör vi!

Den här övningen är en del av kursen

Fördjupad djupinlärning med PyTorch

Visa kurs

Övningsinstruktioner

  • Definiera train_transforms genom att sätta samman fem transformationer: en slumpmässig horisontell spegling, slumpmässig rotation (med vinkel från 0 till 45 grader), slumpmässig automatisk kontrastjustering, konvertering till tensor samt storleksändring till 64 × 64 pixlar.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

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

dataset_train = ImageFolder(
  "clouds_train",
  transform=train_transforms,
)
dataloader_train = DataLoader(
  dataset_train, shuffle=True, batch_size=16
)
Redigera och kör kod