From 13822094362a7f7049137989ba39c317a79ac198 Mon Sep 17 00:00:00 2001 From: Lei Lei <18294546+Crescent-Saturn@users.noreply.github.com> Date: Thu, 28 Nov 2024 13:56:36 -0500 Subject: [PATCH] fix: wrong torch cat dim when saving colors+depths A minor type in example/simple_trainer_2dgs.py --- examples/simple_trainer_2dgs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/simple_trainer_2dgs.py b/examples/simple_trainer_2dgs.py index 10900858..0ef299be 100644 --- a/examples/simple_trainer_2dgs.py +++ b/examples/simple_trainer_2dgs.py @@ -914,7 +914,7 @@ def render_traj(self, step: int): # write images canvas = torch.cat( - [colors, depths.repeat(1, 1, 3)], dim=1 if width > height else 1 + [colors, depths.repeat(1, 1, 3)], dim=0 if width > height else 1 ) canvas = (canvas.cpu().numpy() * 255).astype(np.uint8) canvas_all.append(canvas)