frankenGAN networks
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.

README.md

bikeGAN

this is the fork of bicycleGAN which is used to train and run GANs for frankenGAN. the interactive system which uses these networks is chordatlas.

running

requirements:

  • nvidia GPU (CUDA 8+)
  • pytorch 1.4
  • visdom
  • dominate

the entry point is test_interactive.py which listens to the ./input folders for new inputs, and writes them to ./output (these folders should exist). it will download the pre-trained model weights the first time your run it. Once it is running, set chordatlas's bikeGAN file location (in the settings menu) to the bikeGAN root directory (the one containing this file).

alternatively, use the docker container with nvidia-docker:

nvidia-docker run -v $(pwd)/input:/home/user/bikegan/input -v $(pwd)/output:/home/user/bikegan/output -it --rm twak/bikegan

cite

if you use this project, please cite frankenGAN

@article{frankengan,
  title = {FrankenGAN: Guided Detail Synthesis for Building Mass-Models Using Style-Synchonized GANs},
  author = {Tom Kelly and Paul Guerrero and Anthony Steed and Peter Wonka and Niloy J. Mitra},
  year = {2018},
  journal = {{ACM} Transactions on Graphics},
  volume = {37},
  number = {6},
  doi = {10.1145/3272127.3275065},
}