Skip to main content

Research Framework for easy and efficient training of GANs based on Pytorch

Project description

TorchGAN

Framework for easy and efficient training of GANs based on Pytorch

Project Status: Active – The project has reached a stable, usable state and is being actively developed. Downloads Downloads Downloads License

Stable Documentation Latest Documentation Codecov Binder Open In Colab PyPI version

TorchGAN is a Pytorch based framework for designing and developing Generative Adversarial Networks. This framework has been designed to provide building blocks for popular GANs and also to allow customization for cutting edge research. Using TorchGAN's modular structure allows

  • Trying out popular GAN models on your dataset.
  • Plug in your new Loss Function, new Architecture, etc. with the traditional ones.
  • Seamlessly visualize the training with a variety of logging backends.
System / PyTorch Version 1.8 1.9 nightly
Linux py3.8 CI Testing CI Testing CI Testing
Linux py3.9 CI Testing CI Testing CI Testing
OSX py3.8 CI Testing CI Testing CI Testing
OSX py3.9 CI Testing CI Testing CI Testing
Windows py3.9 CI Testing CI Testing CI Testing
Windows py3.9 CI Testing CI Testing CI Testing

Installation

Using pip (for stable release):

  $ pip install torchgan

Using pip (for latest master):

  $ pip install git+https://github.com/torchgan/torchgan.git

From source:

  $ git clone https://github.com/torchgan/torchgan.git
  $ cd torchgan
  $ python setup.py install

Documentation

The documentation is available here

The documentation for this package can be generated locally.

  $ git clone https://github.com/torchgan/torchgan.git
  $ cd torchgan/docs
  $ pip install -r requirements.txt
  $ make html

Now open the corresponding file from build directory.

Tutorials

Binder Open In Colab

The tutorials directory contain a set of tutorials to get you started with torchgan. These tutorials can be run using Google Colab or Binder. It is highly recommended that you follow the tutorials in the following order.

  1. Introductory Tutorials:
  2. Intermediate Tutorials:
  3. Advanced Tutorials:

Supporting and Citing

This software was developed as part of academic research. If you would like to help support it, please star the repository. If you use this software as part of your research, teaching, or other activities, we would be grateful if you could cite the following:

@misc{pal2019torchgan,
    title={{TorchGAN: A Flexible Framework for GAN Training and Evaluation}},
    author={Avik Pal, and Aniket Das},
    year={2019},
    eprint={1909.03410},
    archivePrefix={arXiv},
    primaryClass={cs.LG}
}

List of publications & submissions using TorchGAN (please open a pull request to add missing entries):

Contributing

We appreciate all contributions. If you are planning to contribute bug-fixes, please do so without any further discussion. If you plan to contribute new features, utility functions or extensions, please first open an issue and discuss the feature with us. For more detailed guidelines head over to the official documentation.

Contributors

This package has been developed by

  • Avik Pal (@avik-pal)
  • Aniket Das (@Aniket1998)

This project exists thanks to all the people who contribute.

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

torchgan-0.1.0.tar.gz (47.3 kB view details)

Uploaded Source

Built Distribution

torchgan-0.1.0-py3-none-any.whl (71.7 kB view details)

Uploaded Python 3

File details

Details for the file torchgan-0.1.0.tar.gz.

File metadata

  • Download URL: torchgan-0.1.0.tar.gz
  • Upload date:
  • Size: 47.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.4.2 importlib_metadata/4.8.1 pkginfo/1.7.1 requests/2.23.0 requests-toolbelt/0.9.1 tqdm/4.61.2 CPython/3.9.5

File hashes

Hashes for torchgan-0.1.0.tar.gz
Algorithm Hash digest
SHA256 d55396a72c302aa8b05e503a421ac8f68b8e9e8b87f924a647a048b271c5b246
MD5 5040f62298e40d53c74e50fba0abd838
BLAKE2b-256 8ccd55b0bc098beaee1a85084450d4a37b95048c61eb1598d1186a1197667623

See more details on using hashes here.

File details

Details for the file torchgan-0.1.0-py3-none-any.whl.

File metadata

  • Download URL: torchgan-0.1.0-py3-none-any.whl
  • Upload date:
  • Size: 71.7 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.4.2 importlib_metadata/4.8.1 pkginfo/1.7.1 requests/2.23.0 requests-toolbelt/0.9.1 tqdm/4.61.2 CPython/3.9.5

File hashes

Hashes for torchgan-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 6be3fd633fff65b3120e43b3e5f46cf7529fe2b20db7c3bf467d24b1bb865a12
MD5 ee5df3ed7af8095cfb0f155cb8a88dc9
BLAKE2b-256 fe37bee3edb085c23f999f732c49c4b90ffce60b168ae6b108e0725621e233f9

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page