2D Layout estimation

Line drawing algorithm for segmented layout images generated by the project from https://github.com/fraunhoferhhi/spvloc

Install spvloc dependencies

#Ubuntu
source venv/bin/activate
git clone https://github.com/MRamazan/2D-Room-Layout-Estimation
cd 2D-Room-Layout-Estimation
pip install -r requirements.txt

cd spvloc
# Build and install redner
./data/setup/install_redner.sh
# Install patched version of pyrender
./data/setup/install_pyrender.sh
# if permission denied
# chmod +x ./data/setup/install_redner.sh ./data/setup/install_pyrender.sh
# rerun the scripts

python -m spvloc.tools.download_pretrained_models

cd ../

Testing

Segmented Image

python layout.py --image_path bedroom.jpg

Layout Image

python core.py --image_path bedroom.jpg

Results

Resources

https://github.com/fraunhoferhhi/spvloc