Improving Facade Parsing with Vision Transformers and Line Integration

PDF

Model Structure

Data Preparation

Download Using following command to complie our dataset

cd data/
python facade_data_generation.py --root [your_root]

Training

python train.py --model_name Segmenter --batch_size 4 --root [your_root]

Inference

python inference.py --model_name Segmenter --eval_img [your_root]

Publication

If you want to use this work, please consider citing the following paper.

@article{wang2024improving,
  title={Improving facade parsing with vision transformers and line integration},
  author={Wang, Bowen and Zhang, Jiaxin and Zhang, Ran and Li, Yunqin and Li, Liangzhi and Nakashima, Yuta},
  journal={Advanced Engineering Informatics},
  volume={60},
  pages={102463},
  year={2024},
  publisher={Elsevier}
}