Framework to ease training of generative models based on TensorFlow
Project description
SimpleGAN
Framework to ease training of generative models
SimpleGAN is a framework based on TensorFlow to make training of generative models easier. SimpleGAN provides high level APIs with customizability options to user which allows them to train a generative model with few lines of code.
Requirements
Make sure you have the following packages installed
Installation
Latest stable release:
$ pip install simplegan
Latest Development release:
$ pip install git+https://github.com/grohith327/EasyGAN.git
Getting Started
>>> from simplegan.gan import DCGAN
>>> gan = DCGAN()
>>> train_ds = gan.load_data(use_mnist = True)
>>> samples = gan.get_sample(train_ds, n_samples = 5)
>>> gan.fit(train_ds = train_ds)
>>> generated_samples = gan.generate_samples(n_samples = 5)
To have a look at more examples, check here
Provided models
- Autoencoders
- Generative Adversarial Networks(GANs)
Contributing
We appreciate all contributions. If you are planning to perform bug-fixes, add new features or models, please file an issue and discuss before making a pull request.
Contributors
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for simplegan-0.2.6-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e86b5eb856d974afb912823c939a3b99c3c4f6f542e72fa5fd06cbdfad1dbd71 |
|
MD5 | 52a1f56afa6a35044ba3222dda9d769b |
|
BLAKE2b-256 | 933ad0c34353df8a9df051b108a0f0f57f92a55921a8b3a4f243dad406b27087 |