Commit Graph
258 Commits
Author SHA1 Message Date
evonneng 10d713f84b Consolidate StatsTracker as LocalWriter; cleaned up functions to write (#88)
* removed stats tracker; consolidate writers into local writer

* basic revamp of stats tracker/writers
2022-05-20 08:51:45 -07:00
Matthew Tancik 2287eb3455 Merge pull request #87 from ethanweber/tancik/refactor_ray_samples_3
Refactor ray samples
2022-05-19 18:43:07 -07:00
Matt Tancik 8a6b7c3e46 Refactor ray samples 2022-05-19 18:38:30 -07:00
Matthew Tancik 526db1eca6 Merge pull request #83 from ethanweber/tancik/add_occupancy_to_samplers
Clean up sampler
2022-05-19 14:21:09 -07:00
Matt Tancik 998b98b19a Clean up sampler 2022-05-19 14:17:03 -07:00
Matthew Tancik bdad07855e Merge pull request #82 from ethanweber/tancik/add_occupancy_to_samplers
Add occupancy to samplers
2022-05-18 17:49:27 -07:00
Matt Tancik 22fdcfe235 Add occupancy to samplers 2022-05-18 17:45:31 -07: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