Skip to content

kopero2000/nagyadat

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 

Repository files navigation

nagyadat

transformdata.py:

  • 16-21. sor: szükséges adatok beolvasása .mat file-ból
  • ImageDataset: DataSet megadása, itt lehet beállítani milyen módon legyen transzformálva a képek
  • read_data: adatok szétválogatása: mennyi train, validation, test adat legyen
  • 80-82. sor: dataloader-ek megadása
  • mean_std: átlag, szórás kiszámolása a képekből a normalizáláshoz
  • file végén beállítása a végleges transzformációknak a dataset-en illetve ezek elmentése

train.py:

  • 57.sorig a dataloader-ek, dataset-ek betöltése
  • 60-67.sor: eszköz beállítása a gpura, vgg16 model betöltése, plusz layerek beadása, 40 osztály legyen a végén (10-50 éves korig becsültem)
  • train_model: 100 epoch-ban tanul a train, és valid dataloadereket használtam a train-re és validation-re.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages