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