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>
This commit is contained in:
Ethan WeberandJake Austin authored and GitHub committed 2022-08-10 14:59:45 -07:00
1 parent 670c2a01db
commit 0b042be956
55 files changed
+8595 -8263

No files matched your search

+4 -4
View File
@@ -6,27 +6,27 @@ Data
Image Dataset
----------------
.. automodule:: nerfactory.data.image_dataset
.. automodule:: nerfactory.dataloaders.image_dataset
:members:
:show-inheritance:
Image Sampler
----------------
.. automodule:: nerfactory.data.image_sampler
.. automodule:: nerfactory.dataloaders.image_sampler
:members:
:show-inheritance:
Pixel Sampler
----------------
.. automodule:: nerfactory.data.pixel_sampler
.. automodule:: nerfactory.dataloaders.pixel_sampler
:members:
:show-inheritance:
Structures
----------------
.. automodule:: nerfactory.data.structs
.. automodule:: nerfactory.dataloaders.structs
:members:
:show-inheritance:
+5 -5
View File
@@ -6,34 +6,34 @@ Graphs
Base
----------------
.. automodule:: nerfactory.graphs.base
.. automodule:: nerfactory.models.base
:members:
:show-inheritance:
Instant NGP
----------------
.. automodule:: nerfactory.graphs.instant_ngp
.. automodule:: nerfactory.models.instant_ngp
:members:
:show-inheritance:
Semantic NeRF
----------------
.. automodule:: nerfactory.graphs.semantic_nerf
.. automodule:: nerfactory.models.semantic_nerf
:members:
:show-inheritance:
TensoRF
----------------
.. automodule:: nerfactory.graphs.tensorf
.. automodule:: nerfactory.models.tensorf
:members:
:show-inheritance:
NeRF
----------------
.. automodule:: nerfactory.graphs.vanilla_nerf
.. automodule:: nerfactory.models.vanilla_nerf
:members:
:show-inheritance:
@@ -3,6 +3,6 @@
Ray Losses
===================
.. automodule:: nerfactory.graphs.modules.ray_losses
.. automodule:: nerfactory.models.modules.ray_losses
:members:
:show-inheritance:
@@ -3,6 +3,6 @@
Ray Sampler
===================
.. automodule:: nerfactory.graphs.modules.ray_sampler
.. automodule:: nerfactory.models.modules.ray_sampler
:members:
:show-inheritance: