Group Members: Joseph Lieberman Jacob Barkovitch
Machine Learning Final Project
Our improved StyleGAN2 gives it 3D functionality as well as improved performance.
To run original model on new data: change directory in train_from_folder method.
To run our 3D model on new data: change self.dataset in set_data_src method and change filter number parameters to desired number of data channels.
Contributions:
Joe - Did not have a NVIDIA GPU, so could not work/commit anything to the code. Got on zoom a couple times and worked on code together. Collaborated on Abstract, Final Project Update, Presentation, Report, etc.
Jacob - Worked on the code and suggested the 3-D change. Collaborated on Abstract, Final Project Update, Presentation, Report, etc.