From 8d4a82ef0bc25cfa068be6bc1d360e51c602d30b Mon Sep 17 00:00:00 2001 From: machenmusik Date: Wed, 21 Dec 2022 22:13:20 -0500 Subject: [PATCH] work around mismatch to allow rgb depth side-by-side output (#1156) * use max resolution when not moving or training * work around mismatch to allow rgb depth side-by-side output * Update render.py Co-authored-by: Matthew Tancik --- scripts/render.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/scripts/render.py b/scripts/render.py index 5fa400303..9714265db 100644 --- a/scripts/render.py +++ b/scripts/render.py @@ -90,7 +90,11 @@ def _render_trajectory_video( with ExitStack() as stack: writer = ( - stack.enter_context(media.VideoWriter(path=output_filename, shape=(render_height, render_width), fps=fps)) + stack.enter_context( + media.VideoWriter( + path=output_filename, shape=(render_height, render_width * len(rendered_output_names)), fps=fps + ) + ) if output_format == "video" else None ) @@ -109,6 +113,8 @@ def _render_trajectory_video( ) sys.exit(1) output_image = outputs[rendered_output_name].cpu().numpy() + if output_image.shape[-1] == 1: + output_image = np.concatenate((output_image,) * 3, axis=-1) render_image.append(output_image) render_image = np.concatenate(render_image, axis=1) if output_format == "images":