Commit Graph
259 Commits
Author SHA1 Message Date
evonneng 39b67e6fe9 update benchmarking; add clip to mipnerf lpips 2022-07-07 18:59:23 +00:00
Matt Tancik 31a465c62f Run_actions quit on failed action 2022-07-07 11:35:16 -07:00
Ethan Weber 44c5e3264c Documenting viewer and faster image sending (#86)
* updated three.js version. only send camera information when it moves

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

* name change

* starting viewer docs

* new workflow

* ga

* ga

* ga

* ga

* ga

* ga

* ga

* ga

* ga

* ga

* ga

* ga

* ga

* ga

* ga

* ga

* ga

* homepage

* homepage

* / to ?

* more documentation

* fixed none error

* instant ngp fix

* faster image sending over tcp. instant ngp bin fix. lowest res when moving

* black

* requirements
2022-07-07 00:03:45 -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 d65b50a0e3 Removed hardcoding in benchmarking code (#59)
* updated benchmarking script

* removed annoying benchmarking hardcoding
2022-06-16 20:14:10 -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
evonneng b2648c6522 updated benchmarking to match new config 2022-06-14 00:15:05 +00:00
evonneng bed39763a1 updated benchmarking script (#32) 2022-06-08 17:39:15 -07:00
evonneng 5433f831ac add licensing headers (#27)
* add licensing headers

* add licensing to run actions

* rerun headers file

* remove trailing white spaces
2022-06-08 13:31:13 -07:00
Ethan Weber d16f87a957 fixing blender scene training with vanilla nerf 2022-06-08 18:00:51 +00: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
evonneng 9fa87173fd added make file run in debug 2022-06-07 03:04:00 +00:00
evonneng e8118f0569 merge 2022-06-03 17:51:01 +00:00
evonneng 394b1748ea update readmes 2022-06-03 17:49:27 +00:00
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
evonneng 3e86682f69 merge 2022-05-31 22:10:29 +00:00
evonneng 19109270fe update getting around readme 2022-05-31 21:50:26 +00:00
Matt Tancik 1a8ad9a763 Change name to pyrad 2022-05-31 14:15:06 -07:00
evonneng 84d01ddd25 add print to beginning of train (#114)
* added print unrolled config to beginning of train
2022-05-30 11:42:58 -07:00
evonneng fbcf3cb831 Eng/better configs (#113)
* update possible config setup

* finished porting remaning configs

* fix bugs

* change mattprot to radiance in config
2022-05-30 10:47:49 -07:00
Matt Tancik b26408eb7a Adios mattport 2022-05-30 09:24:12 -07:00
evonneng 9d2b34ab60 Added auto find gpu; added method name to arg list (#111)
* update benchmarking for better practice

* fixed logic for finding/distributing across GPUS

* added method to argument list
2022-05-30 09:01:52 -07:00
evonneng 5e9d064db6 update benchmarking for better practice (#107) 2022-05-24 12:37:45 -07: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
evonneng 1f719b49c0 Merge branch 'master' of github.com:ethanweber/mattport 2022-05-18 22:33:13 +00:00
evonneng c19deabc2e benchmarking details; todo: might need to move this out 2022-05-18 22:33:06 +00:00
Matthew Tancik 443fdef553 Run black on ipynb (#80) 2022-05-18 09:36:38 -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
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
evonneng 6a70eaa40a fixed up benchmarking test script more versatile 2022-05-12 17:32:53 +00:00
evonneng fda2d448c4 done adding test benchmark script 2022-05-12 00:45:20 +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 9671ec3c33 update readme 2022-05-11 17:59:35 +00: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
evonneng 3a35bcda91 adding scripts for easy local run (#59) 2022-05-05 13:48:34 -07:00
Ethan Weber 0b82c1188c Colmap type of dataset (from "friends" project) (#56)
* code to process the friends data. need to add masking

* code to process the friends data. need to add masking

* black

* trying to fix black

* black fixed

* updating black version
2022-05-05 08:45:30 -07:00
Ethan Weber 023e6a3f92 merging masters new config 2022-04-29 22:29:43 +00:00
Ethan Weber 387ad93027 merging 2022-04-29 22:10:36 +00:00
evonneng a93dace958 modularizing hydra config and setting defaults (#46)
* lint

* reconfigured modules in yaml
2022-04-29 14:39:43 -07:00
Ethan Weber 7b118750ec merged master into this 2022-04-29 21:37:10 +00:00
Ethan Weber 6b17ad51c6 cpu support and multi gpu progress 2022-04-29 21:29:13 +00:00
evonneng 86b1011ae1 Printing profile on sig kill. (#44)
* 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
2022-04-29 11:10:34 -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 cf7034f0d1 added test set (using training data for now) 2022-04-22 03:45:06 +00:00
Ethan Weber b6f878f227 merge 2022-04-21 20:29:43 +00:00
evonneng 4d1a6b2dbd cleanup and added logic to train for load/save checkpoints (#32)
* added checkpointing logic

* small nits

* check DDP running correctly

* cleanup and added logic to train for load/save checkpoints

* lint
2022-04-21 11:39:44 -07:00
Ethan Weber be6fbe799e most code for nerf running 2022-04-19 22:32:10 +00:00
evonneng b8e2713643 Changed Graph to nn.ModuleDict; Added DDP logic to seup_graph; locallyrunning DDP 2022-04-15 23:35:00 +00:00