Aan de slagGa gratis aan de slag

Afbeeldingsgegevensset

Laten we beginnen met het bouwen van een Torch Dataset van afbeeldingen. Je gebruikt die om de data te verkennen en later om het in een model te voeren.

De trainingsdata voor de wolkenclassificatietaak staat in de volgende mapstructuur:

clouds_train
  - cirriform clouds
    - 539cd1c356e9c14749988a12fdf6c515.jpg
    - ...
  - clear sky
  - cumulonimbus clouds
  - cumulus clouds
  - high cumuliform clouds
  - stratiform clouds
  - stratocumulus clouds

Er zijn zeven mappen binnen clouds_train, elk met één wolkentype (of een heldere lucht). In elk van deze mappen staan de bijbehorende afbeeldingsbestanden.

De volgende imports zijn alvast voor je gedaan:

from torchvision.datasets import ImageFolder
from torchvision import transforms

Deze oefening maakt deel uit van de cursus

Gevorderde Deep Learning met PyTorch

Cursus bekijken

Oefeninstructies

  • Combineer twee transformaties en wijs ze toe aan train_transforms: eerst om de afbeelding om te zetten naar een tensor, en daarna om de afbeelding te schalen naar 128 bij 128.
  • Gebruik ImageFolder om dataset_train te definiëren. Geef het het pad naar de data ("clouds_train") en de eerder gedefinieerde transformaties mee.

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

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

# Create Dataset using ImageFolder
dataset_train = ____(
    ____,
    transform=____,
)
Code bewerken en uitvoeren