search

zouchuhang / LayoutNetv2

PyTorch implementation of our IJCV paper: "Manhattan Room Layout Reconstruction from a Single 360 image: A Comparative Study of State-of-the-art Methods"
https://arxiv.org/pdf/1910.04099.pdf
MIT License
221 stars 34 forks source link

LayoutNetv2 Visualization #1

Open thoang3 opened 5 years ago

thoang3 commented 5 years ago

Hi @zouchuhang,

Thank you for your great work! I am wondering how to perform visualization as in the pytorch version of LayoutNet ( https://github.com/sunset1995/pytorch-layoutnet )? I assume we'd need to modify "visual.py", but "visual_preprocess.py" and "visual_3d_layout.py" are more or less similar? Thank you!

zouchuhang commented 5 years ago

Hi @thoang3. Yes you can use "visual_3d_layout.py", just use our output "cor_id" in "test_PC.py" L 164 as the input to L123 in "visual_3d_layout.py".