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

Improve numerical stability of _torch_impl.project_cov3d_ewa #182

Conversation

oseiskar
Copy link
Contributor

Fixes the a minimal set of numerical issues in _torch_impl so that it can be used in actual training and not just unit tests. This makes testing and optimizing various things a lot easier (example here). Similar to one of the goals in #172. However, #172 seems to be still a work-in-progress, so this PR still makes sense too and should not conflict with #172.

The numerical issues were mostly related to:

  1. Computing things for Gaussians that are behind the current camera
  2. Trying to differentiate $\sqrt x$ too close to $x = 0$

@jb-ye
Copy link
Collaborator

jb-ye commented May 28, 2024

Looks good to me. Have you tested the test_project_gaussians.py ?

@oseiskar
Copy link
Contributor Author

Looks good to me. Have you tested the test_project_gaussians.py ?

Yes. Still passing. (Just double checked after realizing the passing CI build does not apparently test CUDA things)

Copy link
Collaborator

@jb-ye jb-ye 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 to me

@jb-ye jb-ye merged commit 74aba63 into nerfstudio-project:main May 28, 2024
2 checks passed
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

2 participants