Set de date cu augmentări
Ai construit deja setul de date cu imagini de nori și modelul convoluțional pentru clasificarea diferitelor tipuri de nori. Înainte de antrenare, hai să adaptăm setul de date adăugând augmentările care pot îmbunătăți performanța modelului la clasificarea norilor.
Codul pentru configurarea Dataset-ului și DataLoader-ului este deja pregătit și ar trebui să-ți fie familiar. Sarcina ta este să definești compoziția de transformări care vor fi aplicate imaginilor de intrare pe măsură ce sunt încărcate.
Reține că anterior redimensionai imaginile la 128 x 128 pentru o afișare mai clară, dar acum vei folosi unele mai mici pentru a accelera antrenarea. Așa cum vei vedea mai târziu, 64 x 64 este suficient de mare pentru ca modelul să învețe.
from torchvision import transforms a fost deja executat pentru tine, deci să începem!
Acest exercițiu face parte din cursul
Deep Learning intermediar cu PyTorch
Instrucțiuni pentru exercițiu
- Definește
train_transformscompunând cinci transformări: o răsturnare orizontală aleatorie, o rotație aleatorie (cu un unghi între 0 și 45 de grade), o ajustare automată aleatorie a contrastului, conversia la tensor și redimensionarea la 64 x 64 pixeli.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# Define transforms
train_transforms = transforms.Compose([
____,
____,
____,
____,
____,
])
dataset_train = ImageFolder(
"clouds_train",
transform=train_transforms,
)
dataloader_train = DataLoader(
dataset_train, shuffle=True, batch_size=16
)