Total3DUnderstanding
Total3DUnderstanding:从单个图像对室内场景进行联合布局,对象姿势和网格重建聂银雨,韩晓光,郭世辉,郑玉建,张建,张建军在。
安装
此实现使用Python 3.6, 和cudatoolkit 9.0。 我们建议使用部署环境。
使用conda安装:
conda env create -f environment.yml
conda activate Total3D
用pip安装:
pip install -r requirements.txt
演示版
预训练的模型可以在下载。 我们在这里还提供了预训练的网格生成网络。 将预训练的模型放在
out/pretrained_models
下面演示了一个演示,以了解该方法的工作原理。 这里使用vtk可视化3D场景。 输出将保存在“演示/输出”下。 您也可以使用此脚本来玩玩具。
1