Commit Graph
149 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 bc6c6fa62a Add equirectangular camera 2022-05-17 17:54:45 -07:00
Matt Tancik 529d27e00d Add missing file 2022-05-17 11:15:26 -07:00
Matt Tancik 685f620313 refactor logging 2022-05-17 11:06:02 -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
evonneng 8f6c2ff894 nit 2022-05-12 00:52:44 +00:00
evonneng 07fa53d7cc Merge branch 'master' of github.com:ethanweber/mattport into eng/benchmarking 2022-05-12 00:45:35 +00:00
evonneng fda2d448c4 done adding test benchmark script 2022-05-12 00:45:20 +00:00
Matthew Tancik e78d7af264 Add CP decomposition (#70) 2022-05-11 17:23:25 -07:00
evonneng 6a77ca89aa Merge branch 'master' of github.com:ethanweber/mattport 2022-05-11 00:09:42 +00:00
evonneng 98a74c67de update lint error 2022-05-11 00:09:33 +00:00
Matthew Tancik 2ebe34f01d Use torchmetrics (#68) 2022-05-10 17:06:49 -07:00
evonneng a555ef7eb4 cap history in calc avg of stats; tensorboard output to /tmp (#67)
* cap history in calc avg of stats

* fixed too many nested ifs
2022-05-10 16:55:16 -07:00
Matthew Tancik 29ba6d8abc Add ssim and update typechecks (#66)
* Add ssim and update typechecks

* update requirements

* Fix type error
2022-05-10 11:52:32 -07:00
Matt Tancik 2a56fac89c Fix test 2022-05-10 10:22:17 -07:00
Matt Tancik 8c1e9d7f1a Better align with nerf 2022-05-10 10:18:20 -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
Matt Tancik 0e77c731e2 missing file 2022-05-09 11:16:04 -07:00
Matt Tancik b4e45d6fc0 Improve depth vis 2022-05-09 11:11:00 -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
Matt Tancik cf9c557b7b Add colormaps to logs 2022-05-06 18:07:12 -07:00
Matt Tancik 2268a9152b Add disparity maps 2022-05-05 13:36:25 -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
Matt Tancik fba0e93aca Fix typing 2022-05-04 16:12:49 -07:00
Matt Tancik 5e3daee0ae update lint 2022-05-04 15:56:27 -07:00
Matt Tancik 3507467bf4 update lint 2022-05-04 15:53:20 -07:00
Matt Tancik fc282f8857 Add multiple color options 2022-05-04 15:47:54 -07:00
Matt Tancik ba28d0cdb2 Add training with white backgrounds 2022-05-04 13:45:53 -07:00
Matt Tancik e0bf0bbe6a Disable gradients on samplers 2022-05-03 17:13:04 -07:00
Matt Tancik c09a06e0de Add encoder visualizations 2022-05-03 16:10:19 -07:00
Matt Tancik a10eb5caef Add basic NGP encoder 2022-05-02 13:27:30 -07:00
Matt Tancik 7622515e92 Fix lint 2022-04-29 18:46:02 -07:00
Matt Tancik 55cf759435 Refactor weights 2022-04-29 18:37:20 -07:00
Matt Tancik 065b94ea2c Merge branch 'master' of github.com:ethanweber/mattport into tancik/stratified_and_pdf_sampling 2022-04-29 15:37:52 -07:00
Matt Tancik 076c5bf0b7 Add stratified sampling and enable pdf sampling 2022-04-29 15:37:33 -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