Quick fix for eval script failing for depth-nerfacto (and other methods 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 <jmysliwiec@cyclomedia.com>
This commit is contained in:
jMysliwiecandJakub Mysliwiec authored and GitHub committed 2023-06-08 08:22:46 -07:00
1 parent 3b0f7581d8
commit 5511f5f4ed
1 file changed
+3
+3
View File
@@ -25,6 +25,7 @@ from typing import Literal, Optional, Tuple
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