Adding SpotLessSplats: Ignoring Distractors in 3D Gaussian Splatting

[lilygoli]

Adding support for SpotLessSplats: Ignoring Distractors in 3D Gaussian Splatting

Project Page | Paper

Reproduced Results

The results in the SpotLessSplats paper are reproduced on gsplat codebase. The results on gsplat are reported without enabling the appearance modeling method of gsplat.

Vanilla 3DGS on Crab (2): SpotLessSplats on Crab (2):

The reproduced results of Fig. 8 in the paper on gsplat:

	Android	Statue	Crab (2)	Yoda	Mountain	Fountain	Corner	Patio	Spot	Patio High	Average
3DGS	23.23	21.45	30.03	29.7	20.02	21.49	22.34	16.77	18.93	17.09	22.10
RobustFilter	24.34	22.46	34.15	34.91	21.20	21.91	25.66	17.9	23.21	20.22	24.59
SLS-agg	25.01	22.76	34.96	35.06	21.66	22.66	25.77	22.58	24.37	22.72	25.75
SLS-mlp	25.05	22.85	35.33	35.40	21.67	22.51	25.84	22.68	25.06	23.12	25.95
SLS-mlp + UBP	24.96	22.65	35.30	35.26	21.31	22.30	26.36	22.2	25.12	23.00	25.84

The original results in the paper:

The effect of UBP with $\kappa=10^{-14}$ on gsplat:

	Compression Factor	PSNR drop
bicycle	2.21x	-0.32
garden	3.35x	-0.41
stump	2.53x	-0.23

Installation

Installation process is similar to the main gsplat branch. Make sure to pip install -r requirements.txt under the examples directory.

Run Experiments

You can run experiments with the robust masking as below:

To run the SLS-mlp version run:

python spotless_trainer.py  --data_dir [data directory] --data_factor 8   --result_dir [result directory] --loss_type robust --semantics --no-cluster --train_keyword "clutter" --test_keyword "extra" 

To run the SLS-agg version run:

python spotless_trainer.py  --data_dir [data directory] --data_factor 8   --result_dir [result directory] --loss_type robust --semantics --cluster --train_keyword "clutter" --test_keyword "extra" 

To run the RobustFilter version run:

python spotless_trainer.py  --data_dir [data directory] --data_factor 8   --result_dir [result directory] --loss_type robust --no-semantics --no-cluster --train_keyword "clutter" --test_keyword "extra" 

To run baseline 3DGS run:

python spotless_trainer.py  --data_dir [data directory] --data_factor 8   --result_dir [result directory] --loss_type l1 --no-semantics --no-cluster --train_keyword "clutter" --test_keyword "extra" 

For enabling utilization-based pruning (UBP) add --ubp to the runs.

Benchmarking

To run all the experiments together run:

./sls_benchmark.sh

Preparing Datasets

The RobustNeRF dataset and NeRF On-the-go dataset are used for experiments. The Stable Diffusion features used for these scenes can be found here.

To extract these features on your own datasets you can run the Jupyter notebook ./examples/datasets/sd_feature_extraction.ipynb.

We assume that the image files have prefixes determining clean (clean), cluttered train data (clutter) and clean test data (extra). You can process datasets like NeRF On-the-go datasets that provide a JSON file with these tags with running:

python ./examples/datasets/prep_data.py

[liruilong940607]

Great work! After chatting with @lilygoli offline we decide to not merging it until we figure out a position for this approach to live inside the gsplat library. But we'll leave this PR open for ppl to acknowledge the existence of this solution.