Dnerf viewer and training fix (#1065)

* dnerf viewer and training fix * Update viewer_utils.py
nerfstudio-project · Dec 1, 2022 · 1437a2f · 1437a2f
1 parent 8c3f093
commit 1437a2f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/nerfstudio/cameras/cameras.py b/nerfstudio/cameras/cameras.py
@@ -677,7 +677,7 @@ def _generate_rays_from_coords(
         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,

diff --git a/nerfstudio/viewer/server/viewer_utils.py b/nerfstudio/viewer/server/viewer_utils.py
@@ -757,7 +757,7 @@ def _render_image_in_viewer(self, camera_object, graph: Model, is_training: bool
         )
 
         times = self.vis["renderingState/render_time"].read()
-        if times:
+        if times is not None:
             times = torch.tensor([float(times)])
 
         camera = Cameras(