3D gaussian splatting 高斯的使用方法及训练

github链接:https://github.com/graphdeco-inria/gaussian-splatting

!!github上有视频教学链接,十分详细,可以直接跟着视频学习。

首先我们需要去github上将项目clone或者下载到我们的电脑上。但是在开始运行之前我们通常需要下载一些软件,并把他们加入到环境变量之中去。(最好是使用命令clone,我是直接下压缩包载的,会缺少两个库。clone不会出现这个bug)

# SSH
    git clone [git@github.com](mailto:git@github.com):graphdeco-inria/gaussian-splatting.git --recursive
# HTTPS
    git clone [https://github.com/graphdeco-inria/gaussian-splatting](https://github.com/graphdeco-inria/gaussian-splatting) --recursive

运行的环境要求:

  • CUDA 11.x 和对应的pytorch版本
  • 论文中说明需要24G的显存,实际测试下来也没有那么高,可能是我的图片大小并不大
  • cmake,论文中使用的版本是3.24

需要安装的软件:

  • annaconda:这是给项目提供了运行的环境
  • git:用于下载项目代码(这个可以不用)
  • VS:论文中使用了Visual studio 2019(使用vscode也一样)
  • Colmap: 下载colmap,并加入环境变量
  • ImageMagick:是一个免费开源的跨平台软件套件用于显示、创建、转换、修改和编辑光栅图像。
  • FFMPEG:用于制作数据集从视频中抽取帧

注:如果是Windwos系统的话,需要在项目的根目录运行这行代码:

SET DISTUTILS_USE_SDK=1 # Windows only

如果之前有pytorch的环境的话,基本上不需要再安装环境,如果没有ngp环境,可以使用以下conda命令进行环境的创建和搭建:

conda env create --file environment.yml
conda activate gaussian_splatting

如果是想要训练自己的数据,需要经历如下几个步骤:

(1)数据准备
假设我拍摄的是一段人脸的视频(face.mp4),首先我们需要构建一些文件目录:data → face → face.mp4, data → face → input;

我们进入到face目录下:

ffmpeg -i face.mp4 -qscale:v 1 -qmin 1 -vf fps=10 %04d.jpg

然后将所有生成的图片都移动到input文件夹中,然后我们切换到根目录并激活虚拟环境。

接下来我们就要使用脚本进行数据准备了,这就是位姿的准备了实际上就是使用colmap进行位姿恢复。

python convert.py -s <location> [--resize] #If not resizing, ImageMagick is not needed

# 我使用的命令
python convert.py -s ./data/face

(2)训练

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

# 我使用的命令
python train.py -s ./data/face

如果你遇到了下面的问题,就按照下面的方法来:

在运行这段代码的时候我发现,他会报一个这样的错:

from diff_gaussian_rasterization import GaussianRasterizationSettings, GaussianRasterizer
ModuleNotFoundError: No module named 'diff_gaussian_rasterization'

这表明我们的项目中和环境中都是没有这个库的,于是我尝试了使用pip、apt和conda的安装,都表示没有。经过查找发现在作者的github下有一个仓库中包含这个库:https://github.com/graphdeco-inria/diff-gaussian-rasterization

那么我们可能需要手动安装这个库:
先下载这个库,然后运行下列命令:

sudo apt-get install libglm-dev 
python setup.py build
python setup.py install 
pip install plyfile

这是一个光栅化引擎。不出所料,又缺少了一个库simple_knn._C
然后,在查找bug的过程中,我并没有发现这个库,作者也没有提供。

于是,我尝试了另一种方法。由于之前是直接下载项目解压的,可能出了一些问题,于是我重新使用git clone了一下。并且按照github的要求,配置了一个新的环境:

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

conda env create --file environment.yml

训练总耗时:10分钟26秒

训练过后,我们可以在项目的根目录生成一个output文件,并且会生成一个随机的文件夹名字,里面包含四个文件。我们可以将这个文件夹名改成我们自己想要的名字。

(3)查看模型并渲染

Ubuntu 22.4:

# Dependencies

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

# Project setup

cd SIBR_viewers

cmake -Bbuild . -DCMAKE_BUILD_TYPE=Release # add -G Ninja to build faster

cmake --build build -j24 --target install

如果是Ubuntu的话,这个软件将会保存到:/gaussian-splatting/SIBR_viewers/install/bin/

Windows:

cd SIBR_viewers

cmake -Bbuild .

cmake --build build --target install --config RelWithDebInfo

你需要去下载一个软件,用于查看渲染结果,这里给出链接:https://repo-sam.inria.fr/fungraph/3d-gaussian-splatting/binaries/viewers.zip

然后运行:

cd install/bin

./SIBR_gaussianViewer_app -m ~/Documents/gaussian-splatting/output/face

正如论文所描述的3D高斯就是由一个个的椭球组成的。

这是点云的效果。

这是渲染效果。

我们可以通过awsdqr来控制其在水平方向上的移动,通过控制ijkluo来控制视角的变换。

该算法还是非常清晰的,效果挺好的。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,732评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,496评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,264评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,807评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,806评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,675评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,029评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,683评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 41,704评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,666评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,773评论 1 332
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,413评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,016评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,978评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,204评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,083评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,503评论 2 343

推荐阅读更多精彩内容