frankenGAN networks
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
data
datasets update README Jan 17, 2018
docker
models fixed bug: style vector encoded from images was chosen randomly from … May 31, 2018
options
scripts
util
.gitignore
.pylintrc added options needed for my networks May 18, 2018
LICENSE
README.md
test.py code refactoring; remove unused code Apr 16, 2018
test_interactive.py
train.py code refactoring; remove unused code Apr 16, 2018
video.py code refactoring; remove unused code Apr 16, 2018

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