Commit Graph
36 Commits
Author SHA1 Message Date
Matthew Tancik 304a759e85 Tancik/scene contraction (#70)
Contract the scene to a sphere of radius 2. This is the same contraction used in MipNeRF 360.
Add mipnerf 360 graph and config. It currently doesn't contain the other advancements made in the paper (sample proposal network, regularizers)
2022-06-29 18:28:21 -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
Matt Tancik cc79708df7 Cleanup mipnerf implementation 2022-06-09 17:26:58 -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
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
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
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
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 cf9c557b7b Add colormaps to logs 2022-05-06 18:07:12 -07:00
Matt Tancik c164fd5207 Update vscode settings 2022-05-06 13:50:23 -07:00
Matt Tancik 0be41fe4dd Use pyproject.toml 2022-05-06 11:57:40 -07:00
Matt Tancik a10eb5caef Add basic NGP encoder 2022-05-02 13:27:30 -07:00
Ethan Weber 6b17ad51c6 cpu support and multi gpu progress 2022-04-29 21:29:13 +00: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 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
Matt Tancik a32c681a7a Rearrange nerf files 2022-04-21 14:51:51 -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 32af0082c7 Fix linting issues 2022-04-20 14:10:59 -07:00
Matt Tancik 4325c86997 Get code running 2022-04-20 12:24:08 -07:00
Matt Tancik 5ee9669b2c Add sampler 2022-04-20 12:24:08 -07:00
Ethan Weber be6fbe799e most code for nerf running 2022-04-19 22:32:10 +00: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
Matt Tancik 5da5a0deac Add type checking 2022-04-15 17:53:22 -07:00
Matt Tancik a96088054a Fix renderer, add test 2022-04-15 16:58:28 -07: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
Matt Tancik c2a851ed92 Fix linting errors throughout repo 2022-04-11 17:04:08 -07:00
Ethan Weber 6512793b29 Merge branch 'master' into ethan/initial_trainer_with_config 2022-04-07 13:25:32 -07:00
Ethan Weber 5691d00d42 fixed most linting errors 2022-04-07 20:22:48 +00:00
Ethan Weber a9c0d5b97f hydra config and starting trainer class 2022-04-06 16:46:47 +00:00
Matt Tancik 219f51a5ac add sampler boilerplate 2022-04-05 15:06:36 -07:00
Matt Tancik 69977da8e6 update camera structure 2022-04-05 14:33:25 -07:00
Matt Tancik deceb34ad7 swap lint in vscode settings 2022-04-05 11:49:01 -07:00
Ethan Weber b63d5854a1 reworked structure to use packages 2022-04-02 20:07:41 +00:00