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

[Fix] Pixel Sampler setup in VanillaDataManager for Cameras incl. metadata (#2678)

Browse Source 
fix pixel sampler setup for cameras with metadata other than fisheye crop radius

Co-authored-by: Matias Turkulainen <30566358+maturk@users.noreply.github.com>
This commit is contained in:
Tobias Fischer 2023-12-14 19:07:41 +01:00 committed by GitHub
parent a894c6d938
commit 622e26b511
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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