Commit Graph
20 Commits
Author SHA1 Message Date
Ethan Weber 841e03a9a3 Default to train dataset when eval isn't specified in config (#265)
default to train dataset when eval isn't in config
2022-08-10 15:29:56 -07:00
Ethan WeberandJake Austin 0b042be956 Pipeline v2 (#258)
* Model class done, pipeline class still needs more visualization stuffs

* Renaming the BaseDataContainer to DataloaderOutputs and adding documentation

* Pushing more documentation, removing some trivial functions

* Updating dict typing

* Removing config dicts for collider

* Passing tests

* Pushing shell new dataloader

* Saving work, need to fix NGP model's get_loss_dict and Pipeline

* V1 of the ngp pipeline should be done. Need train script and debugging

* Updating with latest instant-ngp code

* Syntax updates from latest ngp graph code

* Linter, removing unneeded code in new NGP graph code

* Style

* Should be good now. Linter is happy and the code is verbatum current instant-ngp

* Everything done except novel view testing code

* Fixing some nitpicks

* Debugging dataloader

* merged master

* WOP. integrating pipeline with configs and notebook demo

* removing model folder

* a lot of name changing from Graph to Model

* in progress... changing trainer to use pipeline... broken right now

* pipeline code runs

* switched to val dataset

* makes run_eval.py work

* updated configs for pipeline change

* fixing test_run_train script

* fixing black and linting

* fix docs

* small updates to trainer

Co-authored-by: Jake Austin <jake-austin@berkeley.edu>
2022-08-10 14:59:45 -07:00
Matt Tancik fd9c72ddc0 Add missing alpha colors to blender datasets 2022-08-01 11:24:31 -07:00
Matthew Tancik 58212543f5 Fix typing errors (#229) 2022-07-29 14:17:38 -07:00
Matthew Tancik a957e8c501 Move datasets to classes (#227) 2022-07-28 18:41:39 -07:00
Ruilong Li(李瑞龙) 0777883fb6 make scene collider optional because NGP does not need it (#210) 2022-07-26 19:34:29 -07:00
Ruilong Li(李瑞龙) 31c2c2608d Make DensityGrid plug-and-play for the base Graph and the base Sampler. (#205) 2022-07-26 18:02:18 -07:00
Matthew Tancik 6dc704c551 Rename repo to nerfactory (#204)
* Rename to nerfactory

* Fix missing files
2022-07-25 13:52:20 -07:00
Matthew Tancik 441dcd5314 Add mix precision training (#172)
* Add mix precision training

* Disable mixed precision for cpu
2022-07-20 13:21:06 -07:00
evonneng 928a414468 adding sliders for min/max value user defined (#179)
* added sliders for min/max values... todo debug glitchy if camera moves

* fixed camera changing bug and removed sliders from config
2022-07-19 19:42:17 -07:00
Matthew Tancik ab35e28871 Cache full dataset by default (#157)
Cache full dataset by default
2022-07-15 17:52:15 -07:00
Ethan Weber 0f779c2885 quick fix to hydra issue on changing working directory 2022-07-13 23:27:56 +00:00
Ethan Weberandevonneng 56661b5d9a Viewer Dynamic Resolution Scaling (DRS) (#87)
* updated three.js version. only send camera information when it moves

* changing code back to updating camera constantly because packets seem to be lost

* interrupting function during rendering

* small linting fixes

* added logic for interrupting rendering + separted out visualizer logic from trainer

Co-authored-by: evonneng <evonne_ng@berkeley.edu>
2022-07-06 15:47:08 -07:00
Ethan Weberandevonneng bf13ed3cf3 Integrating WebRTC into viewer (#73)
* restructuring for webrtc integration

* some webrtc code but not working

* webrtc connects when muted

* the flag is real!

* updating camera in websocket bridge. moved webrtc to existing websocket connection instead of http request. need to change video track

* rudimentary adding real-time imaging

* added free TURN server so code works everywhere

* slow rendering works during training with the viewer

* fix tensor device and update default model save dir

* forgot to include trainer updates

* removing some code that isn't needed

* quick fix to bins. not the most accurate however

* cleaned up code and added some config logic

* black

* adding license

Co-authored-by: evonneng <evonne_ng@berkeley.edu>
2022-07-05 14:52:49 -07:00
Ethan Weber fefbad2592 Fixed frustum device and updated model checkpoint defaults (#78)
* fix tensor device and update default model save dir

* removed viewer reference
2022-07-04 17:22:39 -07:00
Matt Tancik 73a2449bae Fix hydra bug 2022-06-25 18:43:10 -07:00
evonneng ee4e6b24f4 Added structured config support (#58)
* added typed config; idealy would convert graph to typed too....

* added graph as config
2022-06-17 08:46:03 -07:00
evonneng 91ef54963c refactoring configs; pulled network out of graph; created trainer heirarchy (#51)
* changed config graph structure

* removed network udner graphs; added new heirarchy trainer

* fixed callback in occupancy graph

* lint

* changed register callback to setter method
2022-06-15 09:06:43 -07:00
evonneng f67dca6f5b refactor moving setup functions to associated classes; cleanup eval and benchmark scripts (#48)
* refactor code to move graph/dataset setup to respective files

* consolidated eval and benchmark script

* moved setup optimizers to its own thing
2022-06-13 19:05:07 -07:00
Ethan Weber 457788d689 dataloader logic, new folder structure, flatter configs (#26)
* renaming dataloader to data. adding some dataloader classes

* restructuring configs

* fixed instant ngp tcnn direction rescaling

* working through test cases

* black and lint and tests
2022-06-08 09:01:56 -07:00