1. Learn
    2. /
  2. Kurser
    3. /
  3. Deep Learning intermediar cu PyTorch
Connected

övning

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!

Instruktioner

100 XP
  • Definește train_transforms compunâ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.