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

added __repr__ to Camera()

Browse Source
This commit is contained in:
xvdp 2023-11-04 08:29:03 -07:00
parent 27ef163bdf
commit f934e701b2
2 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
scene/cameras.py
View File

@ -56,6 +56,16 @@ class Camera(nn.Module):
self.full_proj_transform = (self.world_view_transform.unsqueeze(0).bmm(self.projection_matrix.unsqueeze(0))).squeeze(0) self.full_proj_transform = (self.world_view_transform.unsqueeze(0).bmm(self.projection_matrix.unsqueeze(0))).squeeze(0)
self.camera_center = self.world_view_transform.inverse()[3, :3] self.camera_center = self.world_view_transform.inverse()[3, :3]
def __repr__(self):
format_string = self.__class__.__name__ + '()\n'
for k, v in self.__dict__.items():
if torch.is_tensor(v) and v.numel() > 16:
format_string +=f" {k}:\t{tuple(v.shape)}\n"
else:
format_string += f"{k}:\t{v}\n"
return format_string
class MiniCam: class MiniCam:
def __init__(self, width, height, fovy, fovx, znear, zfar, world_view_transform, full_proj_transform): def __init__(self, width, height, fovy, fovx, znear, zfar, world_view_transform, full_proj_transform):
self.image_width = width self.image_width = width

4
scene/gaussian_model.py
View File

@ -60,7 +60,7 @@ class GaussianModel:
def __repr__(self): def __repr__(self):
format_string = self.__class__.__name__ + '()' format_string = self.__class__.__name__ + '()\n'
for k, v in self.__dict__.items(): for k, v in self.__dict__.items():
if torch.is_tensor(v): if torch.is_tensor(v):
format_string +=f" {k}:\t{tuple(v.shape)}\n" format_string +=f" {k}:\t{tuple(v.shape)}\n"
@ -84,7 +84,7 @@ class GaussianModel:
self.optimizer.state_dict(), self.optimizer.state_dict(),
self.spatial_lr_scale, self.spatial_lr_scale,
) )
def restore(self, model_args, training_args): def restore(self, model_args, training_args):
(self.active_sh_degree, (self.active_sh_degree,
self._xyz, self._xyz,