1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Intermediate Deep Learning with PyTorch
Connected

cvičení

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!

Pokyny

100 XP
  • 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ů.