Unverified Commit 2d8d6b43 authored by Ruilong Li(李瑞龙)'s avatar Ruilong Li(李瑞龙) Committed by GitHub
Browse files

Update train_ngp_nerf.py

parent e7bd2074
...@@ -200,7 +200,7 @@ if __name__ == "__main__": ...@@ -200,7 +200,7 @@ if __name__ == "__main__":
t * args.cone_angle, min=render_step_size t * args.cone_angle, min=render_step_size
) )
# filter out the points that are not in the near far plane. # filter out the points that are not in the near far plane.
if (near_plane is not None) and (near_plane is not None): if (near_plane is not None) and (far_plane is not None):
step_size = torch.where( step_size = torch.where(
(t > near_plane) & (t < far_plane), (t > near_plane) & (t < far_plane),
step_size, step_size,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment