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
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 naar128bij128. - Gebruik
ImageFolderomdataset_trainte 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=____,
)