Quick fix for eval script failing for depth-nerfacto (and other metho…

…ds with a non-standard dataset) (#2047) dirty fix for the eval_utils script that fails because generics don't work with yaml configs Co-authored-by: Jakub Mysliwiec <[email protected]>
nerfstudio-project · Jun 8, 2023 · 5511f5f · 5511f5f
1 parent 3b0f758
commit 5511f5f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/nerfstudio/utils/eval_utils.py b/nerfstudio/utils/eval_utils.py
@@ -25,6 +25,7 @@
 import torch
 import yaml
 
+from nerfstudio.configs.method_configs import all_methods
 from nerfstudio.data.datamanagers.base_datamanager import VanillaDataManagerConfig
 from nerfstudio.engine.trainer import TrainerConfig
 from nerfstudio.pipelines.base_pipeline import Pipeline
@@ -87,6 +88,8 @@ def eval_setup(
     config = yaml.load(config_path.read_text(), Loader=yaml.Loader)
     assert isinstance(config, TrainerConfig)
 
+    config.pipeline.datamanager._target = all_methods[config.method_name].pipeline.datamanager._target
+
     if eval_num_rays_per_chunk:
         config.pipeline.model.eval_num_rays_per_chunk = eval_num_rays_per_chunk