Bernhard Kerbl*, Andreas Meuleman*, Georgios Kopanas, Michael Wimmer, Alexandre Lanvin, George Drettakis (* indicates equal contribution)
This repository contains the official authors' implementation associated with the paper "A Hierarchical 3D Gaussian Representation for Real-Time Rendering of Very Large Datasets". We explain the different steps required to run our algorithm. We use a "toy example" of 1500 images organized in 2 chunks to illustrate each step of the method and facilitate reproduction. The full datasets presented in the paper will be released as soon as the data protection process is completed (please stay tuned).
Bibliography:
@Article{hierarchicalgaussians24,
author = {Kerbl, Bernhard and Meuleman, Andreas and Kopanas, Georgios and Wimmer, Michael and Lanvin, Alexandre and Drettakis, George},
title = {A Hierarchical 3D Gaussian Representation for Real-Time Rendering of Very Large Datasets},
journal = {ACM Transactions on Graphics},
number = {4},
volume = {43},
month = {July},
year = {2024},
url = {https://repo-sam.inria.fr/fungraph/hierarchical-3d-gaussians/}
}
Please note that the code release is currently in alpha. We intend to provide fixes for issues that are experienced by users, due to difficulties with setups and/or environments that we did not test on. The below steps were successfully tested on Windows and Ubuntu 22. We appreciate the documentation of issues by users and will try to address them. Furthermore, there are several points that we will integrate in the coming weeks:
- Datasets: We will add links for large-scale datasets that are currently undergoing auditing.
- Windows binaries: Once we have sufficiently tested them, we will add pre-compiled binaries for the viewers on Windows.
- Direct conversion of legacy 3DGS models: we are testing the conversion of scenes trained with vanilla 3DGS to hierarchical models. Once the quality is assured and we have concluded testing, we will document the necessary steps to do so.
- Streaming from disk: currently, data is streamed on-demand to the GPU, however, the viewed dataset must fit into memory. This can become prohibitive in the hierarchy merger and real-time viewer. We will adapt the code to allow dynamic streaming from disk soon.
- Reduce real-time viewer resource usage: the storage configuration for the real-time viewer is unoptimized, and so is the speed. Users can define a VRAM budget for the scene, but it is not used as efficiently as it could be. We will iterate towards making sure that higher quality settings can be achieved with lower budgets and better framerates. We will try to make the budget so that it effectively limits the total application VRAM, including framebuffer structs.
Make sure to clone the repo using --recursive
:
git clone https://github.com/graphdeco-inria/hierarchical-3d-gaussians.git --recursive
cd hierarchical-3d-gaussians
We tested on Ubuntu 22.04 and Windows 11 using the following:
- CMake 3.22.1
- gcc/g++ 11.4.0 or Visual Studio 2019
- CUDA (11.8, 12.1 or 12.5)
- COLMAP 3.9.1 (for preprocessing only). Linux: build from source. Windows: add the path to the COLMAP.bat directory to the PATH environment variable.
conda create -n hierarchical_3d_gaussians python=3.12 -y
conda activate hierarchical_3d_gaussians
# Replace cu121 with cu118 if using CUDA 11.x
pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu121
pip install -r requirements.txt
To enable depth loss, download the model weights of one of these methods:
- Depth Anything V2 (suggested): download from Depth-Anything-V2-Large and place it under
submodules/Depth-Anything-V2/checkpoints/
. - DPT (used in the paper): download from dpt_large-midas-2f21e586.pt and place it under
submodules/DPT/weights/
.
cd submodules/gaussianhierarchy
cmake . -B build -DCMAKE_BUILD_TYPE=Release
cmake --build build -j --config Release
cd ../..
For Ubuntu 22.04, install dependencies:
sudo apt install -y cmake libglew-dev libassimp-dev libboost-all-dev libgtk-3-dev libopencv-dev libglfw3-dev libavdevice-dev libavcodec-dev libeigen3-dev libxxf86vm-dev libembree-dev
Clone the hierarchy viewer and build:
cd SIBR_viewers
git clone https://github.com/graphdeco-inria/hierarchy-viewer.git src/projects/hierarchyviewer
cmake . -B build -DCMAKE_BUILD_TYPE=Release -DBUILD_IBR_HIERARCHYVIEWER=ON -DBUILD_IBR_ULR=OFF -DBUILD_IBR_DATASET_TOOLS=OFF -DBUILD_IBR_GAUSSIANVIEWER=OFF
cmake --build build -j --target install --config Release
Our method has two main stages: Reconstruction, that takes a (usually large) set of images as input and outputs a "merged hierarchy", and Runtime, that displays the full hierarchy in real-time.
Reconstruction has two main steps: 1) Preprocessing the input images and 2) Optimization. We present these in detail next. For each step we have automatic scripts that perform all the required steps, and we also provide details about the individual components.
To get started, prepare a dataset or download and extract the toy example.
The dataset should have sorted images in a folder per camera in ${DATASET_DIR}/inputs/images/
and optional masks (with .png
extension) in ${DATASET_DIR}/inputs/masks/
. Masks will be multiplied to the input images and renderings before computing loss.
You can also work from our full scenes. As we provide them calibrated and subdivided, you may skip to Generate monocular depth maps. The datasets:
In the following, replace ${DATASET_DIR}
with the path to your dataset or set DATASET_DIR:
# Bash:
DATASET_DIR=<Path to your dataset>
# PowerShell:
${DATASET_DIR} = "<Path to your dataset>"
To skip the reconstruction and only display scenes, download pretrained hierarchies and scaffolds, place them under
${DATASET_DIR}/output/
and follow the viewer instructions. The pretrained hierarchies:
As in 3dgs we need calibrated cameras and a point cloud to train our hierarchies on.
The first step is to generate a "global colmap". The following command uses COLMAP's hierarchical mapper, rectify images and masks, and align and scale the sparse reconstruction to facilitate subdivision.
python preprocess/generate_colmap.py --project_dir ${DATASET_DIR}
Using calibrated images
If your dataset already has COLMAP (with 2D and 3D SfM points) and rectified images, they should be placed under ${DATASET_DIR}/camera_calibration/rectified
. As they still need alignment, run:
python preprocess/auto_reorient.py --input_path ${DATASET_DIR}/camera_calibration/rectified/sparse --output_path ${DATASET_DIR}/camera_calibration/aligned/sparse/0
This step takes ~ 47 minutes on our example dataset using a RTX A6000, more details on each steps of the script here.
Once the "global colmap" generated, it should be split into chunks. We also run a per-chunk bundle adjustment as COLMAP's hierarchical mapper is faster but less accurate (if your global colmap is accurate, you can skip this time consuming step with --skip_bundle_adjustment
).
python preprocess/generate_chunks.py --project_dir ${DATASET_DIR}
This step takes ~ 95 minutes on our example dataset using a RTX A6000, more details on each steps of the script here.
note that by using
--use_slurm
you can refine the chunks in parallel, remember to set your slurm parameters inpreprocess/prepare_chunks.slurm
(gpu, account, etc ...).
In order to use depth regularization when training each chunks, depth maps must be generated for each rectified image. Then, depth scaling parameters needs to be computed as well, these two steps can be done using:
python preprocess/generate_depth.py --project_dir ${DATASET_DIR}
Now you should have the following file structure, it is required for the training part:
project
└── camera_calibration
├── aligned
│ └── sparse/0
│ ├── images.bin
│ ├── cameras.bin
│ └── points3D.bin
├── chunks
│ ├── 0_0
│ └── 0_1
│ .
│ .
│ .
│ └── m_n
│ ├── center.txt
│ ├── extent.txt
│ └── sparse/0
│ ├── cameras.bin
│ ├── images.bin
│ ├── points3d.bin
│ └── depth_params.json
└── rectified
├── images
├── depths
└── masks
The scene training process is divided into five steps; 1) we first train a global, coarse 3D Gaussian splatting scene ("the scaffold"), then 2) train each chunk independently in parallel, 3) build the hierarchy, 4) optimize the hierarchy in each chunk and finally 5) consolidate the chunks to create the final hierarchy.
Make sure that you correctly set up your environment and built the hierarchy merger/creator
The full_train.py
script performs all these steps to train a hierarchy from a preprocessed scene. While training, the progress can be visualized with the original 3DGS remote viewer (build instructions).
python scripts/full_train.py --project_dir ${DATASET_DIR}
Command Line Arguments
Input aligned colmap.
Path to rectified images.
Path to rectified depths.
Path to rectified masks.
Path to input chunks folder.
Name the conda env you created earlier.
Path to output dir.
Flag to enable parallel training using slurm (False
by default).
note that by using
--use_slurm
, chunks will be trained in parallel, to exploit e.g. multi-GPU setups. To control the process, remember to set your slurm parameters incoarse_train.slurm
,consolidate.slurm
andtrain_chunk.slurm
(gpu, account, etc ...)
This step takes ~ 171 minutes on our example dataset using a RTX A6000, more details on each steps of the script here.
The real-time viewer is based on SIBR, similar to original 3DGS. For setup, please see here
The hierarchical real-time viewer is used to vizualize our trained hierarchies. It has a top view
that displays the structure from motion point could as well as the input calibrated cameras in green. The hierarchy chunks are also displayed in a wireframe mode.
After installing the viewers, you may run the compiled SIBR_gaussianHierarchyViewer_app in <SIBR install dir>/bin/
. Controls are described here.
If not a lot of VRAM is available, add --budget <Budget for the parameters in MB>
(by default set to 16000, assuming at least 16 GB of VRAM). Note that this only defines the budget for the SCENE representation. Rendering will require some additional VRAM (up to 1.5 GB) for framebuffer structs. Note that the real-time renderer assumes that CUDA/OpenGL Interop is available on your system (see the original 3DGS documentation for more details).
The interface includes a field for tau (size limit)
which defines the desired granularity setting. Note that tau = 0
will try to render the complete dataset (all leaf nodes). If the granularity setting exceeds the available VRAM budget, instead of running out of memory, the viewer will auto-regulate and raise the granularity until the scene can fit inside the defined VRAM budget.
SIBR_viewers/install/bin/SIBR_gaussianHierarchyViewer_app --path ${DATASET_DIR}/camera_calibration/aligned --scaffold ${DATASET_DIR}/output/scaffold/point_cloud/iteration_30000 --model-path ${DATASET_DIR}/output/merged.hier --images-path ${DATASET_DIR}/camera_calibration/rectified/images
Command Line Arguments for Real-Time Viewer
Path to a trained hierarchy.
Specifies which of state to load if multiple are available. Defaults to latest available iteration.
Argument to override model's path to source dataset.
Takes two space separated numbers to define the resolution at which real-time rendering occurs, 1200
width by default. Note that to enforce an aspect that differs from the input images, you need --force-aspect-ratio
too.
Path to rectified input images to be viewed in the top view.
Index of CUDA device to use for rasterization if multiple are available, 0
by default.
Amount of VRAM memory that may be used for the hierarchical 3DGS scene representation.
note that in our experiments we used colmap 3.9.1 with cuda support
the parameters of each colmap commands as well as our scripts are the ones we used in the example dataset.
More details on these parameters can be found here
-
Create a
project
folder and create the required folders to have the following file structure:project ├── camera_calibration │ ├── aligned │ ├── rectified │ └── unrectified └── output
-
Generate a
database.db
in theunrectified
subfolder by extracting features from images:
Input image folder should be organised by subfolders per camera.cd project/unrectified colmap feature_extractor --database_path database.db --image_path <path to images> --ImageReader.single_camera_per_folder 1 --ImageReader.default_focal_length_factor 0.5 --ImageReader.camera_model OPENCV
-
Create a custom
matching.txt
file using:cd hierarchical_3d_gaussians python preprocess/make_colmap_custom_matcher.py --image_path <path to images> --output_path <matching.txt file path>
<matching.txt file path>
will contain pairs of camera indices that are close using the image order and the gps data when it is available.Command Line Arguments
--n_gps_neighbours
Number of closest neighbors to add if gps data is available (
25
by default).--n_loop_closure_match_per_view
Number of matches to add to each camera provided in [loop_matches], it is not used when no custom [loop_matches] is provided (
5
by default).--image_path
Path to input
images
folder.--output_path
Path to output file with the image pairs to match.
--n_seq_matches_per_view
For each image, match with the next [n_seq_matches_per_view] images of the other cameras (
0
by default).--n_quad_matches_per_view
For each image, add [n_quad_matches_per_view] matches that will range from
$image$ to$image+2^{\text{n_quad_matches_per_view}}$ images of the other cameras (10
by default).--loop_matches
Custom matches that can be added by hand (
None
by default).
-
The previously created
matching.txt
file will be used with the feature matching:cd ${DATASET_DIR}/unrectified colmap matches_importer --database_path <database.db> --match_list_path <matching.txt file path>
-
Launch the
hierarchical mapper
to create the scene colmap:note that this step will take a lot more time than the previous steps, it took ~39 minutes on the example dataset.
colmap hierarchical_mapper --database_path <database.db> --image_path <path to images> --output_path <sparse> --Mapper.ba_global_function_tolerance=0.000001
-
Remove floating cameras and feature points that don't have sfm points, to make the colmap lighter:
cd hierarchical_3d_gaussians python preprocess/simplify_images.py --base_dir ${DATASET_DIR}/unrectified/sparse/0
Command Line Arguments
--base_dir
Path to a colmap folder having
images
,cameras
andpoints3D
files.--mult_min_dist
Points at distance > [mult_min_dist]*median_dist_neighbors are removed (
10
by default).--model_type
Colmap model type, can be either
bin
ortxt
(bin
by default).
-
Undistort calibrated cameras, resulting images will be used during training:
cd ${DATASET_DIR} colmap image_undistorter --image_path <path to images> --input_path <unrectified/sparse/0> --output_path <rectified> --output_type COLMAP --max_image_size 2048
If alpha masks are used, they should be undistorted the same way as images. Please find instructions on how to do it in the
generate_colmap.py
script. -
Align and scale the rectified colmap:
The rectified colmap is aligned and scaled to be metric so that it can be easily cut into chunks later.cd hierarchical_3d_gaussians python preprocess/auto_reorient.py --input_path <project_dir/rectified/sparse> --output_path <project_dir/aligned>
Command Line Arguments
--input_path
Path to input colmap dir.
--output_path
Path to output colmap dir.
--upscale
Custom upscaling factor, is automatically computed if not set (
0
by default).--target_med_dist
Median distance of all calibrated cameras to their 3D points for the scene to be roughly metric, experimentally found. Ignored if
upscale
is set (20
by default)--model_type
Colmap model type, can be either
bin
ortxt
(bin
by default).
The last preprocessing step is to divide the colmap into chunks, each chunk will have its own colmap that will be refined with two rounds of bundle adjustment and triangulation:
-
Cut the calibration under
project/camera_calibration/aligned
into chunks, each chunk has its own colmap:python preprocess/make_chunk.py --base_dir <project/aligned/sparse/0> --images_dir <project/rectified/images> --output_path <project/raw_chunks>
Command Line Arguments
Path to input colmap dir.
Path to rectified images.
Chunks are cubes of extents [chunk_size] (
100
by default).Padding for the global bounding box (
0.2
by default).Threshold to discard blurry images: if their laplacians are < mean - lapla_thresh * std (
1
by default).Min nb of cameras in each chunk (
100
by default).Max nb of cameras in each chunk (
2000
by default).Path to output chunks folder.
Add cameras that are far from the chunk (
False
by default).Colmap model type, can be either
bin
ortxt
(bin
by default).
-
Refine each chunk by applying two rounds of
triangulation
andbundle adjustment
:## do this for each chunk python preprocess/prepare_chunk.py --raw_chunk <path to raw chunk> --out_chunk <path to output chunk> --images_dir <project/rectified/images> --depths_dir <project/rectified/depths> --preprocess_dir <path to hierarchical_gaussians/preprocess_dir>
Command Line Arguments
Path to repo/preprocess.
Path to rectified images.
Path to depths.
Path to the unrefined chunk.
Path to the output refined chunk.
Make sure to have the depth estimator weights.
-
- Using Depth Anything V2 (prefered):
cd submodules/Depth-Anything-V2 python run.py --encoder vitl --pred-only --grayscale --img-path [path_to_input_images_dir] --outdir [path_to_output_depth_dir]
- Using DPT:
cd submodules/DPT python run_monodepth.py -t dpt_large -i [path_to_input_images_dir] -o [path_to_output_depth_dir]
- Using Depth Anything V2 (prefered):
-
this file will be used for the depth regularization for single chunk training. It needs to be generated for each chunk.
cd ../../ python preprocess/make_depth_scale.py --base_dir [path to colmap] --depths_dir [path to output depth dir]
Make sure that you correctly set up repositories and environments
-
Coarse optimization
To allow consistent training of all chunks, we create a basic scaffold and skybox for all ensuing steps:python train_coarse.py -s <path to project/aligned> -i <../rectified/images> --skybox_num 100000 --position_lr_init 0.00016 --position_lr_final 0.0000016 --model_path <path to output scaffold>
-
Single chunk training
It is recommended to train using depth regularization to have better results, especially if your scene contains textureless surfaces such as roads. Make sure you generated depth mapspython -u train_single.py -s [project/chunks/chunk_name] --model_path [output/chunks/chunk_name] -i [project/rectified/images] -d [project/rectified/depths] --alpha_masks [project/rectified/masks] --scaffold_file [output/scaffold/point_cloud/iteration_30000] --skybox_locked --bounds_file [project/chunks/chunk_name]
-
Per chunk hierarchy building
Make sure you followed the steps to generate the hierarchy creator executable file. Now we will generate a hierarchy in each chunk:# Linux: submodules/gaussianhierarchy/build/GaussianHierarchyCreator [path to output chunk point_cloud.ply] [path to chunk colmap] [path to output chunk] [path to scaffold] # Windows: submodules/gaussianhierarchy/build/Release/GaussianHierarchyCreator.exe [path to output chunk point_cloud.ply] [path to chunk colmap] [path to output chunk] [path to scaffold]
-
Single chunk post-optimization
python -u train_post.py -s [project/chunks/chunk_name] --model_path [output/chunks/chunk_name] --hierarchy [output/chunks/chunk_name/hierarchy_name.hier] --iterations 15000 --feature_lr 0.0005 --opacity_lr 0.01 --scaling_lr 0.001 --save_iterations -1 -i [project/rectified/images] --alpha_masks [project/rectified/masks] --scaffold_file [output/scaffold/point_cloud/iteration_30000] --skybox_locked --bounds_file [project/chunks/chunk_name]
-
Consolidation Make sure you followed the steps to generate the hierarchy merger executable file. Now we will consolidate and merge all the chunk hierarchies:
# Linux: submodules/gaussianhierarchy/build/GaussianHierarchyMerger [path to output/trained_chunks] "0" [path to chunk colmap] [path to output merged.hier] [list of all the chunk names] # Windows: submodules/gaussianhierarchy/build/Release/GaussianHierarchyMerger.exe [path to output/trained_chunks] "0" [path to chunk colmap] [path to output merged.hier] [list of all the chunk names]
The beginning of each .slurm
script must have the following parameters:
#!/bin/bash
#SBATCH --account=xyz@v100 # your slurm account (ex: xyz@v100)
#SBATCH --constraint=v100-32g # the gpu you require (ex: v100-32g)
#SBATCH --ntasks=1 # number of process you require
#SBATCH --nodes=1 # number of nodes you require
#SBATCH --gres=gpu:1 # number of gpus you require
#SBATCH --cpus-per-task=10 # number of cpus per task you require
#SBATCH --time=01:00:00 # maximal allocation time
Note that the slurm scripts have not been thouroughly tested.
We use a test.txt file that is read by the dataloader and splits into train/test sets when --eval
is passed to the training scripts. This file should be present in sprase/0/
for each chunk and for the aligned "global colmap" (if applicable).
The single chunks we used for evaluation:
To run the evaluations on a chunk:
python train_single.py -s ${CHUNK_DIR} --model_path ${OUTPUT_DIR} -d depths --exposure_lr_init 0.0 --eval --skip_scale_big_gauss
# Windows: build/Release/GaussianHierarchyCreator
submodules/gaussianhierarchy/build/GaussianHierarchyCreator ${OUTPUT_DIR}/point_cloud/iteration_30000/point_cloud.ply ${CHUNK_DIR} ${OUTPUT_DIR}
python train_post.py -s ${CHUNK_DIR} --model_path ${OUTPUT_DIR} --hierarchy ${OUTPUT_DIR}/hierarchy.hier --iterations 15000 --feature_lr 0.0005 --opacity_lr 0.01 --scaling_lr 0.001 --eval
python render_hierarchy.py -s ${CHUNK_DIR} --model_path ${OUTPUT_DIR} --hierarchy ${OUTPUT_DIR}/hierarchy.hier_opt --out_dir ${OUTPUT_DIR} --eval
Ensure that the test.txt is present in all sparse/0/
folders. preprocess/copy_file_to_chunks.py
can help copying it to each chunk.
Then, the scene can be optimized with eval
:
python scripts/full_train.py --project_dir ${DATASET_DIR} --extra_training_args '--exposure_lr_init 0.0 --eval'
The following renders the test set from the optimized hierarchy. Note that the current implementation loads the full hierarchy in GPU memory.
python render_hierarchy.py -s ${DATASET_DIR} --model_path ${DATASET_DIR}/output --hierarchy ${DATASET_DIR}/output/merged.hier --out_dir ${DATASET_DIR}/output/renders --eval --scaffold_file ${DATASET_DIR}/output/scaffold/point_cloud/iteration_30000
We generally disable exposure optimization for evaluations. If you want to use it, you can optimize exposure on the left half of the test image and evaluate on their right half. To achieve this, remove --exposure_lr_init 0.0
from the commands above and add --train_test_exp
to all training scripts.