FreeGAN - Simple machine learning project

Generate your own GAN models for free

Change this value in training.py num_epochs = 100 Higher value, better result. This value changes how much iterations training model does through dataset.

how to start script:

python main.py

What it does

• Load pictures dataset
• Train model
• Saves trained model into .h5 file
• Saves generated pictures into a folder
• Saves telemetry for each epoch in .csv file
• Resume training

In progress

-Main menu (80% done)

TODO

• Main Menu (Reset model, Resume training, Change dataset, Resolution picker)
• Fancy graphs, realtime image viewer, Iteration counter (epochs),
• Training values tweak so model is more accurate
• Tweaking model values straight from program, resizing resolution of dataset, multiple datasets.

You can put any dataset to train model. I decided to use anime pictures.

Demo timelapse (4h model progress)

Old version:

https://github.com/jstarzon/Anime-GAN-Neural-network