A composable GAN built for developers, researchers, and artists.

HyperGAN is in pre-release and open beta.

Logos generated with examples/colorizer

See more on the hypergan youtube

• About
• Documentation
• Changelog
• Quick start
  • Requirements
  • Install
  • Train
• API
  • Using a trained hypergan model
  • Training a gan
  • Examples
  • Tutorials
• The pip package hypergan
  • Training
  • Sampling
  • Additional Arguments
  • Running on CPU
  • Troubleshooting
  • Development Mode
• Datasets
  • Creating a Dataset
  • Downloadable Datasets
  • Cleaning up data
• Features
• Showcase
• Sponsors
• Contributing
• Versioning
• Citation

HyperGAN builds generative adversarial networks in pytorch and makes them easy to train and share.

For a general introduction to GANs see http://blog.aylien.com/introduction-generative-adversarial-networks-code-tensorflow/

Join the community discord.

• Gitbook documentation

See the full changelog here: Changelog.md

OS: Windows, OSX, Linux

For training:

GPU: Nvidia, GTX 1080+ recommended

1. Install HyperGAN For users: pip3 install hypergan

For developers: Download this repo and run python3 setup.py develop

2. Test it out

• hypergan train preset:celeba -s 128x128x3

3. Join the community

• Once you've made something cool, be sure to share it on the Discord (https://discord.gg/t4WWBPF).

  hypergan new mymodel

This will create a mymodel.json based off the default configuration. You can change configuration templates with the -c flag.

  hypergan new mymodel -l

See all configuration templates with --list-templates or -l.

  hypergan train folder/ -s 32x32x3 -c mymodel --resize

import hypergan as hg

Note this API is currently under work in 1.0. If you are reading this before 1.0 is released check the examples.

See the gitbook documentation for more details.

my_gan = hg.GAN('model.hypergan')
batch_sample = my_gan.sample()

gan = hg.GAN("default.json", inputs=hg.inputs.ImageLoader(...))
trainable_gan = hg.TrainableGAN(gan)
for step in trainable_gan.train():
    print("I'm on step ", step)

See the examples https://github.com/hypergan/HyperGAN/tree/master/examples

See the tutorials https://hypergan.gitbook.io/hypergan/tutorials

pip install hypergan

  # Train a 32x32 gan with batch size 32 on a folder of pngs
  hypergan train [folder] -s 32x32x3 -b 32 --config [name]

  hypergan sample [folder] -s 32x32x3 -b 32 --config [name] --sampler batch_walk --save_samples

By default hypergan will not save training samples to disk. To change this, use --save_samples.

To see a detailed list, run

  hypergan -h

You can switch the backend with:

  hypergan [...] -B cpu

Don't train on CPU! It's too slow.

Make sure that your cuda, nvidia drivers, pillow, pytorch, and pytorch vision are the latest version.

Check the discord for help.

If you wish to modify hypergan

git clone https://github.com/hypergan/hypergan
cd hypergan
python3 setup.py develop

Make sure to pip3 uninstall hypergan to avoid version conflicts.

To build a new network you need a dataset.

Datasets in HyperGAN are meant to be simple to create. Just use a folder of images. Nested folders work too.

HyperGAN is built to be resilient to all types of unclean data. By default images are resized then cropped if necessary.

See --nocrop, --random_crop and --resize for additional image scaling options.

A list of features in the 1.0 release:

• API
• CLI
• Viewer - an electron app to explore and create models
• Cross platform - Windows, OSX, Linux
• Inference - Add AI content generation to your project
• Training - Train custom models using accelerated parallel training backends
• Sharing - Share built models with each other. Use them in python projects as hypergan models, or in any project as onxx models
• Customizable - Define custom architectures in the json, or replace any component with your own pytorch creation
• Data - Built to work on unclean data and multiple data types
• Unsupervised learning
• Unsupervised alignment - Align one distribution to another or discover new novel distributions.
• Transfer learning
• Online learning

Submit your showcase with a pull request!

For more, see the #showcase room in

We are now accepting financial sponsors. Sponsor to (optionally) be listed here.

https://github.com/sponsors/hypergan

Contributions are welcome and appreciated! We have many open issues in the Issues tab. Join the discord.

See how to contribute.

HyperGAN uses semantic versioning. http://semver.org/

TLDR: x.y.z

• x is incremented on stable public releases.
• y is incremented on API breaking changes. This includes configuration file changes and graph construction changes.
• z is incremented on non-API breaking changes. z changes will be able to reload a saved graph.

  HyperGAN Community
  HyperGAN, (2016-2020+), 
  GitHub repository, 
  https://github.com/HyperGAN/HyperGAN

HyperGAN comes with no warranty or support.