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 rgbs sigmas to be contiguous

Browse Source
This commit is contained in:
Ruilong Li 2022-07-27 22:50:11 +00:00
parent c27dc26e03
commit af176ddf32
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nerfactory/graphs/instant_ngp.py
View File

@ -100,7 +100,12 @@ class NGPGraph(Graph):
accumulated_color,
alive_ray_mask,
) = nerfactory_cuda.VolumeRenderer.apply(
packed_info, ray_samples.frustums.starts, ray_samples.frustums.ends, sigmas, rgbs, opacities
packed_info,
ray_samples.frustums.starts,
ray_samples.frustums.ends,
sigmas.contiguous(),
rgbs.contiguous(),
opacities,
)
accumulated_depth = torch.clip(accumulated_depth, t_min[:, None], t_max[:, None])
accumulated_color = accumulated_color + colors.WHITE.to(accumulated_color) * (1.0 - accumulated_weight)