3D Gaussian Splatting (Packaged Python Version)
This repo is the refactored python training and inference code for 3D Gaussian Splatting. Forked from commit a2a91d9093fd791fb01f556fa717f8d9f2cfbdd7. We refactored the original code following the standard Python package structure, while keeping the algorithms used in the code identical to the original version.
Features
- organize the code as a standard Python package
- exposure compensation
- camera and 3DGS parameters joint training
- depth regularization
- local relative depth regularization
- image mask
- integrated gsplat backend
- integrated 2DGS from gsplat
Downstream Projects
- InstantSplat(Refactored): Z. Fan et al., "InstantSplat: Sparse-view Gaussian Splatting in Seconds," CVPR NRI 2024
- reduced-3dgs(Refactored): P. Papantonakis et al., "Reducing the Memory Footprint of 3D Gaussian Splatting," Proceedings of the ACM on Computer Graphics and Interactive Techniques, vol. 7, no. 1, May 2024
- 3dgs-mcmc(Refactored): S. Kheradmand et al., "3D Gaussian Splatting as Markov Chain Monte Carlo," NeurIPS Spotlight Presentation 2024
- lapis-gs(Refactored): Y. Shi et al., "LapisGS: Layered Progressive 3D Gaussian Splatting for Adaptive Streaming," 3DV 2025
- gscompressor(Original): Compress 3DGS scenes by Draco.
- gsvvcompressor(Original): A modular video compression framework for Gaussian Splatting models.
- TrackerSplat(Original): D. Yin et al., "TrackerSplat: Exploiting Point Tracking for Fast and Robust Dynamic 3D Gaussians Reconstruction," SIGGRAPH Asia 2025
Install
Prerequisites
- Pytorch (>= v2.4 recommended)
- CUDA Toolkit (12.4 recommended, match with PyTorch version)
- gsplat
PyPI Install
pip install --upgrade gaussian-splatting
or build latest from source:
pip install wheel setuptools
pip install --upgrade git+https://github.com/yindaheng98/gaussian-splatting.git@master --no-build-isolation
Development Install
git clone --recursive https://github.com/yindaheng98/gaussian-splatting
cd gaussian-splatting
pip install tqdm plyfile tifffile numpy opencv-python pillow open3d
pip install git+https://github.com/nerfstudio-project/gsplat.git
pip install --target . --upgrade . --no-deps
Quick Start
- Download dataset (T&T+DB COLMAP dataset, size 650MB):
wget https://repo-sam.inria.fr/fungraph/3d-gaussian-splatting/datasets/input/tandt_db.zip -P ./data
unzip data/tandt_db.zip -d data/
- Train 3DGS with densification (same with original 3DGS)
python -m gaussian_splatting.train -s data/truck -d output/truck -i 30000 --mode densify
- Render or view 3DGS
python -m gaussian_splatting.render -s data/truck -d output/truck -i 30000 --mode densify
python -m gaussian_splatting.viewer -d output/truck -i 30000
- Joint training 3DGS and camera (load the trained 3DGS)
python -m gaussian_splatting.train -s data/truck -d output/truck-camera -i 30000 --mode camera -l output/truck/point_cloud/iteration_30000/point_cloud.ply
- Render 3DGS with optimized cameras
python -m gaussian_splatting.render -s data/truck -d output/truck-camera -i 30000 --mode camera --load_camera output/truck-camera/cameras.json
💡 This repo does not contain code for creating dataset. If you want to create your own dataset, please refer to InstantSplat or use convert.py.
💡 See .vscode/launch.json for more example. See gaussian_splatting.train and gaussian_splatting.render for full options.
2D Gaussian Splatting (2DGS)
- Train 2DGS with densification
python -m gaussian_splatting.train -s data/truck -d output/truck-2dgs -i 30000 --mode normal-densify --backend gsplat-2dgs
- Render or view 2DGS
python -m gaussian_splatting.render -s data/truck -d output/truck-2dgs -i 30000 --backend gsplat-2dgs
python -m gaussian_splatting.viewer -d output/truck-2dgs -i 30000 --backend gsplat-2dgs
- Joint training 2DGS and camera (load the trained 2DGS)
python -m gaussian_splatting.train -s data/truck -d output/truck-2dgs-camera -i 30000 --mode normal-camera -l output/truck-2dgs/point_cloud/iteration_30000/point_cloud.ply --backend gsplat-2dgs
- Render 2DGS with optimized cameras
python -m gaussian_splatting.render -s data/truck -d output/truck-2dgs-camera -i 30000 --mode camera --load_camera output/truck-2dgs-camera/cameras.json --backend gsplat-2dgs
(Optional) Generate depth maps before training
- Prepare Depth-Anything-V2
git clone https://github.com/DepthAnything/Depth-Anything-V2.git
mkdir checkpoints
wget -O checkpoints/depth_anything_v2_vitl.pth https://huggingface.co/depth-anything/Depth-Anything-V2-Large/resolve/main/depth_anything_v2_vitl.pth?download=true
- Generate depth maps
# (Recommanded) save depth map as floating-point tiff file
python tools/run_depth_anything_v2.py --encoder vitl --img-path data/truck/images --outdir data/truck/depths
# (not Recommanded) save depth map as uint8 png file
python Depth-Anything-V2/run.py --encoder vitl --pred-only --grayscale --img-path data/truck/images --outdir data/truck/depths
API Usage
Gaussian models
GaussianModel is the basic 3DGS model.
from gaussian_splatting import GaussianModel
gaussians = GaussianModel(sh_degree).to(device)
If you want cameras-3DGS joint training, use CameraTrainableGaussianModel, the rendering process is different.
from gaussian_splatting import CameraTrainableGaussianModel
gaussians = CameraTrainableGaussianModel(sh_degree).to(device)
save and load params:
gaussians.save_ply("output/truck/point_cloud/iteration_30000/point_cloud.ply")
gaussians.load_ply("output/truck/point_cloud/iteration_30000/point_cloud.ply")
init 3DGS with sparse point cloud extracted by colmap:
from gaussian_splatting.dataset.colmap import colmap_init
colmap_init(gaussians, "data/truck")
Dataset
Basic colmap dataset:
from gaussian_splatting.dataset.colmap import ColmapCameraDataset, colmap_init
dataset = ColmapCameraDataset("data/truck")
save to JSON and load JSON dataset:
dataset.save_cameras("output/truck/cameras.json")
from gaussian_splatting import JSONCameraDataset
dataset = JSONCameraDataset("output/truck/cameras.json")
Dataset with trainable cameras:
from gaussian_splatting import TrainableCameraDataset
dataset = TrainableCameraDataset("data/truck") # init cameras from colmap
dataset = TrainableCameraDataset.from_json("output/truck/cameras.json") # init cameras from saved json
Inference
for camera in dataset:
out = gaussians(camera)
image = out["render"]
... # compute loss, save image or others
Trainers
gaussian_splatting.trainer contains a series of trainers for optimizing 3DGS models.
Core Trainers
Basic training methods that handle fundamental optimization tasks:
BaseTrainer only optimize the 3DGS parameters, without densification or camera optimization.
from gaussian_splatting.trainer import BaseTrainer
trainer = BaseTrainer(
gaussians,
scene_extent=dataset.scene_extent(),
... # see gaussian_splatting/trainer/base.py for full options
)
BaseDensificationTrainer optimize the 3DGS parameters with densification.
from gaussian_splatting.trainer import BaseDensificationTrainer
trainer = BaseDensificationTrainer(
gaussians,
scene_extent=dataset.scene_extent(),
... # see gaussian_splatting/trainer/densifier/densifier.py for full options
)
BaseCameraTrainer jointly optimize the 3DGS parameters and cameras, without densification.
from gaussian_splatting.trainer import BaseCameraTrainer
trainer = BaseCameraTrainer(
gaussians,
scene_extent=dataset.scene_extent(),
dataset=dataset,
... # see gaussian_splatting/trainer/camera_trainable.py for full options
)
BaseDepthTrainer optimize the 3DGS parameters with depth regularization.
from gaussian_splatting.trainer import BaseDepthTrainer
trainer = BaseDepthTrainer(
gaussians,
scene_extent=dataset.scene_extent(),
... # see gaussian_splatting/trainer/base.py for full options
)
DepthCameraTrainer integrated BaseCameraTrainer with depth regularization.
from gaussian_splatting.trainer import SHLiftTrainer
trainer = SHLiftTrainer(
gaussians,
scene_extent=dataset.scene_extent(),
dataset=dataset,
... # see gaussian_splatting/trainer/sh_lift.py for full options
)
Enhanced Trainers
Gaussian Splatting paper also introduce two tricks "opacity reset" and "lifting SH", they are also included. The basic methods can be integrated with opacity reset and lifting SH. For example:
BaseOpacityResetDensificationTrainer integrated BaseDensificationTrainer with opacity reset.
from gaussian_splatting.trainer import BaseOpacityResetDensificationTrainer
trainer = OpacityResetDensificationTrainer(
gaussians,
scene_extent=dataset.scene_extent(),
... # see gaussian_splatting/trainer/combinations.py for full options
)
DepthOpacityResetDensificationTrainer integrated BaseOpacityResetDensificationTrainer with depth regularization.
from gaussian_splatting.trainer import DepthOpacityResetDensificationTrainer
trainer = DepthOpacityResetDensificationTrainer(
gaussians,
scene_extent=dataset.scene_extent(),
... # see gaussian_splatting/trainer/combinations.py for full options
)
BaseSHLiftOpacityResetDensificationTrainer integrated BaseOpacityResetDensificationTrainer with lifting SH.
from gaussian_splatting.trainer import BaseSHLiftOpacityResetDensificationTrainer
trainer = BaseSHLiftOpacityResetDensificationTrainer(
gaussians,
scene_extent=dataset.scene_extent(),
... # see gaussian_splatting/trainer/combinations.py for full options
)
DepthSHLiftOpacityResetDensificationTrainer integrated DepthOpacityResetDensificationTrainer with lifting SH.
from gaussian_splatting.trainer import DepthSHLiftOpacityResetDensificationTrainer
trainer = DepthSHLiftOpacityResetDensificationTrainer(
gaussians,
scene_extent=dataset.scene_extent(),
... # see gaussian_splatting/trainer/combinations.py for full options
)
Similarly, there are BaseOpacityResetDensificationCameraTrainer, DepthOpacityResetDensificationCameraTrainer, BaseSHLiftOpacityResetDensificationCameraTrainer, DepthSHLiftOpacityResetDensificationCameraTrainer that integrated the above with camera training.
For more, please refer to train.py and trainer/combinations.py.
Training
To use any trainer:
for camera in dataset:
loss, out = trainer.step(camera)
Discussion on Additional Features
Local Relative Depth Regularization
Problem: Global Depth Rescaling Limitation
The default implementation uses DepthAnythingV2 for depth estimation (tools/run_depth_anything_v2.py). These estimated depth maps are then scaled using one global factor per scene (trainer/depth.py or in the github.com/graphdeco-inria/gaussian-splatting/utils/make_depth_scale.py).
However, this approach suffers from local inaccuracies due to limitations inherent in monocular depth predictions.
As demonstrated below, monocular depth estimation frequently introduces local distortions with global rescaling (for instance, people are pouring wine, but the spout is not positioned directly above the wine glass.):
Using globally scaled depth alone results in artifacts and incorrectly placed surfaces during rendering:
Overlaying rendered depth map with DepthAnythingV2-estimated depth map manifests these shortcomings clearly. While background walls and foreground table approximately match ground truth, depth estimates for people remain significantly inaccurate:
Root Cause: Spatial Error Patterns in DepthAnythingV2
Although the output depth estimation from DepthAnythingV2 appears visually plausible when inspected independently (as illustrated in the figure below), local depth scale variations remain substantial.
Therefore, a single global scaling cannot account adequately for these local discrepancies.
Solution: Local relative depth regularization
Considering that DepthAnythingV2 produces relatively accurate local depth relationships, this repo introduces local relative depth regularization strategy. Specifically, the strategy involves:
- Divide the depth map into small overlapping windows.
- Compute scaling and offset corrections individually per window.
- Apply these local corrections to guide model predictions.
Implementation details are provided in the function compute_local_relative_depth_loss in trainer/depth.py.
The resulting improvements are clearly visible, significantly reducing artifacts:
Overlaying it with DepthAnythingV2-estimated depth map:
Local regularization notably improves background alignment (e.g., walls), but some inaccuracies remain for complex foreground shapes such as people. This clearly highlights inherent limitations and persistent spatial error patterns in the monocular DepthAnythingV2 estimations.
3D Gaussian Splatting for Real-Time Radiance Field Rendering
Bernhard Kerbl*, Georgios Kopanas*, Thomas Leimkühler, George Drettakis (* indicates equal contribution)
| Webpage | Full Paper | Video | Other GRAPHDECO Publications | FUNGRAPH project page |
| T&T+DB COLMAP (650MB) | Pre-trained Models (14 GB) | Viewers for Windows (60MB) | Evaluation Images (7 GB) |
This repository contains the official authors implementation associated with the paper "3D Gaussian Splatting for Real-Time Radiance Field Rendering", which can be found here. We further provide the reference images used to create the error metrics reported in the paper, as well as recently created, pre-trained models.
Abstract: Radiance Field methods have recently revolutionized novel-view synthesis of scenes captured with multiple photos or videos. However, achieving high visual quality still requires neural networks that are costly to train and render, while recent faster methods inevitably trade off speed for quality. For unbounded and complete scenes (rather than isolated objects) and 1080p resolution rendering, no current method can achieve real-time display rates. We introduce three key elements that allow us to achieve state-of-the-art visual quality while maintaining competitive training times and importantly allow high-quality real-time (≥ 30 fps) novel-view synthesis at 1080p resolution. First, starting from sparse points produced during camera calibration, we represent the scene with 3D Gaussians that preserve desirable properties of continuous volumetric radiance fields for scene optimization while avoiding unnecessary computation in empty space; Second, we perform interleaved optimization/density control of the 3D Gaussians, notably optimizing anisotropic covariance to achieve an accurate representation of the scene; Third, we develop a fast visibility-aware rendering algorithm that supports anisotropic splatting and both accelerates training and allows realtime rendering. We demonstrate state-of-the-art visual quality and real-time rendering on several established datasets.
BibTeX
@Article{kerbl3Dgaussians,
author = {Kerbl, Bernhard and Kopanas, Georgios and Leimk{\"u}hler, Thomas and Drettakis, George},
title = {3D Gaussian Splatting for Real-Time Radiance Field Rendering},
journal = {ACM Transactions on Graphics},
number = {4},
volume = {42},
month = {July},
year = {2023},
url = {https://repo-sam.inria.fr/fungraph/3d-gaussian-splatting/}
}1 maintainer
Release files for gaussian-splatting 2.10.3
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| gaussian_splatting-2.10.3.tar.gz | 4.5 MB | Details |
Built distributions (wheels)
| File | Reset | |||
|---|---|---|---|---|
| gaussian_splatting-2.10.3-cp312-cp312-win_amd64.whl | CPython 3.12 | CPython 3.12 | Windows x86-64 | Details |
| gaussian_splatting-2.10.3-cp311-cp311-win_amd64.whl | CPython 3.11 | CPython 3.11 | Windows x86-64 | Details |
| gaussian_splatting-2.10.3-cp311-cp311-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl | CPython 3.11 | CPython 3.11 | Linux glibc 2.27+ x86-64, Linux glibc 2.28+ x86-64 | Details |
| gaussian_splatting-2.10.3-cp310-cp310-win_amd64.whl | CPython 3.10 | CPython 3.10 | Windows x86-64 | Details |
| gaussian_splatting-2.10.3-cp310-cp310-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl | CPython 3.10 | CPython 3.10 | Linux glibc 2.27+ x86-64, Linux glibc 2.28+ x86-64 | Details |
Total release size: 22.4 MB
Release files / gaussian_splatting-2.10.3.tar.gz
| Download URL | gaussian_splatting-2.10.3.tar.gz |
|---|---|
| Size | 4.5 MB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
a9f9c11bb12387221284808615e087c5b9e8a01a2b8f3d42a3ab086bf64452ab
|
|
BLAKE2b-256 checksum How to use checksums |
f6636f68d9f14660dc13c2dcdce230472f8030fc7f5bef13f14c02aa9f5d35c0
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Sep 8, 2026.
Transparency logRelease files / gaussian_splatting-2.10.3-cp312-cp312-win_amd64.whl
| Download URL | gaussian_splatting-2.10.3-cp312-cp312-win_amd64.whl |
|---|---|
| Size | 979.9 kB |
| Tags | CPython 3.12 Windows x86-64 |
|
SHA-256 checksum How to use checksums |
75b51fa831d6823bf524d502621173f2c7a2caf41ca8a8f29a9b6088aab16479
|
|
BLAKE2b-256 checksum How to use checksums |
d39d9d53c3f86ef3cdc7adb2bd96c59faac00b81f1b44f909ea3252a74daddf3
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Sep 8, 2026.
Transparency logRelease files / gaussian_splatting-2.10.3-cp311-cp311-win_amd64.whl
| Download URL | gaussian_splatting-2.10.3-cp311-cp311-win_amd64.whl |
|---|---|
| Size | 978.6 kB |
| Tags | CPython 3.11 Windows x86-64 |
|
SHA-256 checksum How to use checksums |
2010638b89e539dda976d006dc9bd9131c50dad0f913b1aac58d68219b79f85c
|
|
BLAKE2b-256 checksum How to use checksums |
37d43bf8fcf1bbf2cf8e146e50f8aaccb37934f50755cc62deb811fc9ab15c97
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Sep 8, 2026.
Transparency logRelease files / gaussian_splatting-2.10.3-cp311-cp311-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl
| Download URL | gaussian_splatting-2.10.3-cp311-cp311-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl |
|---|---|
| Size | 7.5 MB |
| Tags | CPython 3.11 Linux glibc 2.27+ x86-64 Linux glibc 2.28+ x86-64 |
|
SHA-256 checksum How to use checksums |
0d542760f5dc9f4947189d16d13576ad680f7cede1682bf04df3b5409ba9a257
|
|
BLAKE2b-256 checksum How to use checksums |
3286086f6ad30609a7aece2af3a01c29d630e80870cba55bbe9d0d35da994f39
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Sep 8, 2026.
Transparency logRelease files / gaussian_splatting-2.10.3-cp310-cp310-win_amd64.whl
| Download URL | gaussian_splatting-2.10.3-cp310-cp310-win_amd64.whl |
|---|---|
| Size | 976.3 kB |
| Tags | CPython 3.10 Windows x86-64 |
|
SHA-256 checksum How to use checksums |
216859dfdbd6df2443d36df24f6f1cd67b80cfa2f9625db368b46efca5f8e012
|
|
BLAKE2b-256 checksum How to use checksums |
33e1448a315f8a13cbfd5759a6cae21e29885df6fea157ac9bd32c566c008c94
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Sep 8, 2026.
Transparency logRelease files / gaussian_splatting-2.10.3-cp310-cp310-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl
| Download URL | gaussian_splatting-2.10.3-cp310-cp310-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl |
|---|---|
| Size | 7.5 MB |
| Tags | CPython 3.10 Linux glibc 2.27+ x86-64 Linux glibc 2.28+ x86-64 |
|
SHA-256 checksum How to use checksums |
1f6353c68e266223ae69155261ef498228883e7c8c2a12d725135029b99cc81f
|
|
BLAKE2b-256 checksum How to use checksums |
7b803b8b1c97f06dcaf9812c7b96255e1cf3acbb47796b72a99f15c09d28eb0d
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Sep 8, 2026.
Transparency log
