From eff62f7c6c4e46a22fa2ec7dccf64a5b663a7983 Mon Sep 17 00:00:00 2001 From: Matthew Tancik Date: Thu, 6 Oct 2022 12:22:27 -0700 Subject: [PATCH] Don't apply near plane during eval (#716) --- nerfstudio/model_components/scene_colliders.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/nerfstudio/model_components/scene_colliders.py b/nerfstudio/model_components/scene_colliders.py index 46980e6e5..88428f650 100644 --- a/nerfstudio/model_components/scene_colliders.py +++ b/nerfstudio/model_components/scene_colliders.py @@ -82,7 +82,8 @@ class AABBBoxCollider(SceneBoxCollider): ).values # clamp to near plane - nears = torch.clamp(nears, min=self.near_plane) + near_plane = self.near_plane if self.training else 0 + nears = torch.clamp(nears, min=near_plane) return nears, fars @@ -115,6 +116,7 @@ class NearFarCollider(SceneBoxCollider): def forward(self, ray_bundle: RayBundle) -> RayBundle: ones = torch.ones_like(ray_bundle.origins[..., 0:1]) - ray_bundle.nears = ones * self.near_plane + near_plane = self.near_plane if self.training else 0 + ray_bundle.nears = ones * near_plane ray_bundle.fars = ones * self.far_plane return ray_bundle