UniqAI/gaussian-splatting
1
0
Fork 0
mirror of https://github.com/graphdeco-inria/gaussian-splatting synced 2025-06-26 18:18:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

added __repr__ to GaussianModel

Browse Source
This commit is contained in:
xvdp 2023-11-04 03:52:01 -07:00
parent 25b3cb8cc9
commit 27ef163bdf
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
scene/gaussian_model.py
View File

@ -58,6 +58,17 @@ class GaussianModel:
self.spatial_lr_scale = 0 self.spatial_lr_scale = 0
self.setup_functions() self.setup_functions()
def __repr__(self):
format_string = self.__class__.__name__ + '()'
for k, v in self.__dict__.items():
if torch.is_tensor(v):
format_string +=f" {k}:\t{tuple(v.shape)}\n"
else:
format_string += f"{k}:\t{v}\n"
return format_string
def capture(self): def capture(self):
return ( return (
self.active_sh_degree, self.active_sh_degree,