mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-14 03:44:02 +08:00
Ethan Weber
5fb3d14304
* 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
21 lines
551 B
Python
21 lines
551 B
Python
"""
|
|
NeRF-W (NeRF in the wild) implementation.
|
|
TODO:
|
|
"""
|
|
|
|
from mattport.nerf.graph.vanilla_nerf import NeRFGraph
|
|
from mattport.structures.rays import RayBundle
|
|
|
|
|
|
class NerfWGraph(NeRFGraph):
|
|
"""NeRF-W graph"""
|
|
|
|
def __init__(self, intrinsics=None, camera_to_world=None, **kwargs) -> None:
|
|
super().__init__(intrinsics=intrinsics, camera_to_world=camera_to_world, **kwargs)
|
|
|
|
def get_outputs(self, ray_bundle: RayBundle):
|
|
raise NotImplementedError
|
|
|
|
def get_loss_dict(self, outputs, batch):
|
|
raise NotImplementedError
|