Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Refactor splatfacto gaussian params with a ParameterDict #2946

Merged
merged 12 commits into from
Mar 1, 2024

Conversation

kerrj
Copy link
Collaborator

@kerrj kerrj commented Feb 22, 2024

This simplifies subclassing, and simplifies some of the logic for culling/splitting.

@jb-ye
Copy link
Collaborator

jb-ye commented Feb 23, 2024

The change looks good to me. Some suggestions:

(1) test PLY exporter end-to-end to safeguard the workflow is fine.
(2) since the change disallow loading checkpoint from previous versions. I would suggest we bump a major version number like (1.1.0) to distinguish from previous versions. User can still use earlier versions to view their assets.

@jb-ye
Copy link
Collaborator

jb-ye commented Feb 23, 2024

One more thing: when rebasing on top of #2945 We need to change the parameter names for splatfacto-big.

@kerrj
Copy link
Collaborator Author

kerrj commented Feb 28, 2024

This should be ready to merge, fixed all the mentioned changes and also added loading logic to make it backwards-compatible with old checkpoints.

@maturk
Copy link
Collaborator

maturk commented Mar 1, 2024

LGTM! Can we merge this?

Copy link
Collaborator

@brentyi brentyi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

looks good + worked for me!

@brentyi brentyi enabled auto-merge (squash) March 1, 2024 10:49
@brentyi brentyi merged commit da2f833 into main Mar 1, 2024
2 checks passed
@brentyi brentyi deleted the justin/paramdict branch March 1, 2024 10:52
LingzheZhao added a commit to WU-CVGL/BAD-Gaussians that referenced this pull request Mar 21, 2024
Michael-Spleenlab pushed a commit to Michael-Spleenlab/nerfstudio that referenced this pull request Apr 26, 2024
…project#2946)

* refactor splatfacto with parameterdict to simplify subclassing, defining new gaussian features

* stray comment

* fix exp_avg key error

* backwards compatibility for checkpoint

* backwards comp fix

---------

Co-authored-by: Matias Turkulainen <30566358+maturk@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

4 participants