From f77a8c97cc872ac9e043fe5996d3ec893040f107 Mon Sep 17 00:00:00 2001 From: Hardik Dava <39372750+hardikdava@users.noreply.github.com> Date: Tue, 16 Jul 2024 08:55:37 +0200 Subject: [PATCH] fixing error --- nerfstudio/scripts/render.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/scripts/render.py b/nerfstudio/scripts/render.py index c5b8c576075..9463259ce58 100644 --- a/nerfstudio/scripts/render.py +++ b/nerfstudio/scripts/render.py @@ -842,7 +842,7 @@ def update_config(config: TrainerConfig) -> TrainerConfig: # Map to color spaces / numpy if is_raw: output_image = output_image.cpu().numpy() - elif output_name == "rgba": # for RGBA image just do simple numpy conversion + elif output_name == "rgba": output_image = output_image.detach().cpu().numpy() elif is_depth: output_image = (