Commit Graph
252 Commits
Author SHA1 Message Date
evonneng 59eeaca93b removed stats tracker; consolidate writers into local writer 2022-05-20 02:19:44 +00:00
Matthew Tancik 4dee65b480 Merge pull request #81 from ethanweber/tancik/refactor_occupancy
Refactor occupancy. Work in progress, still a bit gross.

TODOs:
+ Remove Field.py and refactor VanillaNeRF
+ Move masking to field_module
+ Decorator or some other fancy python thing for occupancy
+ Clean up documentation
2022-05-18 14:33:31 -07:00
Matt Tancik 97cb328468 Refactor occupancy 2022-05-18 14:27:46 -07:00
Matthew Tancik 443fdef553 Run black on ipynb (#80) 2022-05-18 09:36:38 -07:00
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
Matthew Tancik 5c2d8c2a0b Merge pull request #78 from ethanweber/tancik/add_equirectangular_camera
Add equirectangular camera
2022-05-17 17:58:36 -07:00
Matt Tancik bc6c6fa62a Add equirectangular camera 2022-05-17 17:54:45 -07:00
Matthew Tancik c5ea5c3e93 Merge pull request #77 from ethanweber/tancik/update_writer_no_dict
refactor logging
2022-05-17 11:19:20 -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
Matthew Tancik 486650f0ad Merge pull request #76 from ethanweber/tancik/refactor_field_heads
refactor field heads
2022-05-16 18:55:59 -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
Matthew Tancik 752df417ce Merge pull request #75 from ethanweber/tancik/spherical_harmoincs
Tancik/spherical harmoincs
2022-05-16 13:48:47 -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 6a70eaa40a fixed up benchmarking test script more versatile 2022-05-12 17:32:53 +00:00
evonneng 39db2e8a27 Merge branch 'master' of github.com:ethanweber/mattport into eng/benchmarking 2022-05-12 15:41:21 +00:00
evonneng 1dc574281c Added eval script for benchmarking blender dataset (#71)
* fixed train iteration to much longer

* done adding test benchmark script
2022-05-11 17:56:39 -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 78f8b88d58 merging 2022-05-11 21:47:52 +00:00
evonneng a15185ea47 fixed train iteration to much longer 2022-05-11 19:22:20 +00:00
evonneng 58cdcc2b41 update model saving 2022-05-11 18:27:42 +00:00
evonneng 1563afe51c update readme (#69) 2022-05-11 11:03:15 -07:00
evonneng 9671ec3c33 update readme 2022-05-11 17:59:35 +00: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 a9b069775f Merge branch 'eng/logging_cleanup' 2022-05-10 23:58:49 +00: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
evonneng 71b8f9fc80 fixed too many nested ifs 2022-05-10 23:52:06 +00:00
evonneng 010a7e4e1f cap history in calc avg of stats 2022-05-10 22:59:41 +00: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
Matthew Tancik 6340f975db Merge pull request #65 from ethanweber/tancik/nerf_alignment
Better align with nerf
2022-05-10 10:27:14 -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
Matthew Tancik e83ce4f61b Merge pull request #63 from ethanweber/tancik/improve_depth_vis
Improve depth vis
2022-05-09 11:20:11 -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 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
Matthew Tancik 5cb00c286d Merge pull request #61 from ethanweber/tancik/colormap_logs
Add colormaps to logs
2022-05-06 18:10:17 -07:00
Matt Tancik cf9c557b7b Add colormaps to logs 2022-05-06 18:07:12 -07:00