Aan de slagGa gratis aan de slag

Gegevensset met augmentaties

Je hebt al de beeldgegevensset gemaakt van wolkenfoto's en het convolutionele model om verschillende wolkentypes te classificeren. Voordat je het gaat trainen, passen we de gegevensset aan door augmentaties toe te voegen die de classificatieprestaties van het model kunnen verbeteren.

De code om de Dataset en DataLoader in te stellen is al voor je voorbereid en zou bekend moeten zijn. Jouw taak is om de samenstelling van transforms te definiëren die op de invoerafbeeldingen worden toegepast zodra ze worden geladen.

Let op: eerder wijzigde je de grootte van afbeeldingen naar 128 bij 128 om ze netjes weer te geven, maar nu gebruik je kleinere beelden om het trainen te versnellen. Zoals je later zult zien, is 64 bij 64 groot genoeg voor het model om te leren.

from torchvision import transforms is al voor je uitgevoerd, dus aan de slag!

Deze oefening maakt deel uit van de cursus

Gevorderde Deep Learning met PyTorch

Cursus bekijken

Oefeninstructies

  • Definieer train_transforms door vijf transformaties te combineren: een willekeurige horizontale spiegeling, willekeurige rotatie (met een hoek van 0 tot 45 graden), willekeurige automatische contrastaanpassing, omzetten naar een tensor en schalen naar 64 bij 64 pixels.

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

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

dataset_train = ImageFolder(
  "clouds_train",
  transform=train_transforms,
)
dataloader_train = DataLoader(
  dataset_train, shuffle=True, batch_size=16
)
Code bewerken en uitvoeren