Commit Graph
138 Commits
Author SHA1 Message Date
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
evonneng 609fcb04f6 Set up profiler and moved stats tracker to its own class; set up configurables for stats tracker and profiler (#42)
* 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
2022-04-28 15:04:00 -07:00
Matt Tancik 69fa901788 Append input to encoding 2022-04-27 12:08:39 -07:00
evonneng 23b70634cb Remove clutter for logging in terminal (#40)
* 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
2022-04-26 12:12:57 -07:00
evonneng 5d4ffa210a Prettier printing of times. (#39)
* 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
2022-04-25 09:17:19 -07:00
Ethan Weber 9912c812c0 Fixing Tuple converting to int error. 2022-04-25 08:45:35 -07:00
Ethan Weber 11d9c7271a black 2022-04-23 19:51:05 +00:00
Ethan Weber 80ad17700c pylint and test cases for dataloader 2022-04-23 19:45:53 +00:00
Ethan Weber 4c7d07116e added caching for dataloader 2022-04-23 17:36:57 +00:00
Ethan Weber 5f1baa5a7b Merge branch 'master' of github.com:ethanweber/mattport 2022-04-23 15:26:57 +00:00
Ethan Weber 078dc6aa04 removing dotdict 2022-04-23 15:26:53 +00:00
evonneng ca76f70d49 Added timing logic within trainer class (#37)
* 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
2022-04-22 17:33:02 -07:00