Fix render for instant-ngp

nerfstudio-project · Nov 13, 2024 · 449f408 · 449f408
1 parent 6b60855
commit 449f408
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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 = (