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.
requirements:
- nvidia GPU (CUDA 8+)
- pytorch 1.4
- watchdog
- visdom
- dominate
- opencv
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. You may need to run the container as the same user who will add the inputs (i.e. without sudo nvidia-docker):
mkdir input output
nvidia-docker run -v $(pwd)/input:/home/user/bikegan/input -v $(pwd)/output:/home/user/bikegan/output -it --rm twak/bikegan
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},
}