Zacznij terazZacznij za darmo

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

Zobacz kurs

Instrukcje do ćwiczenia

  • Połącz dwie transformacje: pierwszą, która przetwarza obraz na tensor, oraz drugą, która zmienia rozmiar obrazu do 128 na 128 pikseli – przypisz je do zmiennej train_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=____,
)
Edytuj i uruchom kod