mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-12 19:04:03 +08:00
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:
1 parent
3b0f7581d8
commit
5511f5f4ed
1 file changed
+3
@@ -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
|
||||
|
||||
|
||||
Reference in new issue
Block a user