ganslate - a simple and extensible GAN image-to-image translation framework

For comprehensive documentation, visit: https://ganslate.readthedocs.io/en/latest/

Note: The documentation is still in progress! Suggestions and contributions are welcome!

ganslate is a PyTorch framework which aims to make GAN image-to-image translation more accessible to both beginner and advanced project with:

• Simple configuration system
• Extensibility for other datasets or architectures
• Documentation and video walk-throughs (soon)

Features

• 2D and 3D support
• Mixed precision
• Distributed training
• Tensorboard and Weights&Biases logging
• Natural and medical image support
• A range of generator and discriminator architectures

Available GANs

• Pix2Pix (paper)
• CycleGAN (paper)
• RevGAN (paper)
• CUT (Contrastive Unpaired Translation) (paper)

Projects

ganslate was used in:

• Project 1
• Project 2

Citation

If you used ganslate in your project, please cite:

@software{ibrahim_hadzic_2021_5494572,
  author       = {Ibrahim Hadzic and
                  Suraj Pai and
                  Chinmay Rao and
                  Jonas Teuwen},
  title        = {ganslate-team/ganslate: Initial public release},
  month        = sep,
  year         = 2021,
  publisher    = {Zenodo},
  version      = {v0.1.0},
  doi          = {10.5281/zenodo.5494572},
  url          = {https://doi.org/10.5281/zenodo.5494572}
}