室内SfMLearner
ECCV2020论文的PyTorch实施:
于泽浩*,雷金*,
(*平等贡献)
入门
安装
pip install -r requirements.txt
然后安装pytorch与
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
Pytorch版本> = 0.4.1会很好用。
下载预训练的模型
请从下载预训练的模型并提取:
tar -xzvf ckpts.tar.gz
rm ckpts.tar.gz
对单个图像的预测
运行以下命令以在单个图像上进行预测:
python inference_single_image.py --image_path=/path/to/image
默认情况下,脚本会将预测深度保存到同一文件夹
评估
从下载测试数据并放入./data。
cd data
评论0
最新资源