tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 18:44:21 +08:00
Code

rename to pixel_sampler

Browse Source
This commit is contained in:
Ethan Weber 2023-08-01 02:09:03 +00:00
parent f115dc784e
commit aaeee724b2
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nerfstudio/configs/method_configs.py
View File

@ -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,

4
nerfstudio/data/datamanagers/base_datamanager.py
View File

@ -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
)