tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00
Code

Don't apply near plane during eval (#716)

Browse Source
This commit is contained in:
Matthew Tancik 2022-10-06 12:22:27 -07:00 committed by GitHub
parent 9252a46273
commit eff62f7c6c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
nerfstudio/model_components/scene_colliders.py
View File

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