Fix errors

DveloperY0115 · Jun 27, 2022 · e101029 · e101029
1 parent 257a634
commit e101029
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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"),
             )