Gaussian Splatting代码安装部署(windows)

项目网址 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

下载源码

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

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

2.创建虚拟环境

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

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

但是要注意他用的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

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

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

下载conda和vs2019联动的插件

conda install -c anaconda vs2019_win-64

下载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

安装其他依赖

pip install submodules\diff-gaussian-rasterization
pip install submodules\simple-knn
pip install plyfile
pip install tqdm

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

二、跑提供的模型

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

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

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

python train.py -s data/tandt/train

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

cd SIBR_viewers
cmake -Bbuild .
cmake --build build --target install --config RelWithDebInfo

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

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

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

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

我这里是

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

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

点击红色方框的下拉框,三个选项一次是渲染后的效果,点云和高斯(其中高斯比较卡),在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),如果在下载时出现一些问题可以参考这个。
同时里面也给出了制作自己数据集的方法。

  • 34
    点赞
  • 49
    收藏
    觉得还不错? 一键收藏
  • 11
    评论
评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值