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
Pokyny k cvičení
- Definuj
train_transformsslož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
)