复现SUDS代码,主要进行环境配置,数据预处理,训练,查看PSNR渲染指标
1、环境配置
根据SUDS提供的environment.yml文件创建环境,由于安装总是出现环境问题,或者某些包无法下载的问题,如图,在安装torch-scatter时显示缺少文件
所以分布配置环境
首先创建环境,安装cuda11.6 pytorch1.13.1
create -n suds python=3.9.16
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia
再进行pip install安装,同样出现缺少文件问题:fatal error: cusolverDn.h: No such file or directory
这里参考这篇博客:链接
sudo apt install nvidia-cuda-dev
以上安装之后的确可以继续安装,但是在进行实例拷贝到其他服务器时,出现了问题:
Failed to initialize NVML: Driver/library version mismatch
我后来也参考这篇github的解决方法,修改PATH变量,但是没有成功,后面再看解决方法
接着需要安装tiny-cuda和nerfstudio,作者提供的都是他当时使用的版本,所以直接安装最新的版本可能会出现版本问题。
安装步骤在这篇博客
2、数据预处理
尽管不知道环境是否配置成功,但是在进行数据预处理时,运行的命令是可以得到深度图、掩码、DINO特征数据的,根据作者提供的命令就可以得到以下格式的数据
在进行以下命令时,使用的0006序列有270*2=540张图片,所以视角很久,需要6个小时左右得到dino_correspondences_02、dino_correspondences_03文件