From 27801d2a4786de0f94c280c0fbf27ac717dd0d22 Mon Sep 17 00:00:00 2001 From: Kei Ishikawa <30857855+kstoneriv3@users.noreply.github.com> Date: Wed, 10 Jul 2024 12:47:22 +0900 Subject: [PATCH] Update the docstring of `SplatfactoModel.get_outputs` (#3289) Co-authored-by: Justin Kerr --- nerfstudio/models/splatfacto.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/nerfstudio/models/splatfacto.py b/nerfstudio/models/splatfacto.py index d5e653e0d..37856e389 100644 --- a/nerfstudio/models/splatfacto.py +++ b/nerfstudio/models/splatfacto.py @@ -687,11 +687,11 @@ class SplatfactoModel(Model): return background def get_outputs(self, camera: Cameras) -> Dict[str, Union[torch.Tensor, List]]: - """Takes in a Ray Bundle and returns a dictionary of outputs. + """Takes in a camera and returns a dictionary of outputs. Args: - ray_bundle: Input bundle of rays. This raybundle should have all the - needed information to compute the outputs. + camera: The camera(s) for which output images are rendered. It should have + all the needed information to compute the outputs. Returns: Outputs of model. (ie. rendered colors)