Skip to content

pablodawson/few-shot-gaussian-splatting

 
 

Repository files navigation

Few-shot Gaussian Splatting

This is my unofficial implementation of: Depth-Regularized Optimization for 3D Gaussian Splatting in Few-Shot Images (Jaeyoung Chung, Jeongtaek Oh, Kyoung Mu Lee) Paper link

TLDR: Gaussian Splatting requires a large number of training images, or it tends to overfit the training data, because of the lack of geometry cues. They employ some tricks to mitigate this issue and train with sparse (2-5) views.

Comparison (5 views)

Test PNSR 8.55 vs 19.45 (!)

They achieve this by:

Depth regularization

  • Create a dense depth map for each image (monocular depth estimation), then align it with the points obtained with COLMAP.

  • Obtain the l1 loss between the aligned depth map and the rendered depth.

  • Stop training when the depth loss starts to rise. I found it more effective to do this based on the test-set loss as it's more stable, could be wrong though.

Smoothness regularization

  • They create a smoothness loss, making sure depth values are close on similar positions.

Hyperparameters

  • 3rd degree spherical harmonics are too complex for sparse views. Limit them to 1. (This seems to be the biggest factor in my tests)
  • Opacity reset does not help either. Remove it.

Train

Same as the original. Add input images to your dataset path, inside a folder called input.

Then convert:

python convert.py -s dataset_path

A COLMAP dataset will be created, with the corresponding aligned depth maps in the depth_aligned folder.

Then, as usual:

python train.py -s dataset_path --eval

About

Few shot regularization for gaussian splatting

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 99.0%
  • Other 1.0%