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