MCPcopy Index your code
hub / github.com/ennauata/houseganpp

github.com/ennauata/houseganpp @main

Chat with this repo
repository ↗ · DeepWiki ↗ · + Follow
324 symbols 2,078 edges 63 files 12 documented · 4%
What it actually does AI analysis from the code graph — generated when you open this
loading…
README

House-GAN++

Code and instructions for our paper: House-GAN++: Generative Adversarial Layout Refinement Network towards Intelligent Computational Agent for Professional Architects, CVPR 2021. Project website.

Data

alt text We have used the RPLAN dataset, which offers 60k vector-graphics floorplans designed by professional architects. Qualitative and quantitative evaluations based on the three standard metrics (i.e., realism, diversity, and compatibility) in the literature demonstrate that the proposed system outperforms the current-state-of-the-art by a large margin.

Demo

image Please check out our live demo.

Running pretrained models

See requirements.txt for checking the dependencies before running the code

For running a pretrained model check out the following steps: - Run python test.py. - Check out the results in output folder.

Training models

  • Download the raw RPLAN dataset.
  • Run this script for processing the dataset and extracting JSON files.
  • The extracted JSON files serves directly as input to our dataloader.

Citation

Please consider citing our work.

@inproceedings{nauata2021house,
  title={House-GAN++: Generative Adversarial Layout Refinement Network towards Intelligent Computational Agent for Professional Architects},
  author={Nauata, Nelson and Hosseini, Sepidehsadat and Chang, Kai-Hung and Chu, Hang and Cheng, Chin-Yi and Furukawa, Yasutaka},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  pages={13632--13641},
  year={2021}
}

Contact

If you have any question, feel free to contact me at nnauata@sfu.ca.

Acknowledgement

This research is partially supported by NSERC Discovery Grants, NSERC Discovery Grants Accelerator Supplements, DND/NSERC Discovery Grant Supplement, and Autodesk. We would like to thank architects and students for participating in our user study.

Core symbols most depended-on inside this repo

conv_block
called by 24
models/models.py
scale_dimension
called by 8
misc/intersections.py
conv_block
called by 7
models/models_improved.py
compute_gradient_penalty
called by 7
models/models.py
helperDoIntersect
called by 5
misc/intersections.py
conv_block
called by 5
misc/old/autoencoder_dataset.py
conv_block
called by 5
testing/reconstruction_dataset.py
make_sequence
called by 4
dataset/floorplan_dataset_maps_functional_high_res.py

Shape

Function 238
Method 61
Class 25

Languages

Python100%

Modules by API surface

models/model_resnet.py25 symbols
testing/reconstruct.py16 symbols
models/models_improved.py16 symbols
dataset/floorplan_dataset_maps_functional_high_res.py16 symbols
models/models.py12 symbols
testing/pytorch_visualize.py11 symbols
misc/utils.py10 symbols
testing/evaluate_parallel.py9 symbols
scripts/optimization_valid_houses_v3.py9 symbols
misc/old/coordconv.py9 symbols
testing/reconstruction_dataset.py8 symbols
scripts/optimizing_compatibility_valid_houses.py8 symbols

For agents

$ claude mcp add houseganpp \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact