【3DGS】Ubuntu20.04系统搭建3D Gaussian Splatting及可视化环境

有理想的摆烂女 2024-07-18 08:37:02 阅读 77

目录

安装CUDA

下载Gaussian Splatting的源码

创建Gaussian Splatting虚拟环境

下载数据集并训练

模型可视化

更新gcc和g++

更新cmake

安装eigen3.4.0

安装opencv4.9.0

安装远程可视化


安装CUDA

官方说we used 11.8, known issues with 11.6。因此需要确保CUDA版本高于11.8

CUDA官网链接:https://developer.nvidia.com/cuda-toolkit-archive

<code>wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run

sudo sh cuda_11.8.0_520.61.05_linux.run

安装完成后,配置并更新环境变量:

vim ~/.bashrc

#添加路径(此处展示的是默认路径,根据自己的路径来)

export PATH=/usr/local/cuda-11.8/bin${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

source ~/.bashrc

使用下面的命令查看你的CUDA版本:

nvcc -V

下载Gaussian Splatting的源码

注意:最后的--recursive特别重要,不加的话submodules和sibr_viewers中部分东西装不上,影响后续环境配置。

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

创建Gaussian Splatting虚拟环境

使用下面的命令执行,会自动为你创建一个名为 gaussian_splatting 的虚拟环境。并在安装完所需要的依赖库后激活。

conda env create --file environment.yml

conda activate gaussian_splatting

下载数据集并训练

数据集:3D Gaussian Splatting for Real-Time Radiance Field Rendering (inria.fr)。

scenes数据集包含了4个场景的图片,并已经通过convert.py进行colmap转换,可以直接用来做train.py。

我新建了一个data文件夹用来放数据和输出结果,然后可以开始训练了(-s后接数据路径-m后接输出路径):

<code>python train.py -s data/truck/ -m data/truck/output

训练速度真的很快!其中L1是L1损失;PSNR是图像峰值信噪比,单位是dB,数值越大代表MSE越小,MSE越小代表两张图片越接近,失真就越小,因此PSNR值数值越大表示失真越小。

生成的output文件夹结构如下(程序会保存训练7000轮与30000轮时的模型):

模型可视化

更新gcc和g++

<code>cd /usr/bin

sudo rm gcc

sudo ln -s gcc-9 gcc

sudo rm g++

sudo ln -s g++-9 g++

更新cmake

cmake官方列表:Index of /files (cmake.org) (官方文件说recent version, we used 3.24)

#下载

wget https://cmake.org/files/v3.25/cmake-3.25.0-linux-x86_64.tar.gz

#解压

tar -zxvf cmake-3.25.0-linux-x86_64.tar.gz

cmake-3.25.0-Linux-x86_64.tar.gz压缩包里的文件是已经编译过的,解压就可以用!

#将解压出来的包移到 /opt 目录下

sudo mv cmake-3.25.0-linux-x86_64 /opt/cmake-3.25.0

#建立软连接

sudo ln -sf /opt/cmake-3.25.0/bin/* /usr/bin/

查看版本

安装eigen3.4.0

不知道为什么我一直进不去eigen官网https://eigen.tuxfamily.org/index.php?title=Main_Page

后面找到了这个下载链接:发布 · libeigen / eigen · GitLab

(写在前面:555安装这个是因为make opencv的时候提示egien版本不够新,所以就用源码安装了一下。但是呢后续安装colmap的时候又提示eigen版本过高咋咋的,所以我又把它卸载了。大家可以根据自己的情况试试,我也是非常的迷惑。)

<code>#下载

wget https://gitlab.com/libeigen/eigen/-/archive/3.4.0/eigen-3.4.0.zip

#解压

unzip eigen-3.4.0

#进入文件夹

cd eigen-3.4.0

#创建build文件夹并进入

mkdir build

cd build

#cmake构建

cmake..

make

#安装

sudo make install

#复制文件夹(方便以后查找)

sudo cp -r /usr/local/include/eigen3 /usr/include

#查看当前版本

pkg-config --modversion eigen3

安装opencv4.9.0

官网:Releases - OpenCV

github地址:https://github.com/opencv

#下载opencv

git clone https://github.com/opencv/opencv.git

#进入源码文件夹

cd opencv

#下载opencv_contrib

git clone https://github.com/opencv/opencv_contrib.git

安装依赖

sudo apt-get update

sudo apt-get upgrade

sudo apt-get install cmake gfortran

sudo apt-get install python3 python3-dev python3-numpy

sudo apt-get install libjpeg-dev libtiff-dev libgif-dev

sudo apt-get install libgstreamer1.0-dev gstreamer1.0-gtk3

sudo apt-get install libgstreamer-plugins-base1.0-dev gstreamer1.0-gl

sudo apt-get install libavcodec-dev libavformat-dev libswscale-dev

sudo apt-get install libgtk2.0-dev libcanberra-gtk*

sudo apt-get install libxvidcore-dev libx264-dev libgtk-3-dev

sudo apt-get install libtbb2 libtbb-dev libdc1394-22-dev libv4l-dev

sudo apt-get install libopenblas-dev libatlas-base-dev libblas-dev

sudo apt-get install protobuf-compiler

 编译安装

#在opencv文件夹下

mkdir build

cd build

cmake -D WITH_TBB=ON -D WITH_EIGEN=ON -D OPENCV_GENERATE_PKGCONFIG=ON -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local -D OPENCV_EXTRA_MODULES_PATH=../opencv_contrib/modules -D OPENCV_ENABLE_NONFREE=ON ..

sudo make -j8

sudo make install

环境配置

#修改文件

sudo vim /etc/ld.so.conf

#在文件中输入,并保存退出

/usr/local/lib

#执行命令

sudo ldconfig

#修改文件

sudo vim /etc/bash.bashrc

#在文件中输入,并保存退出

PKG_CONFIG_PATH=$PKG_CONFIG_PATH:/usr/local/lib/pkgconfig

export PKG_CONFIG_PATH

#执行命令

source /etc/bash.bashrc

查看版本号:pkg-config --modversion opencv4

安装远程可视化

<code>#安装依赖

sudo apt install -y libglew-dev libassimp-dev libboost-all-dev libgtk-3-dev libopencv-dev libglfw3-dev libavdevice-dev libavcodec-dev libeigen3-dev libxxf86vm-dev libembree-dev

#进入文件夹

cd SIBR_viewers

#因为我是20.04版本,所以还要加一句

#22.04版本不需要加

git checkout fossa_compatibility

#构建安装

cmake -Bbuild . -DCMAKE_BUILD_TYPE=Release

cmake --build build -j24 --target install

在这个过程中出现了非常多的问题hai

大多数都是版本不一致造成的(所以大家可能不会遇到)

我的解决办法就是遇到一个解决一个 耗时很久

终于在众多编译的警告下安装成功(看的真是心脏怦怦跳)

在SIBR_viewers/install/bin文件夹内如下:

接下来就是见证奇迹的时刻(我用的MobaXterm):

<code>./SIBR_gaussianViewer_app -m /home/lyc/3dgs-code/gaussian-splatting/data/truck/output

自采数据集实验

【3DGS】Ubuntu系统下完成3D Gaussian Splatting自采数据集的重建、评估及可视化-CSDN博客



声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。