PyTorch code for our CVPR Paper [Fast and Flexible Indoor Scene Synthesis via Deep Convolutional Generative Models]

Requires PyTorch 0.4 to run, additional python library requirements could be found at /scene-synth/requirements.txt. Run

pip install -r requirements.txt

to install.

We trained our models on the SUNCG dataset. Due to an ongoing legal dispute, SUNCG is currently unavailable to download. As such, we feel that we should not provide anything derived from the dataset, including the pre-trained models, as well as several metadata files that our code rely on. We will update this page in the event that this situation changes in the future.

cat.py, loc.py, orient.py, dims.py contains the four components of our model respectively. fast_synth.py contains the main scene synthesis code. We will update with instructions on them as well.

We have a follow-up of this paper set to appear at SIGGRAPH 2019. In that paper, we introduce a high level "planning" module, formulated as a graph convolutional network, and uses the modules introduced here to fill out/"instantiate" low level details. Please refer to https://github.com/brownvc/planit for more details.