tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-10 18:17:31 +08:00
Code

Use correct pixel area in volumetric sampler (#2268)

Browse Source
This commit is contained in:
hturki 2023-07-25 21:05:40 -04:00 committed by GitHub
parent ad73c5142e
commit 4d80f0ceee
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nerfstudio/model_components/ray_samplers.py
View File

@ -505,14 +505,13 @@ class VolumetricSampler(Sampler):
if camera_indices is not None:
camera_indices = camera_indices[ray_indices]
zeros = torch.zeros_like(origins[:, :1])
ray_samples = RaySamples(
frustums=Frustums(
origins=origins,
directions=dirs,
starts=starts[..., None],
ends=ends[..., None],
pixel_area=zeros,
pixel_area=ray_bundle[ray_indices].pixel_area,
),
camera_indices=camera_indices,
)