当前位置:   article > 正文

Gaussian Splatting代码安装部署(windows)_gaussian splatting部署

gaussian splatting部署

项目网址 https://github.com/graphdeco-inria/gaussian-splatting
电脑配置 Windows 11 RTX 3060

一、环境配置

电脑已有
Anaconda3
Visual Studio 2022
CUDA 11.8

1.提前准备

Visual Studio 2019社区,选择“使用C++进行桌面开发”(他推荐的是2019,但我电脑里同时有201020192022好像也没冲突)
git
conda
Cuda  v.11.8
COLMAP
ImageMagik
FFMPEG
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7

下载源码

git clone https://github.com/graphdeco-inria/gaussian-splatting --recursive
  • 1

推荐直接git,如果在github上下载压缩包的话有两个文件可能有问题。

2.创建虚拟环境

创建环境不要用源码里给的environment.yml。
他这readme写的11.6有问题,然后yml里面写的还是11.6的,很离谱。
在这里插入图片描述
参考了issues332里一个老哥写的

https://github.com/graphdeco-inria/gaussian-splatting/issues/332
  • 1

但是要注意他用的cuda是11.7,如果之前按照yml安装过的话,要先把之前的虚拟环境删除。
我的CUDA版本是11.8,所以按照下面的方法进行:
打开Anaconda Prompt,创建虚拟环境,并激活。

SET DISTUTILS_USE_SDK=1 # Windows only
conda create -n gaussian_splatting python=3.10
conda activate gaussian_splatting
  • 1
  • 2
  • 3

进入下载的源码位置
我的源码位置是D:\PythonProject\gaussian-splatting\gaussian-splatting>
所以

D:   # 转到D盘
cd D:\PythonProject\gaussian-splatting\gaussian-splatting  # 进入对应目录
  • 1
  • 2

下载conda和vs2019联动的插件

conda install -c anaconda vs2019_win-64
  • 1

下载pytorch等一些列东西,一定要注意里面的版本对应关系,比如我是CUDA11.8,下面就是cu118的

pip install torch==2.0.0+cu118 torchvision==0.15.0+cu118 torchaudio==2.0.0+cu118 -f https://download.pytorch.org/whl/torch_stable.html
  • 1

安装其他依赖

pip install submodules\diff-gaussian-rasterization
pip install submodules\simple-knn
pip install plyfile
pip install tqdm
  • 1
  • 2
  • 3
  • 4

其中diff-gaussian-rasterization和simple-knn这两个包最好是在第一步下载源码的时候直接下载下来,如果是自己从github上下载的压缩包解压的,可能有些问题。

二、跑提供的模型

新建一个data文件夹,把作者提供的数据集下载下来

https://repo-sam.inria.fr/fungraph/3d-gaussian-splatting/datasets/input/tandt_db.zip
  • 1

解压后的文件
在这里插入图片描述
训练

python train.py -s data/tandt/train
  • 1

训练时会生成一个7000步的和一个完整的,在./output目录下面有一个乱码的文件夹
在这里插入图片描述
可视化:安装提供的软件

cd SIBR_viewers
cmake -Bbuild .
cmake --build build --target install --config RelWithDebInfo
  • 1
  • 2
  • 3

如果有问题话直接下载编译好的到根目录里

https://repo-sam.inria.fr/fungraph/3d-gaussian-splatting/binaries/viewers.zip
  • 1

最后的文件结构大概是这样的
在这里插入图片描述
可视化生成的模型

.\viewers\bin\SIBR_gaussianViewer_app -m ./output/(生成的乱码)
  • 1

我这里是

.\viewers\bin\SIBR_gaussianViewer_app -m ./output/eb5f984a-7
  • 1

在这里插入图片描述
在可视化界面里
在这里插入图片描述

点击红色方框的下拉框,三个选项一次是渲染后的效果,点云和高斯(其中高斯比较卡),在fps模式下可以鼠标旋转,wasd前后左右移动。
最上面FPS的下拉框里可以改变显示方式。

三、跑自己的模型

参考 https://www.bilibili.com/read/cv26465887/

四、其他

【3D Gaussian Splatting本地部署【windows系统免环境配置】】 https://www.bilibili.com/video/BV1Fe411o7dU/?share_source=copy_web&vd_source=52aa07429dd1f9ba501f033dded6e8ff
提供了一套不用自己配环境的代码(直接给了envs),如果在下载时出现一些问题可以参考这个。
同时里面也给出了制作自己数据集的方法。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Monodyee/article/detail/101375
推荐阅读
相关标签
  

闽ICP备14008679号