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

[DRAFT] gsplat refactor #269

Conversation

rtabrizi
Copy link
Collaborator

@rtabrizi rtabrizi commented Jul 7, 2024

ongoing branch of changes for the gsplat modularization/refactor

@liruilong940607 liruilong940607 merged commit da01272 into nerfstudio-project:rtabrizi/refactor Jul 8, 2024
@Atticuszz
Copy link

incredible! man!

liruilong940607 added a commit that referenced this pull request Jul 17, 2024
* [DRAFT] gsplat refactor (#269)

* initial commit, rough design of modules

* moved strategy.py into gsplat dir

* latest commit for 1:1 meeting

* removed ops, losses, and strategies now inherit from an abstract 'Strategy' class

* GSs init handled in private method

* format

* class Gaussian3D

* WIP: _strategy

* simple_trainer no longer uses Gaussians object and instead uses ParameterDict with newly-added 'activations' dict

* simple_trainer no longer uses Gaussians object and instead uses ParameterDict with newly-added 'activations' dict

* created mcmc and revised strategy, simplified training scripts

* update _strategy

* minor

* implemented ruilongs' suggested changes, no longer using separate class for revised ADS heuristics

* added docstrings, removed overriding activations attribute, fixed device argument passing

* mcmc and default are all trainable

* cleanup

* minor

* minor

* minor

* minor

* hardcode activations into strategy

* minor cleanup

* minor

* seperate the strategy in to individual files

* change optimizers from List to Dict

* switch to dataclass

* stateless strategy

* minor cleanup

* revised_opacity

* _update_param_with_optimizer

* individual ops

* means3d -> means

* starting docs

* docs

* docs

* docs

* docs

* docs

* minor fix to pass test

* move BINOMS to strategy state

* nits

* nit fix

* viewcode

* relocation func cleanup

* nits in docs

* cleanup imports

* minor cleanup

* support screen size pruning / splitting

---------

Co-authored-by: Ryan Tabrizi <[email protected]>
Co-authored-by: Ruilong Li <[email protected]>
Co-authored-by: rtabrizi <[email protected]>
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.

3 participants