Update freesplatter/models/renderer/gaussian_utils.py

freesplatter/models/renderer/gaussian_utils.py

@@ -70,6 +70,13 @@ def build_scaling_rotation(s, r):
     return L
 
 
+def build_covariance_from_scaling_rotation(scaling, scaling_modifier, rotation):
+    L = build_scaling_rotation(scaling_modifier * scaling, rotation)
+    actual_covariance = L @ L.transpose(1, 2)
+    symm = strip_symmetric(actual_covariance)
+    return symm
+
+
 class Camera(nn.Module):
     def __init__(self, C2W, fxfycxcy, h, w):
         """
@@ -129,11 +136,6 @@ class Camera(nn.Module):
 
 class GaussianModel:
     def setup_functions(self, scaling_activation_type='sigmoid', scale_min_act=0.001, scale_max_act=0.3, scale_multi_act=0.1):
-        def build_covariance_from_scaling_rotation(scaling, scaling_modifier, rotation):
-            L = build_scaling_rotation(scaling_modifier * scaling, rotation)
-            actual_covariance = L @ L.transpose(1, 2)
-            symm = strip_symmetric(actual_covariance)
-            return symm
 
         if scaling_activation_type == 'exp':
             self.scaling_activation = torch.exp