From c15fa1819df7b33aeb63862d8ea8b168d291d5de Mon Sep 17 00:00:00 2001 From: Nicholas Pfaff Date: Tue, 4 Jul 2023 16:25:20 +0100 Subject: [PATCH] Make background_color a required argument in blend_background --- nerfstudio/model_components/renderers.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/nerfstudio/model_components/renderers.py b/nerfstudio/model_components/renderers.py index bb813ca7f..81bb90437 100644 --- a/nerfstudio/model_components/renderers.py +++ b/nerfstudio/model_components/renderers.py @@ -133,7 +133,7 @@ class RGBRenderer(nn.Module): cls, rgb: Tensor, opacity: Tensor, - background_color: Optional[Tensor] = None, + background_color: Tensor, ) -> Float[Tensor, "*bs 3"]: """Blends the background color into the image. @@ -147,9 +147,7 @@ class RGBRenderer(nn.Module): """ if len(opacity.shape) == 1: opacity = opacity.unsqueeze(-1) - blended_rgb = ( - rgb * opacity + background_color.to(rgb.device) * (1 - opacity) if background_color is not None else rgb - ) + blended_rgb = rgb * opacity + background_color.to(rgb.device) * (1 - opacity) return blended_rgb def forward(