Will Hoover

Week of April 8 – April 14 2024

This week was the first week of the second month of our project. We are focusing this month on getting the StyleGAN2 architecture up and running for the project. This was a very complicated architecture to get running on the default values, so we have taken steps to try and mitigate some of the frustrations we have encountered so far. I am working on this part of the project in the ‘generator_’ branch of the git repository, I have split the script we built the network into modular components that are properly fed to each other and successfully trains from it.

FaceCraft Folder heirarchy for the

This will allows us to adjust each individual component as we need to, or add any relevant component into our project without too much delay in implementing it into the flow. Once I got the StyleGAN2 network running without flags, I had it do some test training just to make sure that each component that I implemented was working correctly. It was able to find the images, start training off the 1000 sample images I provided it and generate a blob.

Results after 1 epoch

Results after 4 epochs

This was all very successful but there was a snag, the Onyx image was not being created when the checkpoint was logged. This has become the focal point of my recent attention, with the ability to export any trained model it will allow the other team members to start making front end tools that will interact with the final model and testing their functionality all while I am training and fine tuning the final model we will be shipping.

Once I get everything regarding training, saving, exporting and loading setup with proper flags to let me know whats going on and recording the proper data that would be needed to further tune the model I will be increasing the training time with the expectations of improved results.