Zbiór danych z obrazami
Zacznijmy od zbudowania zbioru danych Torch zawierającego obrazy. Wykorzystasz go do eksploracji danych, a później – do przekazywania ich do modelu.
Dane treningowe do zadania klasyfikacji chmur są zapisane w następującej strukturze katalogów:
clouds_train
- cirriform clouds
- 539cd1c356e9c14749988a12fdf6c515.jpg
- ...
- clear sky
- cumulonimbus clouds
- cumulus clouds
- high cumuliform clouds
- stratiform clouds
- stratocumulus clouds
Wewnątrz katalogu clouds_train znajduje się siedem folderów – każdy odpowiada jednemu typowi chmur (lub bezchmurnemu niebu). W każdym z tych folderów są zapisane odpowiadające mu pliki obrazów.
Następujące importy zostały już wykonane za ciebie:
from torchvision.datasets import ImageFolder
from torchvision import transforms
To ćwiczenie jest częścią kursu
Głębokie uczenie z PyTorch – poziom średnio zaawansowany
Instrukcje do ćwiczenia
- Połącz dwie transformacje: pierwszą, która przetwarza obraz na tensor, oraz drugą, która zmienia rozmiar obrazu do
128na128pikseli – przypisz je do zmiennejtrain_transforms. - Użyj
ImageFolder, aby zdefiniowaćdataset_train, przekazując ścieżkę do katalogu z danymi ("clouds_train") oraz wcześniej zdefiniowane transformacje.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# Compose transformations
train_transforms = ____([
transforms.____,
transforms.____,
])
# Create Dataset using ImageFolder
dataset_train = ____(
____,
transform=____,
)