Skip to content

GitLab

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

remove shortcut for zero samples (#276)

parent 3eef9208
Hide whitespace changes
Inline Side-by-side
Showing with 12 additions and 10 deletions
nerfacc/volrend.py
View file @ d8de4301
......@@ -88,11 +88,12 @@ def rendering(
# Query sigma/alpha and color with gradients
if rgb_sigma_fn is not None:
if t_starts.shape[0] != 0:
rgbs, sigmas = rgb_sigma_fn(t_starts, t_ends, ray_indices)
else:
rgbs = torch.empty((0, 3), device=t_starts.device)
sigmas = torch.empty((0,), device=t_starts.device)
rgbs, sigmas = rgb_sigma_fn(t_starts, t_ends, ray_indices)
# if t_starts.shape[0] != 0:
# rgbs, sigmas = rgb_sigma_fn(t_starts, t_ends, ray_indices)
# else:
# rgbs = torch.empty((0, 3), device=t_starts.device)
# sigmas = torch.empty((0,), device=t_starts.device)
assert rgbs.shape[-1] == 3, "rgbs must have 3 channels, got {}".format(
rgbs.shape
)
......@@ -115,11 +116,12 @@ def rendering(
"rgbs": rgbs,
}
elif rgb_alpha_fn is not None:
if t_starts.shape[0] != 0:
rgbs, alphas = rgb_alpha_fn(t_starts, t_ends, ray_indices)
else:
rgbs = torch.empty((0, 3), device=t_starts.device)
alphas = torch.empty((0,), device=t_starts.device)
rgbs, alphas = rgb_alpha_fn(t_starts, t_ends, ray_indices)
# if t_starts.shape[0] != 0:
# rgbs, alphas = rgb_alpha_fn(t_starts, t_ends, ray_indices)
# else:
# rgbs = torch.empty((0, 3), device=t_starts.device)
# alphas = torch.empty((0,), device=t_starts.device)
assert rgbs.shape[-1] == 3, "rgbs must have 3 channels, got {}".format(
rgbs.shape
)
......
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