Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/torchgan/torchgan
Research Framework for easy and efficient training of GANs based on Pytorch
https://github.com/torchgan/torchgan
computer-vision deep-learning gans generative-adversarial-networks generative-model machine-learning neural-networks python python3 pytorch
Last synced: 20 days ago
JSON representation
Research Framework for easy and efficient training of GANs based on Pytorch
- Host: GitHub
- URL: https://github.com/torchgan/torchgan
- Owner: torchgan
- License: mit
- Created: 2018-09-20T16:41:00.000Z (about 6 years ago)
- Default Branch: master
- Last Pushed: 2022-10-23T03:55:31.000Z (about 2 years ago)
- Last Synced: 2024-10-09T15:21:57.463Z (about 1 month ago)
- Topics: computer-vision, deep-learning, gans, generative-adversarial-networks, generative-model, machine-learning, neural-networks, python, python3, pytorch
- Language: Python
- Homepage: https://torchgan.readthedocs.io/en/latest/
- Size: 22.9 MB
- Stars: 1,398
- Watchers: 29
- Forks: 168
- Open Issues: 28
-
Metadata Files:
- Readme: README.md
- License: LICENSE
- Citation: CITATION.bib
Awesome Lists containing this project
README
# TorchGAN
**Framework for easy and efficient training of GANs based on Pytorch**
[](https://www.repostatus.org/#active)
[](https://pepy.tech/project/torchgan)
[](https://pepy.tech/project/torchgan/month)
[](https://pepy.tech/project/torchgan/week)
[](LICENSE)
[](https://doi.org/10.21105/joss.02606)
[](https://torchgan.readthedocs.io/en/stable/)
[](https://torchgan.readthedocs.io/en/latest/)
[](https://codecov.io/gh/torchgan/torchgan)
[](https://mybinder.org/v2/gh/torchgan/torchgan/master)
[](https://colab.research.google.com/github/torchgan)
[](https://badge.fury.io/py/torchgan)
TorchGAN is a [Pytorch](https://pytorch.org) 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 |  |  |  |  |
| Linux py3.9 |  |  |  |  |
| OSX py3.8 |  |  |  |  |
| OSX py3.9 |  |  |  |  |
| Windows py3.9 |  |  |  |  |
| Windows py3.9 |  |  |  |  |
### Installation
Using pip (for stable release):
```bash
$ pip install torchgan
```
Using pip (for latest master):
```bash
$ pip install git+https://github.com/torchgan/torchgan.git
```
From source:
```bash
$ git clone https://github.com/torchgan/torchgan.git
$ cd torchgan
$ python setup.py install
```
### Documentation
The documentation is available [here](https://torchgan.readthedocs.io/en/latest/)
The documentation for this package can be generated locally.
```bash
$ 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
[](https://mybinder.org/v2/gh/torchgan/torchgan/master)
[](https://colab.research.google.com/github/torchgan)
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:
- [Tutorial 1: Introduction to TorchGAN](https://github.com/torchgan/torchgan/blob/master/tutorials/Tutorial%201.%20Introduction%20to%20TorchGAN.ipynb)
- [Tutorial 2: Custom Loss Functions](https://github.com/torchgan/torchgan/blob/master/tutorials/Tutorial%202.%20Custom%20Loss%20Functions.ipynb)
2. Intermediate Tutorials:
- [Tutorial 3: CycleGAN](https://github.com/torchgan/torchgan/blob/master/tutorials/Tutorial%203.%20CycleGAN.ipynb)
- [Tutorial 4: Self Attention GAN](https://github.com/torchgan/torchgan/blob/master/tutorials/Tutorial%204.%20Self%20Attention%20GAN.ipynb)
3. Advanced Tutorials:
- [Tutorial 5: Adversarial Autoencoder](https://github.com/torchgan/torchgan/blob/master/tutorials/Tutorial%205.%20Adversarial%20Autoencoder.ipynb)
### 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:
```
@article{Pal2021,
doi = {10.21105/joss.02606},
url = {https://doi.org/10.21105/joss.02606},
year = {2021},
publisher = {The Open Journal},
volume = {6},
number = {66},
pages = {2606},
author = {Avik Pal and Aniket Das},
title = {TorchGAN: A Flexible Framework for GAN Training and Evaluation},
journal = {Journal of Open Source Software}
}
```
List of publications & submissions using TorchGAN (please open a pull request to add missing entries):
* [Can GAN-Generated Network Traffic be used to Train Traffic Anomaly Classifiers?](https://ieeexplore.ieee.org/abstract/document/9284901?casa_token=9bEvJ3COOXMAAAAA:pz8TsMSrecv_ip1t9rEI7tYn_S_AQyZE_UrgYZ61vX_3Clu0Y17pFTUEpclAcBja13pPEqOsxypp)
* [Ward2ICU: A Vital Signs Dataset of Inpatients from the General Ward](https://arxiv.org/abs/1910.00752)
### 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.
