conda create -n roomlayout python=3.8
pip install -r requirement.txt

# train
python main.py --mode train

The weight of Structured3d and nyu303 can be downloaded in feijiang 、 baiduNet or google drive

And put them in the weights/(structured3d/nyu303) folder