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

Dnerf viewer and training fix (#1065)

Browse Source 
* dnerf viewer and training fix

* Update viewer_utils.py
This commit is contained in:
Mustafa Berk YALDIZ 2022-11-30 23:07:21 -08:00 committed by GitHub
parent 8c3f093ecb
commit 1437a2fad5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nerfstudio/cameras/cameras.py
View File

@ -677,7 +677,7 @@ class Cameras(TensorDataclass):
pixel_area = (dx * dy)[..., None] # ("num_rays":..., 1)
assert pixel_area.shape == num_rays_shape + (1,)
times = self.times[camera_indices, None] if self.times is not None else None
times = self.times[camera_indices, 0] if self.times is not None else None
return RayBundle(
origins=origins,

2
nerfstudio/viewer/server/viewer_utils.py
View File

@ -757,7 +757,7 @@ class ViewerState:
)
times = self.vis["renderingState/render_time"].read()
if times:
if times is not None:
times = torch.tensor([float(times)])
camera = Cameras(