Merge pull request #19 from RongLiu-Leo/main

avoid a division by zero

Author: liruilong940607

examples/01_dummy_training.py

@@ -100,7 +100,7 @@ def training_step():
         tic = time.time()
         with viewer.lock:
             num_train_rays_per_step = training_step()
-        num_train_steps_per_sec = 1.0 / (time.time() - tic)
+        num_train_steps_per_sec = 1.0 / (max(time.time() - tic, 1e-10))
         num_train_rays_per_sec = num_train_rays_per_step * num_train_steps_per_sec
         # Update the viewer state.
         viewer.render_tab_state.num_train_rays_per_sec = num_train_rays_per_sec

examples/04_gsplat_training.py

@@ -631,7 +631,7 @@ def train(self):
 
             if not cfg.disable_viewer:
                 self.viewer.lock.release()
-                num_train_steps_per_sec = 1.0 / (time.time() - tic)
+                num_train_steps_per_sec = 1.0 / (max(time.time() - tic, 1e-10))
                 num_train_rays_per_sec = (
                     num_train_rays_per_step * num_train_steps_per_sec
                 )
@@ -822,7 +822,7 @@ def eval(self, step: int):
             )  # [1, H, W, 3]
             colors = torch.clamp(colors, 0.0, 1.0)
             torch.cuda.synchronize()
-            ellipse_time += time.time() - tic
+            ellipse_time += max(time.time() - tic, 1e-10)
 
             # write images
             canvas = torch.cat([pixels, colors], dim=2).squeeze(0).cpu().numpy()

nerfview/_renderer.py

@@ -171,7 +171,7 @@ def run(self):
                     else:
                         img, depth = rendered, None
                     self.viewer.render_tab_state.num_view_rays_per_sec = (W * H) / (
-                        time.time() - tic
+                        max(time.time() - tic, 1e-10)
                     )
             except InterruptRenderException:
                 continue