/GAN-JAX

Primary LanguagePython

GAN-JAX

-src

|-architecture : code for the generators and discriminators

|-data : code for the datasets

|-models : training algorithms

TODO :

  • Architecture selection from config (replaces the hardcoded class import)

Methods :

  • CycleGAN

  • CUT

  • StarGAN v1

  • StarGAN v2

Architectures :

  • Resnet

  • Unet

  • Visual transformer

  • Stylegan