Commit Graph
59 Commits
Author SHA1 Message Date
Ethan Weber 146332cfb6 Refactoring (data loading, semantic-nerf/nerf-w fixes, ...) (#8)
* in progress. camera_indices broken

* faster dataloader. starting on pixel sampler logic

* imagesampler and pixelsampler

* lint and black

* continuing merge

* pytest working

* fixed docs

* docs

* minor changes
2022-06-02 14:58:46 -04:00
Matt Tancik 1a8ad9a763 Change name to pyrad 2022-05-31 14:15:06 -07:00
Matt Tancik eb8434ace9 Update test sensitivity 2022-05-31 10:05:20 -07:00
Matt Tancik b26408eb7a Adios mattport 2022-05-30 09:24:12 -07:00
Ethan Weber 3226a1c989 CPU usage improvement and minor changes (#108)
* cpu fixes, masking logic but commented out, nerfw and semanticnerf in progress

* pylint
2022-05-26 10:18:08 -04:00
evonneng e4a878556e removed outdated test configs 2022-05-23 20:49:45 +00:00
Ethan Weber 7d8beaaed4 Tiny-cuda-nn and fixed logic for instant-ngp (#102)
* correcting implementation for instant ngp

* instant ngp with tcnn

* black and pylint

* merged master
2022-05-23 12:05:41 -07:00
Matt Tancik a9e6cad0fb Refactor field heads 2022-05-20 15:36:14 -07:00
Matt Tancik 8a6b7c3e46 Refactor ray samples 2022-05-19 18:38:30 -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
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
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
Matt Tancik cbc85ef450 Fix test 2022-04-29 15:44:19 -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
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
Matt Tancik 03738b1066 Update render_heads to use Module class 2022-04-14 17:31:10 -07:00
evonneng 4b131d4358 lint (#15) 2022-04-12 17:26:29 -07:00
evonneng 1aed15faad changed config loading to hydra for graph generation + tests 2022-04-13 00:04:58 +00:00
Ethan Weber 0b7dfcdc5d Initial ray generation and dataset code (#12)
* ray generator and cameras structure progress

* test renames

* added pass

* conflicts

* fixing some black errors

* reformated files

* merged master and .gitignore

* fixing linting

* removing old folder

* removed return False

* pylint fixes

* update

* updates

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update
2022-04-12 15:08:16 -07:00
evonneng 69ea3ae45e rudimentary code; passing logic for ordering and input size calculation; need to check design decisions 2022-04-12 03:55:31 +00:00