
gsplat 5 分钟上手3D 高斯泼溅重建 3D 场景显存省 4 倍还能多卡加速【免费下载链接】gsplatCUDA accelerated rasterization of gaussian splatting项目地址: https://gitcode.com/GitHub_Trending/gs/gsplat跑满 3 万步迭代一组照片就能变成浏览器里可旋转的 3D 场景显存占用比官方 3D 高斯泼溅实现低最多 4 倍。gsplat 是一个 CUDA 加速的高斯泼溅光栅化开源库解决照片重建真实感 3D 场景这件事适合已有 PyTorch CUDA 环境、想自己搭重建管线的开发者。60 秒跑通4 条命令看到渲染结果开始前只需要一样东西装好 CUDA 的 PyTorch。下面的演示不需要任何数据集用 2000 个高斯去拟合一张图几秒到几分钟就能收敛pip install gsplat git clone https://gitcode.com/GitHub_Trending/gs/gsplat cd gsplat/examples pip install -r requirements.txt python image_fitting.py第一条命令装库首次 import 时会 JIT 编译 CUDA 代码等几分钟image_fitting.py默认用内置的红蓝象限图当目标跑起来后终端持续打印 loss能看到拟合效果逐步稳定下来。完整场景训练收敛起来是什么样看官方 garden 场景的这个过程左边是随机初始化右边是训练后期。核心机制高斯泼溅到底在泼什么把场景想象成一堆彩色烟雾团——没有网格只有几百万个 3D 高斯每个带位置、形状、朝向、不透明度、颜色。颜色用球谐函数表达所以换个视角看颜色会变渲染是 splatting把高斯投影到屏幕上前到后混合比光线行进快得多全程可微像素误差直接反传回每个高斯训练中还会稠密化糊的地方补高斯冗余的删掉default / MCMC 两种策略按任务展开端到端从一组照片重建可旋转的 3D 场景这个任务解决最常见的场景你有一个物体或房间的拍摄照片加 COLMAP 跑出的位姿想要一个能环视的 3D 场景。cd examples python datasets/download_dataset.py # Mip-NeRF 360 → data/360_v2/ python simple_trainer.py mcmcexamples/simple_trainer.py 默认读 garden 场景训练 3 万步在第 7000、30000 步自动存 checkpoint 并写评估指标同时在本机 8080 端口开一个 web viewer。训练中途就能打开localhost:8080实时旋转查看不用等训练结束。怎么验证跑成功results/garden/stats/下会出现val*.json对比官方 3DGS 论文 garden 的 PSNR26.96浏览器里场景结构完整、没有明显漂浮噪点就算过关。鱼眼、滚动快门相机不用预校正的用法这个任务解决非常规镜头针孔畸变、鱼眼、f-theta、滚动快门。关键是原始图像不用先做去畸变——只要有 COLMAP 标定好的畸变参数投影在 CUDA kernel 里直接算。python simple_trainer.py mcmc --with_ut --with_eval3d --data_dir data/zipnerf/londonzipnerf数据就是带强畸变的实拍图专门用来验证这条路注意 3DGUT 路径只支持 MCMC 策略。参数细节radial_coeffs、camera_modelfisheye、rolling_shutter等在 docs/3dgut.md 里有完整列表。怎么验证训练完用python simple_viewer_3dgut.py --ckpt checkpoint 路径查看viewer 里实时调畸变参数渲染应和原图一致。显存压到 1/4 的设置3 个关键开关这个任务解决场景大、照片多显存爆掉的问题。--packedpacked 格式光栅化官方实测比官方 3DGS 实现省最多 4 倍显存、快 15%单次前向略慢训练整体更快--data_factor 8训练图降采样默认 4显存吃紧就翻倍多卡等效 batch ×4所以步数同步缩到 1/4CUDA_VISIBLE_DEVICES0,1,2,3 python simple_trainer.py default \ --packed --steps_scaler 0.25 --data_dir data/360_v2/garden另外rasterization()支持任意批量渲染1 万个高斯时1 个 batch 前向 0.37ms64 个 batch 也只要 3.7ms显存随 batch 线性增长适合离线批量出图。怎么验证训练时nvidia-smi显存有富余4 卡跑完的总耗时约为单卡的 1/4。⚡实时查看器训练完不用等1 条命令看结果这个任务解决训练中途想看效果或加载别处导出的 .ply 文件。python simple_viewer.py --ckpt results/garden/ckpts/ckpt_29999_rank0.pt --port 8080也支持--ply加载标准 3DGS 导出文件--scene_grid 3可以把场景平铺成 3×3 网格快速扫视全局。怎么验证浏览器打开 8080 端口场景可拖拽旋转、帧率稳定不卡。高频问题Q首次运行特别慢或报 CUDA 编译错误PyPI 版首次运行 JIT 编译等几分钟属正常源码安装需要 torch 与 CUDA 版本匹配Windows 编译问题看docs/INSTALL_WIN.md。Q训练 loss 不降或降不下去九成是数据问题默认init_typesfm需要数据目录里有sparse/0点云和正确的位姿再检查--data_factor是否开太大导致图太糊。Q显存 OOM 怎么办加--packed并把data_factor翻倍MCMC 用户可用--strategy.cap-max限制高斯数量官方基准用 100 万还不行就降--sh_degree。Q重建发糊、有伪影或透明窟窿先看 val PSNR明显偏低是位姿问题可试--pose_opt联合优化相机透明窟窿是训练图有透明背景加--random_bkgd随机背景抑制透明。Qviewer 打不开 / 8080 端口被占用--port换个端口纯离线训练不想开 viewer 就加--disable_viewer评估指标仍会照常落盘。跑通上面 garden 场景后拿你自己的一组照片先跑一遍 COLMAP再试试--with_ut在畸变数据上训练对比校正前后两张图就知道 3DGUT 值不值得上了。【免费下载链接】gsplatCUDA accelerated rasterization of gaussian splatting项目地址: https://gitcode.com/GitHub_Trending/gs/gsplat创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考