tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-14 11:54:01 +08:00
Code Activity

Remove point samples

Browse Source
This commit is contained in:
Matt Tancik authored and Matthew Tancik committed 2022-07-01 09:36:59 -07:00
1 parent 191dfc6a34
commit 5a1c8a1c86
13 files changed
+85 -107
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tests/fields/test_fields.py
+3 -3
View File
@@ -5,7 +5,7 @@ import torch
from pyrad.fields.modules.field_heads import FieldHeadNames
from pyrad.fields.instant_ngp_field import TCNNInstantNGPField
from pyrad.cameras.rays import Frustums, PointSamples
from pyrad.cameras.rays import Frustums, RaySamples
def test_tcnn_instant_ngp_field():
@@ -35,8 +35,8 @@ def test_tcnn_instant_ngp_field():
frustum_ends=torch.zeros((*directions.shape[:-1], 1), device=device),
pixel_area=torch.ones((*directions.shape[:-1], 1), device=device),
)
point_samples = PointSamples(frustums=frustums)
field_outputs = field.forward(point_samples)
ray_samples = RaySamples(frustums=frustums)
field_outputs = field.forward(ray_samples)
positions_dtype = positions.dtype
field_outputs_density_dtype = field_outputs[FieldHeadNames.DENSITY].dtype