Začněte nyníZačněte zdarma

Dataset s augmentacemi

Datovou sadu obrázků oblaků a konvoluční model pro klasifikaci jejich typů už máš za sebou. Než model natrénuješ, upravíme datovou sadu – přidáme augmentace, které mohou zlepšit výkon modelu při klasifikaci oblaků.

Kód pro nastavení Dataset a DataLoader je už připravený a měl by ti být povědomý. Tvým úkolem je definovat kompozici transformací, které se budou aplikovat na vstupní obrázky při jejich načítání.

Dřív jsi měnil/a velikost obrázků na 128 × 128 pixelů, aby se hezky zobrazovaly. Teď použijeme menší rozměry, aby trénování proběhlo rychleji. Jak uvidíš, rozlišení 64 × 64 bude pro učení modelu dostatečné.

from torchvision import transforms už je naimportováno, takže se pusťme do toho!

Toto cvičení je součástí kurzu

Intermediate Deep Learning with PyTorch

Zobrazit kurz

Pokyny k cvičení

  • Definuj train_transforms složením pěti transformací: náhodné horizontální překlopení, náhodná rotace (o úhel od 0 do 45 stupňů), náhodná automatická úprava kontrastu, převod na tensor a změna velikosti na 64 × 64 pixelů.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

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

dataset_train = ImageFolder(
  "clouds_train",
  transform=train_transforms,
)
dataloader_train = DataLoader(
  dataset_train, shuffle=True, batch_size=16
)
Upravit a spustit kód