From aaeee724b2a0b5c9016476450c14ff40b8eecd04 Mon Sep 17 00:00:00 2001 From: Ethan Weber Date: Tue, 1 Aug 2023 02:09:03 +0000 Subject: [PATCH] rename to pixel_sampler --- nerfstudio/configs/method_configs.py | 2 +- nerfstudio/data/datamanagers/base_datamanager.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/nerfstudio/configs/method_configs.py b/nerfstudio/configs/method_configs.py index 8010a62d6..48aef03ac 100644 --- a/nerfstudio/configs/method_configs.py +++ b/nerfstudio/configs/method_configs.py @@ -211,7 +211,7 @@ method_configs["depth-nerfacto"] = TrainerConfig( pipeline=VanillaPipelineConfig( datamanager=VanillaDataManagerConfig( _target=VanillaDataManager[DepthDataset], - sampling_config=PairPixelSamplerConfig(), + pixel_sampler=PairPixelSamplerConfig(), dataparser=NerfstudioDataParserConfig(), train_num_rays_per_batch=4096, eval_num_rays_per_batch=4096, diff --git a/nerfstudio/data/datamanagers/base_datamanager.py b/nerfstudio/data/datamanagers/base_datamanager.py index 95dad4823..6933dd475 100644 --- a/nerfstudio/data/datamanagers/base_datamanager.py +++ b/nerfstudio/data/datamanagers/base_datamanager.py @@ -343,7 +343,7 @@ class VanillaDataManagerConfig(DataManagerConfig): """The scale factor for scaling spatial data such as images, mask, semantics along with relevant information about camera intrinsics """ - sampling_config: PixelSamplerConfig = PixelSamplerConfig() + pixel_sampler: PixelSamplerConfig = PixelSamplerConfig() """Specifies the pixel sampler used to sample pixels from images.""" @@ -457,7 +457,7 @@ class VanillaDataManager(DataManager, Generic[TDataset]): is_equirectangular = (dataset.cameras.camera_type == CameraType.EQUIRECTANGULAR.value).all() if is_equirectangular.any(): CONSOLE.print("[bold yellow]Warning: Some cameras are equirectangular, but using default pixel sampler.") - return self.config.sampling_config.setup( + return self.config.pixel_sampler.setup( is_equirectangular=is_equirectangular, num_rays_per_batch=self.config.train_num_rays_per_batch )