added equirectangular warning back

This commit is contained in:
AdamRashid96 committed 2023-06-21 00:53:09 +00:00
1 parent bcf10ed77b
commit 0e2edd9b34
1 file changed
+2
@@ -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
)