feat: output depth in mm

geanrobotics · Apr 17, 2024 · 7110ba3 · 7110ba3
1 parent 849bc5a
commit 7110ba3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/nerfstudio/scripts/render.py b/nerfstudio/scripts/render.py
@@ -155,6 +155,7 @@ def _render_trajectory_video(
                     if is_depth:
                        output_depth = output_image.cpu().numpy()
                        output_depth = output_depth.squeeze(2)
+                       output_depth = (output_depth.astype(np.float32) * 1000).clip(0, 65535).astype(np.uint16)
                        np.save(f"{output_image_dir}/{camera_idx:05d}.npy", output_depth)
                     else:
                         output_image = (