From 449f4080379c59a19b309915032dd3dbddf6c25e Mon Sep 17 00:00:00 2001
From: Jonas Kulhanek <jonas.kulhanek@live.com>
Date: Wed, 13 Nov 2024 09:30:44 +0100
Subject: [PATCH] Fix render for instant-ngp

---
 nerfstudio/scripts/render.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/nerfstudio/scripts/render.py b/nerfstudio/scripts/render.py
index 4eb4a71840..4d131c2b04 100644
--- a/nerfstudio/scripts/render.py
+++ b/nerfstudio/scripts/render.py
@@ -799,6 +799,9 @@ def update_config(config: TrainerConfig) -> TrainerConfig:
                             rgba = pipeline.model.get_rgba_image(outputs=outputs, output_name="rgb")
                             outputs["rgba"] = rgba
 
+                    # Remove non-image outputs
+                    outputs.pop("num_samples_per_ray", None)
+
                     gt_batch = batch.copy()
                     gt_batch["rgb"] = gt_batch.pop("image")
                     all_outputs = (