evonneng
fda2d448c4
done adding test benchmark script
2022-05-12 00:45:20 +00:00
Matthew Tancik
2ebe34f01d
Use torchmetrics ( #68 )
2022-05-10 17:06:49 -07: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
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
Matt Tancik
b4e45d6fc0
Improve depth vis
2022-05-09 11:11:00 -07:00
Matt Tancik
cf9c557b7b
Add colormaps to logs
2022-05-06 18:07:12 -07:00
Matt Tancik
2268a9152b
Add disparity maps
2022-05-05 13:36:25 -07:00
Matt Tancik
fe1df97522
Add accumulation
2022-05-04 17:47:58 -07:00
Matt Tancik
fc282f8857
Add multiple color options
2022-05-04 15:47:54 -07:00
Matt Tancik
ba28d0cdb2
Add training with white backgrounds
2022-05-04 13:45:53 -07:00
Matt Tancik
a10eb5caef
Add basic NGP encoder
2022-05-02 13:27:30 -07:00
Matt Tancik
55cf759435
Refactor weights
2022-04-29 18:37:20 -07:00
Matt Tancik
076c5bf0b7
Add stratified sampling and enable pdf sampling
2022-04-29 15:37:33 -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
Matt Tancik
69fa901788
Append input to encoding
2022-04-27 12:08:39 -07:00
Ethan Weber
9912c812c0
Fixing Tuple converting to int error.
2022-04-25 08:45:35 -07:00
Ethan Weber
80ad17700c
pylint and test cases for dataloader
2022-04-23 19:45:53 +00:00
Ethan Weber
4c7d07116e
added caching for dataloader
2022-04-23 17:36:57 +00:00
Ethan Weber
5099a4f7c5
resizing image
2022-04-22 18:19:39 +00:00
Ethan Weber
f090d9a001
training loop with tensorboard funs
2022-04-22 03:52:35 +00:00
Ethan Weber
4c0df2fde8
merged from master
2022-04-22 03:51:28 +00:00
Ethan Weber
cf7034f0d1
added test set (using training data for now)
2022-04-22 03:45:06 +00:00
Ethan Weber
7e5a5bf524
code runs again, working on integrating test set and optimizers
2022-04-21 21:57:36 +00:00
Matt Tancik
a32c681a7a
Rearrange nerf files
2022-04-21 14:51:51 -07:00
Ethan Weber
0c3a72b8b3
Merge pull request #31 from ethanweber/tancik/dict_outputs
...
Switch to dictionary outputs
2022-04-21 13:28:50 -07:00
evonneng
17d837babf
Integrating logic for setting up optimizers in run_train and config files ( #33 )
...
* added optimizers
* added optimizer logic for setting it up; todo-- fix optimizers and add step logic
* typing out the alphabet
2022-04-21 13:23:18 -07:00
Matt Tancik
755716dbf7
Merge branch 'master' of github.com:ethanweber/mattport into tancik/dict_outputs
2022-04-21 11:42:39 -07:00
Matt Tancik
05f2537605
Switch to dictionary outputs
2022-04-21 11:40:05 -07: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
evonneng
af74b4e95c
check DDP running correctly
2022-04-21 17:06:29 +00:00
Matt Tancik
4325c86997
Get code running
2022-04-20 12:24:08 -07:00
Ethan Weber
9cd5c58fbd
black formatting
2022-04-20 03:18:27 +00:00
Ethan Weber
7819f6da9d
network pass running
2022-04-20 03:17:19 +00:00
Ethan Weber
be6fbe799e
most code for nerf running
2022-04-19 22:32:10 +00:00
Ethan Weber
c2f3c06b6c
brainstorming with matt on vanilla nerf implementation
2022-04-19 00:45:30 +00:00
evonneng
Matthew Tancik
Ethan Weber
Ethan Weber
evonneng