Update the docstring of SplatfactoModel.get_outputs (#3289)

Co-authored-by: Justin Kerr <justin.g.kerr@gmail.com>
This commit is contained in:
Kei IshikawaandJustin Kerr authored and GitHub committed 2024-07-10 03:47:22 +00:00
1 parent 8f5867a0b4
commit 27801d2a47
1 file changed
+3 -3
+3 -3
View File
@@ -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)