Skip to content

dliud/gan

Repository files navigation

CS 229 Final Project (Spring 2020) -- GAN-Based Image Data Augmentation by Nathan Hu and David Liu

To train GANs, run train_gans.py and modify the parameters in the main function (train_size, trial, alpha, num_gans). Make sure the trial is not already used unless you want the previous data to be overwritten. The train the classifier, run train_classifier.py and modify the parameters in the main function (num_real, num_synth, alpha, epoch). Check the saved model path.

About

cs229 final project

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages