4
0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00
Commit Graph

40 Commits

Author SHA1 Message Date
evonneng
800be33cf3
added limit to how many models to save (#546) 2022-09-26 14:17:04 -07:00
Ethan Weber
21ebaceed1
fix viewer checkpointing (#511) 2022-09-21 17:56:18 -07:00
Ethan Weber
7b5a251a30
[wip] dynamic batch size for instant ngp (#501)
* starting dynamic batch size branch

* dynamic batch pipeline code
2022-09-21 14:07:04 -07:00
evonneng
0c43e7c4b2
updated event writing logic/moved save and print to after config set (#502)
* (1) if wandb/tb enabled always eval; (2) move printing of config to after all variables set

* (3) save config after everything is set

* dont change config; just disable eval with decorator
2022-09-21 13:40:56 -07:00
Jake Austin
7cf97348ee
Abstractionv2 (#500)
* Pushing abstraction changes

* Sorting imports
2022-09-20 23:56:36 -07:00
Ethan Weber
1d14263127
eval progress bar with more metrics (#499)
updated eval
2022-09-20 20:07:35 -07:00
evonneng
38f83de64b
removed post init method in config setup (#495)
* mrege

* nits
2022-09-20 14:56:48 -07:00
Matthew Tancik
046b5c216b
Misc cleanup (#494)
* misc cleanup

* Fix dependencies
2022-09-20 09:14:37 -07:00
Ethan Weber
35c8d7e53c
misc viewer fixes including when no samples in a chunk (#491)
viewer fixes
2022-09-19 18:36:26 -07:00
Ethan Weber
4250161922
moving timing around (#487)
* moved timing functions around

* lint
2022-09-19 15:58:40 -07:00
Matthew Tancik
8b388a3e5c
Remove defaults from docstrings. (#480) 2022-09-19 10:04:25 -07:00
evonneng
4e7ef375a2
[wip] logging separated s.t. only one is enabled (#458)
* updated viewer/writer configs

* allow viewer + tb/wandb but with errors; moved localwriter to its own thing

* nits and all

* update test
2022-09-17 17:50:11 -07:00
Ethan Weber
e22eab8c86
updating metrics logging (#460)
* integrated changes

* update

* stratified

* black

* bump version

* upgrade version

* improved wandb logging

* fix split for eval

* comments
2022-09-16 17:27:55 -07:00
Ethan Weber
8973039e65
misc cleanup (#432)
* misc cuda improvements. added cone angle as param

* wandb as default. better organization for experiments

* turn off tensorboard

* disable eval when using viewer

* remove comment

* launch.json
2022-09-15 00:16:41 -07:00
evonneng
5469f2e6b8
Attempt at making viewer robust to random errors (#429)
* catch more specific errors

* reduce complication; simple handle of oom error

* lints
2022-09-14 11:19:19 -07:00
Ethan Weber
9bc39d895c
Export camera path (#398)
* export button working

* view changes

* adding copy to clipboard feature

* camera choice dropdown

* going to add script saving next

* copy paste working. need to add aspect ratio

* aspect overlay kind of working

* lint

* aspect ratio fixed. version specified

* merge

* openrelay back
2022-09-13 11:33:58 -07:00
Matthew Tancik
7a7a5e608a
Dynamic resolution in viewer (#397)
Dynamic resolution
2022-09-11 18:31:54 -07:00
evonneng
d50c71ac43
update docstrings and random cleanups (#387)
* fixing docstrings and small nits; up to fileds

* finished adding and fixing docstrings
2022-09-10 23:09:44 -07:00
evonneng
e7cca834c8
added flag to view dataset from get-go (#366)
* added logic to comp rays/s on train and vis

* added test and fixed vis rays/sec

* added flag to just visualize dataset

* added new script for viewing only dataset
2022-09-08 13:29:17 -07:00
evonneng
7b6ed5496b
added logic to comp rays/s on train and vis (#364)
* added logic to comp rays/s on train and vis

* added test and fixed vis rays/sec
2022-09-07 13:43:53 -07:00
evonneng
56113e6c75
[WIP] dataloading cleanup (#332)
* testing; need to merge

* fixing wrong base file in models/base

* pushed input parser into datasets class to abstract away fetching data in dataloader

* slight name chage and all

* nits

* added dataparser

* added licnese
2022-09-06 13:54:35 -07:00
evonneng
b0f6d75cdf
refactored dataset names + multi-gpu running (#321)
* refactored dataset names + multi-gpu running

* update comments + docs
2022-09-01 18:22:11 -07:00
Matthew Tancik
f910836b8c
Cleanup (#325)
cleanup
2022-09-01 12:36:54 -07:00
Terrance Wang
9b0cf2ab16
Terrance/instant ngp bug (#324)
* adding after iter callbacks in training loop

* moved callback

* fixed network name

Co-authored-by: Ethan Weber <eweb0124@gmail.com>
2022-09-01 12:11:28 -07:00
Terrance Wang
8670f19d2d
Terrance/tensorf (#218)
* added tensorf, updated requirements for windows

* cleaning up imports

* tensorf changes

* initial working version of tensorf

* added callback for feature upsampling

* modifying tests to match changes

* reverting requirements changes

* reverting unnecessary changes

* reverting unnecessary changes

* no more upsampling on step 0

* removed density field, fixed error msg

* modified callbacks and upsampling, reinitializing optimzer in training loop

* deleting tensorf_field file"

* checkpointing progress on restructuring callbacks

* updated callback logic for before and after train capabilities

* fixing upsampling stepping

* merging

* adding tensorf to config, fixing some bugs with new callback

* changed the way callbacks detected steps argument in callback func

* fixing typing

* adding pylint disable no self use to get_training_callbacks

* fixing config

* changing callback in vanilla nerf to allow for no density field

* cleanup

* removing config yaml

Co-authored-by: Ethan Weber <eweb0124@gmail.com>
2022-08-31 23:29:13 -07:00
evonneng
8f42ce4b34
Changing filepaths to Path obj (#303)
changing from using strings to paths
2022-08-30 14:23:04 -07:00
evonneng
e96f7c0617
added logic for printing things (#309)
* added logic for printing things

* dynamic viewer url
2022-08-30 11:04:42 -07:00
Ethan Weber
7e9a56fac4
Config restructure (#283)
* starting docs for pipelines

* working on an example

* clearing output

* starting to implement config changes and add documentation

* wip: removed cicular imports; porting basic config into python; todo: get python setups workin

* circular imports again

* wip: pausing at dataloader sort of working

* break for today

* running vanilla

* wip need to get all the config docs working

* laptop about to die

* instant ngp working

* added all configs for all methods; not tested yet: [mipnerf360, nerfw, semantic_nerf]

* unpacking later

* done with pylinting

* mostly done. just need to do docs and one test

* added import order to docs

* fixing near/far plane and other config things

* making loggers also instantiable via configs

* docs

* lint

* making --help work with dcargs

* moving configs back into flat

* convert list to tuples; fixed post init function with mutable params

* lint

* made run_eval and benchmarking scripts compatible with new config/dcargs

* lint

* added option to load configs from presaved config

Co-authored-by: evonneng <evonne_ng@berkeley.edu>
2022-08-26 17:25:10 -07:00
evonneng
1f1631cec6 removing callable modules from script/ 2022-08-15 21:31:43 +00:00
evonneng
b03755d6ab
[WIP] added preset button controls; moved rendering into timing loop (#270)
added preset button controls; moved rendering into timing loop
2022-08-12 09:53:29 -07:00
evonneng
1499fd1cd7
added stats- ratio train:visualizer (#268)
added stats: ratio train:visualizer
2022-08-11 16:34:57 -07:00
Ethan Weber
841e03a9a3
Default to train dataset when eval isn't specified in config (#265)
default to train dataset when eval isn't in config
2022-08-10 15:29:56 -07:00
Ethan Weber
0b042be956
Pipeline v2 (#258)
* Model class done, pipeline class still needs more visualization stuffs

* Renaming the BaseDataContainer to DataloaderOutputs and adding documentation

* Pushing more documentation, removing some trivial functions

* Updating dict typing

* Removing config dicts for collider

* Passing tests

* Pushing shell new dataloader

* Saving work, need to fix NGP model's get_loss_dict and Pipeline

* V1 of the ngp pipeline should be done. Need train script and debugging

* Updating with latest instant-ngp code

* Syntax updates from latest ngp graph code

* Linter, removing unneeded code in new NGP graph code

* Style

* Should be good now. Linter is happy and the code is verbatum current instant-ngp

* Everything done except novel view testing code

* Fixing some nitpicks

* Debugging dataloader

* merged master

* WOP. integrating pipeline with configs and notebook demo

* removing model folder

* a lot of name changing from Graph to Model

* in progress... changing trainer to use pipeline... broken right now

* pipeline code runs

* switched to val dataset

* makes run_eval.py work

* updated configs for pipeline change

* fixing test_run_train script

* fixing black and linting

* fix docs

* small updates to trainer

Co-authored-by: Jake Austin <jake-austin@berkeley.edu>
2022-08-10 14:59:45 -07:00
Matt Tancik
fc1ba5e630 Use reduce 2022-08-01 17:19:51 -07:00
Matthew Tancik
3abc586427
Remove some type ignores (#238) 2022-08-01 15:01:39 -07:00
Matthew Tancik
6c45f72c57
Typecheck workspace (#231) 2022-07-30 15:38:08 -07:00
Matthew Tancik
58212543f5
Fix typing errors (#229) 2022-07-29 14:17:38 -07:00
Ethan Weber
d13a766b9e
Serializing JSON code for drawable objects (and some documentation) (#222)
* some comments and starting serializing comments. on plane w/o formatter

* fixing pylint and black

* removed changes in file

* revert file

* addressing comments

* remove abstract

Co-authored-by: Ethan Weber <ethanweber@Ethans-MacBook-Pro.local>
2022-07-28 13:55:58 -05:00
Ruilong Li(李瑞龙)
1cd697366e
sorting imports on save; cleanup imports sorting; add isort check (#207) 2022-07-26 14:35:38 -07:00
Matthew Tancik
6dc704c551
Rename repo to nerfactory (#204)
* Rename to nerfactory

* Fix missing files
2022-07-25 13:52:20 -07:00