utils

nerfstudio-project · Aug 1, 2024 · 7edee58 · 7edee58
1 parent c0e7dcb
commit 7edee58
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/examples/utils.py b/examples/utils.py
@@ -143,12 +143,12 @@ def knn(x: Tensor, K: int = 4) -> Tensor:
     return torch.from_numpy(distances).to(x)
 
 
+def rgb_to_sh(rgb: Tensor) -> Tensor:
+    C0 = 0.28209479177387814
+    return (rgb - 0.5) / C0
+
+
 def set_random_seed(seed: int):
     random.seed(seed)
     np.random.seed(seed)
     torch.manual_seed(seed)
-
-
-def rgb_to_sh(rgb: Tensor) -> Tensor:
-    C0 = 0.28209479177387814
-    return (rgb - 0.5) / C0