index.md

GaussianSplatting.jl

Gaussian Splatting algorithm in pure Julia.

Requirements

• Julia 1.10 or higher.
• AMDGPU.jl or CUDA.jl or Metal.jl capable machine.

Install

Add GaussianSplatting.jl package:

] add https://github.com/JuliaNeuralGraphics/GaussianSplatting.jl.git

Usage

GaussianSplatting.jl comes with a GUI application to train & view the gaussians.

1. Add necessary packages:

   ] add AMDGPU # for AMD GPU
   ] add CUDA   # for Nvidia GPU
   ] add Metal  # for Apple GPU

2. Run:

   julia> using AMDGPU; kab = ROCBackend()  # for AMD GPU
   julia> using CUDA; kab = CUDABackend()   # for Nvidia GPU
   julia> using Metal; kab = MetalBackend() # for Apple GPU
   julia> GaussianSplatting.gui(kab, "path-to-colmap-dataset-directory"; scale=1)

Viewer mode

Once you've trained a model and saved it to .bson file you can open it in a viewer-only mode by providing its path.

julia> GaussianSplatting.gui(kab, "path-to-checkpoint.bson")

Alternative, you can load a checkpoint in a training mode (see Usage section) using "Save/Load" tab.

References

• 3D Gaussian Splatting for Real-Time Radiance Field Rendering: https://arxiv.org/abs/2308.04079
• gsplat: https://github.com/nerfstudio-project/gsplat