From e10102995fbf2f71cc5b5f14fc53962c70d55969 Mon Sep 17 00:00:00 2001 From: Seungwoo Yoo Date: Mon, 27 Jun 2022 16:35:35 +0900 Subject: [PATCH] Fix errors --- torch_nerf/runners/runner_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torch_nerf/runners/runner_utils.py b/torch_nerf/runners/runner_utils.py index 69281b5..7e08fe0 100644 --- a/torch_nerf/runners/runner_utils.py +++ b/torch_nerf/runners/runner_utils.py @@ -407,11 +407,11 @@ def visualize_scene( ) # (H * W, C) -> (C, H, W) - pixel_pred = pixel_pred.reshape(img_height, dataset.img_width, -1) + pixel_pred = pixel_pred.reshape(img_height, img_width, -1) pixel_pred = pixel_pred.permute(2, 0, 1) # save the image tvu.save_image( pixel_pred, - os.path.join(pred_img_dir, f"{str(view_idx).zfill(5)}.png"), + os.path.join(pred_img_dir, f"{str(pose_idx).zfill(5)}.png"), )