部署高斯喷射项目gaussian-splatting

硬件要求

• 支持 CUDA 的 GPU，具有 7.0+ 的计算能力
• 24 GB VRAM

软件要求

• Conda
• 用于 PyTorch 扩展的 C++ 编译器（Visual Studio 2019）

• CUDA SDK 11 for PyTorch 扩展，在 Visual Studio 之后安装
• C++ 编译器和 CUDA SDK 必须兼容

拉取源码

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

安装

SET DISTUTILS_USE_SDK=1
conda env create --file environment.yml
conda activate gaussian_splatting

如果报错

C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.8/include\crt/host_config.h(231): fatal error C1083: 无法打开包括文件: “crtdefs.h”: No such file or directory
error: command 'C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.8\\bin\\nvcc.exe' failed with exit code 2 

则执行下面命令

conda install -c anaconda vs2019_win-64
conda env update --file environment.yml

训练

要运行优化器，只需使用下面命令

python train.py -s <path to COLMAP or NeRF Synthetic dataset>

COLMAP合成数据集结构：

 <location>
|---images
|   |---<image 0>
|   |---<image 1>
|   |---...
|---sparse
    |---0
        |---cameras.bin
        |---images.bin
        |---points3D.bin

下载项目提供的SfM数据集

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

查看

下载编译好的查看器并解压

https://repo-sam.inria.fr/fungraph/3d-gaussian-splatting/binaries/viewers.ziphttps://repo-sam.inria.fr/fungraph/3d-gaussian-splatting/binaries/viewers.zip运行实时查看器

cd gaussian-splatting\viewers\bin
SIBR_gaussianViewer_app -m <path to trained model>