There was an error while loading. Please reload this page.
1 parent 6243cc8 commit 6f37836Copy full SHA for 6f37836
1 file changed
gsplat/compression/sort.py
@@ -32,7 +32,7 @@ def sort_splats(splats: Dict[str, Tensor], verbose: bool = True) -> Dict[str, Te
32
sort_keys = ["means", "quats", "scales", "opacities"]
33
if "sh0" in splats:
34
sort_keys.append("sh0")
35
-
+
36
params_to_sort = torch.cat([splats[k].reshape(n_gs, -1) for k in sort_keys], dim=-1)
37
shuffled_indices = torch.randperm(
38
params_to_sort.shape[0], device=params_to_sort.device
0 commit comments