mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-15 12:23:46 +08:00
Remove dataparser
This commit is contained in:
1 parent
57736cfafc
commit
a83d443dc3
9 files changed
+124
-57
No files matched your search
+6
-2
@@ -219,8 +219,12 @@ def main(config: TrainerConfig) -> None:
|
||||
|
||||
config.set_timestamp()
|
||||
if config.data:
|
||||
CONSOLE.log("Using --data alias for --data.pipeline.datamanager.dataparser.data")
|
||||
config.pipeline.datamanager.dataparser.data = config.data
|
||||
CONSOLE.log("Using --data alias for --data.pipeline.datamanager.data")
|
||||
config.pipeline.datamanager.data = config.data
|
||||
|
||||
if config.prompt:
|
||||
CONSOLE.log("Using --prompt alias for --data.pipeline.model.prompt")
|
||||
config.pipeline.model.prompt = config.prompt
|
||||
|
||||
if config.load_config:
|
||||
CONSOLE.log(f"Loading pre-set config from: {config.load_config}")
|
||||
|
||||
@@ -76,16 +76,16 @@ class RunViewer:
|
||||
)
|
||||
while True:
|
||||
viewer_state.vis["renderingState/isTraining"].write(False)
|
||||
self._update_viewer_state(viewer_state, config, pipeline)
|
||||
self._update_viewer_state(viewer_state, pipeline)
|
||||
|
||||
def _update_viewer_state(self, viewer_state: viewer_utils.ViewerState, config: TrainerConfig, pipeline: Pipeline):
|
||||
def _update_viewer_state(self, viewer_state: viewer_utils.ViewerState, pipeline: Pipeline):
|
||||
"""Updates the viewer state by rendering out scene with current pipeline
|
||||
Returns the time taken to render scene.
|
||||
|
||||
"""
|
||||
# NOTE: step must be > 0 otherwise the rendering would not happen
|
||||
step = 1
|
||||
num_rays_per_batch = config.pipeline.datamanager.train_num_rays_per_batch
|
||||
num_rays_per_batch = pipeline.datamanager.get_train_rays_per_batch()
|
||||
with TimeWriter(writer, EventName.ITER_VIS_TIME) as _:
|
||||
try:
|
||||
viewer_state.update_scene(self, step, pipeline.model, num_rays_per_batch)
|
||||
|
||||
Reference in new issue
Block a user