diff --git a/nerfstudio/cameras/cameras.py b/nerfstudio/cameras/cameras.py index 5644e2723..a84d39abf 100644 --- a/nerfstudio/cameras/cameras.py +++ b/nerfstudio/cameras/cameras.py @@ -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, diff --git a/nerfstudio/viewer/server/viewer_utils.py b/nerfstudio/viewer/server/viewer_utils.py index 25a90579a..05be19ea0 100644 --- a/nerfstudio/viewer/server/viewer_utils.py +++ b/nerfstudio/viewer/server/viewer_utils.py @@ -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(