Fix render for instant-ngp

This commit is contained in:
Jonas Kulhanek committed 2024-11-13 09:30:44 +01:00
1 parent 6b60855003
commit 449f408037
1 file changed
+3
+3
View File
@@ -799,6 +799,9 @@ class DatasetRender(BaseRender):
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 = (