From 484ee27120965d61e96d576c7722a9e6c0dcf892 Mon Sep 17 00:00:00 2001 From: rjbaw Date: Tue, 3 Dec 2024 00:27:42 -0500 Subject: [PATCH] Fix evaluation scripts --- README.md | 7 ++++--- gsplat/__init__.py | 1 + 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 5d9dcc77..c4186c26 100644 --- a/README.md +++ b/README.md @@ -40,11 +40,12 @@ To build gsplat from source on Windows, please check [this instruction](docs/INS This repo comes with a standalone script that reproduces the official Gaussian Splatting with exactly the same performance on PSNR, SSIM, LPIPS, and converged number of Gaussians. Powered by gsplat’s efficient CUDA implementation, the training takes up to **4x less GPU memory** with up to **15% less time** to finish than the official implementation. Full report can be found [here](https://docs.gsplat.studio/main/tests/eval.html). ```bash -pip install -r examples/requirements.txt +cd examples +pip install -r requirements.txt # download mipnerf_360 benchmark data -python examples/datasets/download_dataset.py +python datasets/download_dataset.py # run batch evaluation -bash examples/benchmarks/basic.sh +bash benchmarks/basic.sh ``` ## Examples diff --git a/gsplat/__init__.py b/gsplat/__init__.py index 4ee718c2..f9fe29e3 100644 --- a/gsplat/__init__.py +++ b/gsplat/__init__.py @@ -1,6 +1,7 @@ import warnings from .compression import PngCompression +from .optimizers import SelectiveAdam from .cuda._torch_impl import accumulate from .cuda._torch_impl_2dgs import accumulate_2dgs from .cuda._wrapper import (