From 0e2edd9b34f7d9d008116690f903141dab19abff Mon Sep 17 00:00:00 2001 From: AdamRashid96 Date: Wed, 21 Jun 2023 00:53:09 +0000 Subject: [PATCH] added equirectangular warning back --- nerfstudio/data/datamanagers/base_datamanager.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/nerfstudio/data/datamanagers/base_datamanager.py b/nerfstudio/data/datamanagers/base_datamanager.py index ff6933959..688a2a25b 100644 --- a/nerfstudio/data/datamanagers/base_datamanager.py +++ b/nerfstudio/data/datamanagers/base_datamanager.py @@ -432,6 +432,8 @@ class VanillaDataManager(DataManager, Generic[TDataset]): def _get_pixel_sampler(self, dataset: TDataset, *args: Any, **kwargs: Any) -> PixelSampler: """Infer pixel sampler to use.""" 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( is_equirectangular=is_equirectangular, num_rays_per_batch=self.config.train_num_rays_per_batch )