fix: support fix from test

geanrobotics · Apr 4, 2024 · 849bc5a · 849bc5a
1 parent 1ceddd0
commit 849bc5a
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/nerfstudio/scripts/render.py b/nerfstudio/scripts/render.py
@@ -166,14 +166,16 @@ def _render_trajectory_video(
                             .numpy()
                         )
                     render_image.append(output_image)
-                render_image = np.concatenate(render_image, axis=1)
-                if output_format == "images":
-                    if image_format == "png":
-                        media.write_image(output_image_dir / f"{camera_idx:05d}.png", render_image, fmt="png")
-                    if image_format == "jpeg":
-                        media.write_image(
-                            output_image_dir / f"{camera_idx:05d}.jpg", render_image, fmt="jpeg", quality=jpeg_quality
-                        )
+                if not is_depth:
+                    render_image = np.concatenate(render_image, axis=1)
+
+                    if output_format == "images":
+                        if image_format == "png":
+                            media.write_image(output_image_dir / f"{camera_idx:05d}.png", render_image, fmt="png")
+                        if image_format == "jpeg":
+                            media.write_image(
+                                output_image_dir / f"{camera_idx:05d}.jpg", render_image, fmt="jpeg", quality=jpeg_quality
+                            )
                 if output_format == "video":
                     if writer is None:
                         render_width = int(render_image.shape[1])