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

cvičení

Augmentace dat v PyTorchi

Pojďme přidat augmentaci dat do tvého Datasetu a vizuálně zkontrolovat několik obrázků, abychom se ujistili, že se požadované transformace správně aplikují.

Nejdřív přidáš augmentační transformace do train_transforms. Použijeme náhodné horizontální překlopení a rotaci o náhodný úhel od 0 do 45 stupňů. Kód pro vytvoření Datasetu a DataLoaderu zůstává stejný jako předtím. Na závěr obrázek přetvaruješ a zobrazíš, abys viděl/a, jestli jsou nové augmentační transformace viditelné.

Všechny potřebné importy jsou už připravené:

from torchvision.datasets import ImageFolder
from torch.utils.data import DataLoader
from torchvision import transforms
import matplotlib.pyplot as plt

Čas augmentovat fotky oblaků!

Pokyny

100 XP
  • Přidej do train_transforms dvě další transformace: náhodné horizontální překlopení a rotaci o náhodný úhel od 0 do 45 stupňů.
  • Přetvaruj tensor image z DataLoaderu do podoby vhodné pro zobrazení.
  • Zobraz obrázek.