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

65 Commits

Author SHA1 Message Date
Ethan Weber
fb440fedf3
Occupancy grid (#79)
* occupancy grid in progress

* visualizing occupancy grid. currently restructing field and renderer head outputs

* black and pylint. broke nerfw and semantic nerf but it's okay

* get_project_root
2022-05-17 19:09:25 -07:00
Matt Tancik
8ccc98a389 refactor field heads 2022-05-16 18:07:22 -07:00
Ethan Weber
676beebb14
Setting near and far with scene bounds collider. Also LR schedules. (#74)
* some initial cuda code as I learn. using git to save progress.

* first cuda kernel for ray sampling that mostly works

* cleaning up code. working on lr. structs for scene bounds aabb

* added scene bounds collider code

* updated setup

* setup

* setup

* black

* including some files

* exporting env variable

* new file

* git workflows

* github actions

* github actions

* github actions

* removing env var
2022-05-16 14:03:19 -07:00
Matt Tancik
58dd2a7eee lint 2022-05-16 13:36:40 -07:00
Matt Tancik
cea1d115d2 Merge branch 'master' of github.com:ethanweber/mattport 2022-05-16 13:33:59 -07:00
Matt Tancik
cd6342733b Add spherical harmoincs 2022-05-16 12:27:33 -07:00
Ethan Weber
1c1a58c4e5
nerf-w and semantic-nerf initial implementations (#73)
* initial nerfw code

* nerfw should be implmented correctly now

* semantic nerf in progress. data loader is too slow and not clean right now

* added implementation of semantic nerf

* mend
2022-05-13 17:44:16 -07:00
Matthew Tancik
97c9fe2541
Add tensor CP encoding (#72) 2022-05-12 15:13:19 -07:00
Matthew Tancik
e78d7af264
Add CP decomposition (#70) 2022-05-11 17:23:25 -07:00
Ethan Weber
5fb3d14304
Fixed testing and added graph abstractions (#64)
* abstracted some test logic and actually using test images now

* moved logging from trainer to graph

* fixing dataset path

* refactored instant ngp code to use inheritance

* fixed my run_train

* fixed data path

* fixed data dir
2022-05-09 11:54:40 -07:00
evonneng
470812fab6
removed erroneous test default config 2022-05-06 21:57:51 -07:00
evonneng
a498a6cc64
Standardized logging code to all global var; writers now accessible from all files (#62)
* added skeleton for terminal logger

* done skeleton terminal writer logic... todo: actual calls to the write classes

* moved lambda to top function

* added logic to support timing

* prettier printing

* prettier printing

* lint

* separated out timing and other stats variables. printing in dict format

* keeping logging history and wiping rest

* updated logic to calculate total times instead of average times

* made stats tracker configurable

* moved stats tracker and profiler to own class file; implemented basic functionality for profiler

* lint removing duplicate function

* update readme; moved around config variables

* profiler prints on sig kill

* merge

* standardized all logging to global variables
2022-05-06 21:55:33 -07:00
Ethan Weber
0b82c1188c
Colmap type of dataset (from "friends" project) (#56)
* code to process the friends data. need to add masking

* code to process the friends data. need to add masking

* black

* trying to fix black

* black fixed

* updating black version
2022-05-05 08:45:30 -07:00
Matt Tancik
fe1df97522 Add accumulation 2022-05-04 17:47:58 -07:00
evonneng
35145cb400 moved test config to own config dir; using default 2022-05-03 06:10:34 +00:00
Matthew Tancik
d1e3f16d7e
Merge pull request #50 from ethanweber/tancik/refactor_weights
Refactor weights
2022-04-29 18:49:36 -07:00
Matt Tancik
aa9a1a903f Fix lint 2022-04-29 18:40:09 -07:00
Matt Tancik
55cf759435 Refactor weights 2022-04-29 18:37:20 -07:00
evonneng
e874e856ec
changed test config file to inherit from default (#49)
* lint

* reconfigured modules in yaml

* added data module

* changed test config file
2022-04-29 16:18:30 -07:00
Matt Tancik
cbc85ef450 Fix test 2022-04-29 15:44:19 -07:00
Ethan Weber
023e6a3f92 merging masters new config 2022-04-29 22:29:43 +00:00
Ethan Weber
387ad93027 merging 2022-04-29 22:10:36 +00:00
evonneng
a93dace958
modularizing hydra config and setting defaults (#46)
* lint

* reconfigured modules in yaml
2022-04-29 14:39:43 -07:00
Ethan Weber
7b118750ec merged master into this 2022-04-29 21:37:10 +00:00
Ethan Weber
6b17ad51c6 cpu support and multi gpu progress 2022-04-29 21:29:13 +00:00
evonneng
86b1011ae1
Printing profile on sig kill. (#44)
* added skeleton for terminal logger

* done skeleton terminal writer logic... todo: actual calls to the write classes

* moved lambda to top function

* added logic to support timing

* prettier printing

* prettier printing

* lint

* separated out timing and other stats variables. printing in dict format

* keeping logging history and wiping rest

* updated logic to calculate total times instead of average times

* made stats tracker configurable

* moved stats tracker and profiler to own class file; implemented basic functionality for profiler

* lint removing duplicate function

* update readme; moved around config variables

* profiler prints on sig kill
2022-04-29 11:10:34 -07:00
Ethan Weber
dd6f79fc40 using test images. including some local data. black on save. disabled warnings for pytest 2022-04-29 16:23:05 +00:00
evonneng
58f2622f4d added general testing, fixed up config, made writer general 2022-04-28 23:21:01 +00:00
Ethan Weber
80ad17700c pylint and test cases for dataloader 2022-04-23 19:45:53 +00:00
Matt Tancik
05f2537605 Switch to dictionary outputs 2022-04-21 11:40:05 -07:00
Matt Tancik
32af0082c7 Fix linting issues 2022-04-20 14:10:59 -07:00
Matt Tancik
40f0338eb2 Get tests passing 2022-04-20 13:58:39 -07:00
Matt Tancik
5ee9669b2c Add sampler 2022-04-20 12:24:08 -07:00
Ethan Weber
7ff63fe0be Merge branch 'master' into vanilla_nerf_implementation 2022-04-20 03:27:52 +00:00
Ethan Weber
7819f6da9d network pass running 2022-04-20 03:17:19 +00:00
Matt Tancik
aa2525e5cb Update imports 2022-04-18 17:54:29 -07:00
Matt Tancik
83c664e65a Updated mlp tests 2022-04-18 17:51:39 -07:00
Ethan Weber
e5bc28c4fb
Merge branch 'master' into eng/distributed_training 2022-04-17 09:40:30 -07:00
Ethan Weber
e38040ee4d
Image dataset implemented and ray generation working. (#23)
* working data loader code

* adding gitignore

* fixing linting errors

* set black line length in settings file

* update

* fixed filepath

* black format

* tqdm
2022-04-17 09:26:44 -07:00
Matthew Tancik
5df1748666
Merge pull request #20 from ethanweber/tancik/renderers_test
Fix renderer, add test
2022-04-15 17:04:17 -07:00
Matt Tancik
a96088054a Fix renderer, add test 2022-04-15 16:58:28 -07:00
evonneng
b8e2713643 Changed Graph to nn.ModuleDict; Added DDP logic to seup_graph; locallyrunning DDP 2022-04-15 23:35:00 +00:00
evonneng
b2220fef28 Merge branch 'master' of github.com:ethanweber/mattport into eng/distributed_training 2022-04-15 22:22:07 +00:00
evonneng
8e36b8c313
moved tests to their own directory (#19) 2022-04-15 15:19:04 -07:00
evonneng
62d4218421 updated logic within Trainer class to allow for multiprocess training 2022-04-15 21:53:35 +00:00
Matt Tancik
8e852c5771 Rename field heads 2022-04-15 14:38:26 -07:00
evonneng
ac3e7515ef done adding main logic in run_train_nerf to initialize processes with more robust handling of DDP 2022-04-15 21:33:50 +00:00
Matt Tancik
5e24cbf6fe Restructure into field modules 2022-04-15 14:27:43 -07:00
evonneng
7dc9d445b2 added basic distributed logic for trainer 2022-04-15 01:04:17 +00:00
Matt Tancik
469fd08cee Add more tests 2022-04-14 17:46:35 -07:00