ganslate

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

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}
}