reduced-3dgs 1.8.19__cp311-cp311-win_amd64.whl → 1.9.0__cp311-cp311-win_amd64.whl

reduced_3dgs/importance/trainer.py

@@ -75,7 +75,7 @@ def count_render(self: GaussianModel, viewpoint_camera: Camera):
     }
 
 
-def prune_gaussians(model: GaussianModel, dataset: CameraDataset):
+def prune_list(model: GaussianModel, dataset: CameraDataset):
     gaussian_count = torch.zeros(model.get_xyz.shape[0], device=model.get_xyz.device, dtype=torch.int)
     opacity_important_score = torch.zeros(model.get_xyz.shape[0], device=model.get_xyz.device, dtype=torch.float)
     T_alpha_important_score = torch.zeros(model.get_xyz.shape[0], device=model.get_xyz.device, dtype=torch.float)
@@ -84,7 +84,57 @@ def prune_gaussians(model: GaussianModel, dataset: CameraDataset):
         gaussian_count += out["gaussians_count"]
         opacity_important_score += out["opacity_important_score"]
         T_alpha_important_score += out["T_alpha_important_score"]
-    return None
+    return gaussian_count, opacity_important_score, T_alpha_important_score
+
+
+# return importance score with adaptive volume measure described in paper
+def calculate_v_imp_score(gaussians: GaussianModel, imp_list, v_pow):
+    """
+    :param gaussians: A data structure containing Gaussian components with a get_scaling method.
+    :param imp_list: The importance scores for each Gaussian component.
+    :param v_pow: The power to which the volume ratios are raised.
+    :return: A list of adjusted values (v_list) used for pruning.
+    """
+    # Calculate the volume of each Gaussian component
+    volume = torch.prod(gaussians.get_scaling, dim=1)
+    # Determine the kth_percent_largest value
+    index = int(len(volume) * 0.9)
+    sorted_volume, _ = torch.sort(volume, descending=True)
+    kth_percent_largest = sorted_volume[index]
+    # Calculate v_list
+    v_list = torch.pow(volume / kth_percent_largest, v_pow)
+    v_list = v_list * imp_list
+    return v_list
+
+
+def score2mask(percent, import_score: list, threshold=None):
+    sorted_tensor, _ = torch.sort(import_score, dim=0)
+    index_nth_percentile = int(percent * (sorted_tensor.shape[0] - 1))
+    value_nth_percentile = sorted_tensor[index_nth_percentile]
+    thr = min(threshold, value_nth_percentile) if threshold is not None else value_nth_percentile
+    prune_mask = (import_score <= thr)
+    return prune_mask
+
+
+def prune_gaussians(gaussians: GaussianModel, dataset: CameraDataset, prune_type="important_score", prune_percent=0.1, prune_thr=None, v_pow=0.1):
+    gaussian_list, opacity_imp_list, T_alpha_imp_list = prune_list(gaussians, dataset)
+    match prune_type:
+        case "important_score":
+            mask = score2mask(prune_percent, opacity_imp_list, prune_thr)
+        case "v_important_score":
+            v_list = calculate_v_imp_score(gaussians, opacity_imp_list, v_pow)
+            mask = score2mask(prune_percent, v_list, prune_thr)
+        case "max_v_important_score":
+            v_list = opacity_imp_list * torch.max(gaussians.get_scaling, dim=1)[0]
+            mask = score2mask(prune_percent, v_list, prune_thr)
+        case "count":
+            mask = score2mask(prune_percent, gaussian_list, prune_thr)
+        case "T_alpha":
+            # new importance score defined by doji
+            mask = score2mask(prune_percent, T_alpha_imp_list, prune_thr)
+        case _:
+            raise Exception("Unsupportive prunning method")
+    return mask
 
 
 class ImportancePruner(DensifierWrapper):

{reduced_3dgs-1.8.19.dist-info → reduced_3dgs-1.9.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reduced_3dgs
-Version: 1.8.19
+Version: 1.9.0
 Summary: Refactored code for the paper "Reducing the Memory Footprint of 3D Gaussian Splatting"
 Home-page: https://github.com/yindaheng98/reduced-3dgs
 Author: yindaheng98

{reduced_3dgs-1.8.19.dist-info → reduced_3dgs-1.9.0.dist-info}/RECORD

@@ -2,12 +2,12 @@ reduced_3dgs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reduced_3dgs/combinations.py,sha256=FrZLxd3AZlHPSq_WJeEdGWH5zh40rAuV5txxr8HsSPY,7031
 reduced_3dgs/quantize.py,sha256=Y44qHyFdOIqke7NoeqXmyKloS43j-al74ZiNsuZZHbM,2527
 reduced_3dgs/train.py,sha256=jXHdXk05o_ebHjx_VBzcY6fRNn9EdKve6Tf5YC5an0o,9803
-reduced_3dgs/diff_gaussian_rasterization/_C.cp311-win_amd64.pyd,sha256=gmzHvGQ0dckkAtNQtEb47B8q0dqqDkVffANtRPUbU_0,1632256
+reduced_3dgs/diff_gaussian_rasterization/_C.cp311-win_amd64.pyd,sha256=HEU-nPnNHqiNFZ86Lt27A_-MHwRh5zRSG43H38WSCeI,1632256
 reduced_3dgs/diff_gaussian_rasterization/__init__.py,sha256=oV6JjTc-50MscX4XHeIWSgLr3l8Y25knBIs-0gRbJr4,7932
 reduced_3dgs/importance/__init__.py,sha256=neJsbY5cLikEGBQGdR4MjwCQ5VWVikT1357DwL0EtWU,289
 reduced_3dgs/importance/combinations.py,sha256=Hxl0Syx2PD3l2bDzJ29_b2YdfC3etm3BH7gbteujGSU,1730
-reduced_3dgs/importance/trainer.py,sha256=AlkQR-dWD5A43ywLgdV2okSwB8Sg9b1u8Zi99u8Gd4A,5250
-reduced_3dgs/importance/diff_gaussian_rasterization/_C.cp311-win_amd64.pyd,sha256=ygopQDBcQr4DoGoA7nf6hr0GXXtTwUja2nDo0f00IDg,1316352
+reduced_3dgs/importance/trainer.py,sha256=eYMmvVXFdNGEJ6vrOv8hglM0CGUyo0gTret4rsmIAE4,7717
+reduced_3dgs/importance/diff_gaussian_rasterization/_C.cp311-win_amd64.pyd,sha256=f2XoUqAnEZgNL4IMqZaLkU41o-NLxOLBdBb-OT-Zlvw,1315840
 reduced_3dgs/importance/diff_gaussian_rasterization/__init__.py,sha256=Tix8auyXBb_QFQtXrV3sLE9kdnl5zgHH0BbqcFzDp84,12850
 reduced_3dgs/pruning/__init__.py,sha256=E_YxJ9cDV_B6EJbYUBEcuRYMIht_C72rI1VJUXFCLpM,201
 reduced_3dgs/pruning/combinations.py,sha256=UivTfbSMmaWYVi9E4OF-_AZA-WBWniMiX-wKUftezF8,2331
@@ -20,9 +20,9 @@ reduced_3dgs/quantization/wrapper.py,sha256=cyXqfJgo9b3fS7DYXxOk5LmQudvrEhweOebF
 reduced_3dgs/shculling/__init__.py,sha256=nP2BejDCUdCmJNRbg0hfhHREO6jyZXwIcRiw6ttVgqo,149
 reduced_3dgs/shculling/gaussian_model.py,sha256=f8QWaL09vaV9Tcf6Dngjg_Fmk1wTQPAjWhuhI_N02Y8,2877
 reduced_3dgs/shculling/trainer.py,sha256=9hwR77djhZpyf-URhwKHjnLbe0ZAOS-DIw58RzkcHXQ,6369
-reduced_3dgs/simple_knn/_C.cp311-win_amd64.pyd,sha256=8Wj2LMNDF2ge-kdpiM0bdS0VQ6eS_kqGwetWnfJdroE,1263616
-reduced_3dgs-1.8.19.dist-info/licenses/LICENSE.md,sha256=LQ4_LAqlncGkg_mQy5ykMAFtQDSPB0eKmIEtBut0yjw,4916
-reduced_3dgs-1.8.19.dist-info/METADATA,sha256=j7c3qbf3ewfPBjNgY_toNbiEYmTpTdGZ4g4zoiipjUk,13015
-reduced_3dgs-1.8.19.dist-info/WHEEL,sha256=ZhKniLpVlRfVU2a9ty7N0kTvsYDNee6K-jhM1zk8cMw,101
-reduced_3dgs-1.8.19.dist-info/top_level.txt,sha256=PpU5aT3-baSCdqCtTaZknoB32H93UeKCkYDkRCCZMEI,13
-reduced_3dgs-1.8.19.dist-info/RECORD,,
+reduced_3dgs/simple_knn/_C.cp311-win_amd64.pyd,sha256=I8xj0hehTO_hy7d1MLZlQRa9plXq7JkTbCs_yxYSQOA,1263616
+reduced_3dgs-1.9.0.dist-info/licenses/LICENSE.md,sha256=LQ4_LAqlncGkg_mQy5ykMAFtQDSPB0eKmIEtBut0yjw,4916
+reduced_3dgs-1.9.0.dist-info/METADATA,sha256=aUf_QT29FqRs9fkUn8fVi2BpHWfdeGYnB9bxA-hYVgQ,13014
+reduced_3dgs-1.9.0.dist-info/WHEEL,sha256=ZhKniLpVlRfVU2a9ty7N0kTvsYDNee6K-jhM1zk8cMw,101
+reduced_3dgs-1.9.0.dist-info/top_level.txt,sha256=PpU5aT3-baSCdqCtTaZknoB32H93UeKCkYDkRCCZMEI,13
+reduced_3dgs-1.9.0.dist-info/RECORD,,