Commit Graph
54 Commits
Author SHA1 Message Date
evonneng 800be33cf3 added limit to how many models to save (#546) 2022-09-26 14:17:04 -07:00
evonneng 399e5cac6d move outputs to config (#545) 2022-09-26 13:55:09 -07:00
Matthew Tancik 0181ce71a9 Add appearance conditioning to proposal net (#544) 2022-09-26 13:41:17 -07:00
Ethan Weber 383a6075cc [wip] proposal network sampler (#534)
* initial running code

* dozer and poster 6K experiments

* code reorganization

* removed print

* black
2022-09-25 16:56:29 -07:00
Matthew Tancik aea0cfc253 Add dynamic batching to compound model (#535) 2022-09-24 21:40:23 -07:00
Ethan Weber 21ebaceed1 fix viewer checkpointing (#511) 2022-09-21 17:56:18 -07:00
Ethan Weber 7b5a251a30 [wip] dynamic batch size for instant ngp (#501)
* starting dynamic batch size branch

* dynamic batch pipeline code
2022-09-21 14:07:04 -07:00
evonneng 0c43e7c4b2 updated event writing logic/moved save and print to after config set (#502)
* (1) if wandb/tb enabled always eval; (2) move printing of config to after all variables set

* (3) save config after everything is set

* dont change config; just disable eval with decorator
2022-09-21 13:40:56 -07:00
Jake Austin 7cf97348ee Abstractionv2 (#500)
* Pushing abstraction changes

* Sorting imports
2022-09-20 23:56:36 -07:00
Ethan Weber 1d14263127 eval progress bar with more metrics (#499)
updated eval
2022-09-20 20:07:35 -07:00
Matthew Tancik a04427644a Add Adam optimizer (#496)
* Add adam optimizer

* Fix configs

* merge
2022-09-20 15:10:48 -07:00
evonneng 38f83de64b removed post init method in config setup (#495)
* mrege

* nits
2022-09-20 14:56:48 -07:00
evonneng b4d344b026 removed num_samples from base config (#485) 2022-09-20 09:51:37 -07:00
evonneng 211515c633 auto set experiment name (#484)
changed experiment/method name requirements
2022-09-19 13:05:45 -07:00
evonneng c2a2b1de1e changed base method names (#483)
* changed base method names

* fixed tests
2022-09-19 12:43:33 -07:00
evonneng 1879825d04 Set viewer on/logger off for instantngp + compound (#481)
added defaults properly
2022-09-19 11:36:51 -07:00
Matthew Tancik 8b388a3e5c Remove defaults from docstrings. (#480) 2022-09-19 10:04:25 -07:00
Ethan WeberandMatt Tancik 9a39298e2f more config moving (#478)
* Move parser configs to files

* fixed circular import errors

* update to configs

* merge fixes

Co-authored-by: Matt Tancik <tancik@berkeley.edu>
2022-09-18 17:43:02 -07:00
Matthew TancikandEthan Weber 00c3ae73b2 Move dataset parser configs to files (#476)
* Move parser configs to files

* fixed circular import errors

Co-authored-by: Ethan Weber <eweb0124@gmail.com>
2022-09-18 16:27:41 -07:00
Matthew Tancik 2be4625b4f Migrate compound model to nerfacc (#474) 2022-09-18 14:14:30 -07:00
evonneng 4e7ef375a2 [wip] logging separated s.t. only one is enabled (#458)
* updated viewer/writer configs

* allow viewer + tb/wandb but with errors; moved localwriter to its own thing

* nits and all

* update test
2022-09-17 17:50:11 -07:00
Ethan Weber e22eab8c86 updating metrics logging (#460)
* integrated changes

* update

* stratified

* black

* bump version

* upgrade version

* improved wandb logging

* fix split for eval

* comments
2022-09-16 17:27:55 -07:00
Terrance Wang 84024a4e73 compound model (wip) (#427)
* compound model now uses compound field, passing num_images in the config, working on passing image indices through ngp sampler

* renaming cuda func, adding torch compound support

* removing unused imports

* getting num images from dataloader instead of config

* fixing pylint and adding todo

* undoing base config change

* adding cone angle and near plane to compound model config
2022-09-16 14:38:25 -07:00
Ethan Weber 5ccb8277d1 nerfacc integration (#447)
* integrated changes

* update

* stratified

* black

* bump version

* upgrade version
2022-09-16 09:51:19 -07:00
Matthew Tancik 695d9a1777 Add near plane to NGP sampler (#443) 2022-09-15 17:42:42 -07:00
Ethan Weber 1492f84930 installation documentation (#440) 2022-09-15 13:18:44 -07:00
Ethan Weber 8973039e65 misc cleanup (#432)
* misc cuda improvements. added cone angle as param

* wandb as default. better organization for experiments

* turn off tensorboard

* disable eval when using viewer

* remove comment

* launch.json
2022-09-15 00:16:41 -07:00
Alexander Kristoffersen cbebde0d58 Record3D bugfixes (#382)
* fixing bbox and rotation matrix.

* added rotation.

* cleanup.

* fix coordinate system rotation.

* added orientation as config.
2022-09-13 18:03:21 -07:00
Brent Yi 86a805c88c Experimental autocompletion setup scripts (#378)
* Working completion scripts

* generate.py => install.py

* Uninstall for completions, docs/README updates

* Tune emojis, improve concurrency for completion install

* wip

* Cleanup

* Permissions, shebang checks
2022-09-13 12:44:27 -07:00
Brent Yi 61022f7dbe Shorten dataparser subcommand (#420)
* Shorten dataparser subcommand

* Rename
2022-09-13 12:20:52 -07:00
Terrance Wang 47ea7b5b72 Terrance/combined model (#402)
* adding spatial distortion to instantngp field

* adding scene contraction to instant ngp

* rename compound, fixing configs

* adding nerfw conditioning

* added appearance embeddings, scene contraction disabled

* adding docstrings
2022-09-12 21:49:19 -07:00
Ethan Weber de2cdb6cca Friends dataset loading (#389)
command line dataparsing and friends dataset loading improvement
2022-09-11 12:52:27 -07:00
evonneng d50c71ac43 update docstrings and random cleanups (#387)
* fixing docstrings and small nits; up to fileds

* finished adding and fixing docstrings
2022-09-10 23:09:44 -07:00
Matthew Tancik ad4a6c4a6f Tune viewer for faster rendering (#385)
Tune viewer
2022-09-09 17:32:10 -07:00
Brent Yi 97b77c6c5f Bump dcargs, base config simplification (#368)
Updates for dcargs v0.3.0
2022-09-08 19:22:47 -07:00
evonneng e7cca834c8 added flag to view dataset from get-go (#366)
* added logic to comp rays/s on train and vis

* added test and fixed vis rays/sec

* added flag to just visualize dataset

* added new script for viewing only dataset
2022-09-08 13:29:17 -07:00
evonneng 7b6ed5496b added logic to comp rays/s on train and vis (#364)
* added logic to comp rays/s on train and vis

* added test and fixed vis rays/sec
2022-09-07 13:43:53 -07:00
Ethan Weber 7aa6ca9948 Kill the training script if the viewer ports are conflicting with other processes (#352)
kills program if conflicting ports for the viewer
2022-09-07 12:32:31 -07:00
Ethan Weber 512a51f714 Fixed friends dataset w/o caching (#365)
* fixing friends dataloading and nerfw running

* put dataset back to normal
2022-09-07 12:24:29 -07:00
evonneng 81fbee213f removed load; iter_train_time includes load time (#351) 2022-09-06 16:04:43 -07:00
evonneng 56113e6c75 [WIP] dataloading cleanup (#332)
* testing; need to merge

* fixing wrong base file in models/base

* pushed input parser into datasets class to abstract away fetching data in dataloader

* slight name chage and all

* nits

* added dataparser

* added licnese
2022-09-06 13:54:35 -07:00
Matthew Tancik 3841e2b395 Auto orient poses (#342) 2022-09-05 16:56:47 -07:00
Matthew Tancik b61fea8666 Remove downscaling from dataloading (#340) 2022-09-05 10:33:46 -07:00
Matthew Tancik 2ca0695fef Add colmap preprocessing script (#338)
* Add colmap preprocessing script

* Fix lint, add dataparser
2022-09-04 17:39:30 -07:00
Alexander Kristoffersen 18f8a5d0b8 Bug fixes with Record3D and new config system. (#336)
fixed some bugs regarding record3d and new config system.
2022-09-03 14:43:26 -07:00
evonneng b0f6d75cdf refactored dataset names + multi-gpu running (#321)
* refactored dataset names + multi-gpu running

* update comments + docs
2022-09-01 18:22:11 -07:00
Terrance WangandEthan Weber 8670f19d2d Terrance/tensorf (#218)
* added tensorf, updated requirements for windows

* cleaning up imports

* tensorf changes

* initial working version of tensorf

* added callback for feature upsampling

* modifying tests to match changes

* reverting requirements changes

* reverting unnecessary changes

* reverting unnecessary changes

* no more upsampling on step 0

* removed density field, fixed error msg

* modified callbacks and upsampling, reinitializing optimzer in training loop

* deleting tensorf_field file"

* checkpointing progress on restructuring callbacks

* updated callback logic for before and after train capabilities

* fixing upsampling stepping

* merging

* adding tensorf to config, fixing some bugs with new callback

* changed the way callbacks detected steps argument in callback func

* fixing typing

* adding pylint disable no self use to get_training_callbacks

* fixing config

* changing callback in vanilla nerf to allow for no density field

* cleanup

* removing config yaml

Co-authored-by: Ethan Weber <eweb0124@gmail.com>
2022-08-31 23:29:13 -07:00
Matthew Tancik 6ee6934581 Update dataset loaders (#319)
* Update dataset loaders

* Set ngp default to blender
2022-08-31 17:30:56 -07:00
evonneng 8f42ce4b34 Changing filepaths to Path obj (#303)
changing from using strings to paths
2022-08-30 14:23:04 -07:00
evonneng e96f7c0617 added logic for printing things (#309)
* added logic for printing things

* dynamic viewer url
2022-08-30 11:04:42 -07:00