diff --git a/nerfstudio/scripts/render.py b/nerfstudio/scripts/render.py index c5b8c57607..9463259ce5 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 = (