From 622e26b511545bc583ee5122bf4a9dc6bfb63034 Mon Sep 17 00:00:00 2001 From: Tobias Fischer <36965290+tobiasfshr@users.noreply.github.com> Date: Thu, 14 Dec 2023 19:07:41 +0100 Subject: [PATCH] [Fix] Pixel Sampler setup in VanillaDataManager for Cameras incl. metadata (#2678) fix pixel sampler setup for cameras with metadata other than fisheye crop radius Co-authored-by: Matias Turkulainen <30566358+maturk@users.noreply.github.com> --- nerfstudio/data/datamanagers/base_datamanager.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/nerfstudio/data/datamanagers/base_datamanager.py b/nerfstudio/data/datamanagers/base_datamanager.py index 47144c00e..bf12bd604 100644 --- a/nerfstudio/data/datamanagers/base_datamanager.py +++ b/nerfstudio/data/datamanagers/base_datamanager.py @@ -476,9 +476,10 @@ class VanillaDataManager(DataManager, Generic[TDataset]): if is_equirectangular.any(): CONSOLE.print("[bold yellow]Warning: Some cameras are equirectangular, but using default pixel sampler.") - fisheye_crop_radius = ( - None if dataset.cameras.metadata is None else dataset.cameras.metadata["fisheye_crop_radius"] - ) + fisheye_crop_radius = None + if dataset.cameras.metadata is not None and "fisheye_crop_radius" in dataset.cameras.metadata: + fisheye_crop_radius = dataset.cameras.metadata["fisheye_crop_radius"] + return self.config.pixel_sampler.setup( is_equirectangular=is_equirectangular, num_rays_per_batch=num_rays_per_batch,