mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-12 02:45:34 +08:00
Loose assert condition for datamanger in get_average_eval_image_metrics() (#3310)
* loosen datamanger assert * linter
This commit is contained in:
1 parent
5945851598
commit
7736539cc7
1 file changed
+4
-6
@@ -35,9 +35,7 @@ from torch.nn import Parameter
|
||||
from torch.nn.parallel import DistributedDataParallel as DDP
|
||||
|
||||
from nerfstudio.configs.base_config import InstantiateConfig
|
||||
from nerfstudio.data.datamanagers.base_datamanager import DataManager, DataManagerConfig, VanillaDataManager
|
||||
from nerfstudio.data.datamanagers.full_images_datamanager import FullImageDatamanager
|
||||
from nerfstudio.data.datamanagers.parallel_datamanager import ParallelDataManager
|
||||
from nerfstudio.data.datamanagers.base_datamanager import DataManager, DataManagerConfig
|
||||
from nerfstudio.engine.callbacks import TrainingCallback, TrainingCallbackAttributes
|
||||
from nerfstudio.models.base_model import Model, ModelConfig
|
||||
from nerfstudio.utils import profiler
|
||||
@@ -424,9 +422,9 @@ class VanillaPipeline(Pipeline):
|
||||
self, step: Optional[int] = None, output_path: Optional[Path] = None, get_std: bool = False
|
||||
):
|
||||
"""Get the average metrics for evaluation images."""
|
||||
|
||||
"""Get the average metrics for evaluation images."""
|
||||
assert isinstance(self.datamanager, (VanillaDataManager, ParallelDataManager, FullImageDatamanager))
|
||||
assert hasattr(
|
||||
self.datamanager, "fixed_indices_eval_dataloader"
|
||||
), "datamanager must have 'fixed_indices_eval_dataloader' attribute"
|
||||
image_prefix = "eval"
|
||||
return self.get_average_image_metrics(
|
||||
self.datamanager.fixed_indices_eval_dataloader, image_prefix, step, output_path, get_std
|
||||
|
||||
Reference in new issue
Block a user