reduced-3dgs 1.9.1__cp312-cp312-win_amd64.whl → 1.9.3__cp312-cp312-win_amd64.whl

reduced_3dgs/importance/combinations.py

@@ -16,10 +16,11 @@ def BaseImportancePrunerInDensifyTrainer(
         importance_prune_type="comprehensive",
         importance_prune_percent=0.1,
         importance_prune_thr_important_score=None,
-        importance_prune_thr_v_important_score=1.0,
+        importance_prune_thr_v_important_score=3.0,
         importance_prune_thr_max_v_important_score=None,
         importance_prune_thr_count=1,
-        importance_prune_thr_T_alpha=0.01,
+        importance_prune_thr_T_alpha=0.1,
+        importance_prune_thr_T_alpha_avg=0.001,
         importance_v_pow=0.1,
         **kwargs):
     return DensificationTrainerWrapper(
@@ -36,6 +37,7 @@ def BaseImportancePrunerInDensifyTrainer(
             importance_prune_thr_max_v_important_score=importance_prune_thr_max_v_important_score,
             importance_prune_thr_count=importance_prune_thr_count,
             importance_prune_thr_T_alpha=importance_prune_thr_T_alpha,
+            importance_prune_thr_T_alpha_avg=importance_prune_thr_T_alpha_avg,
             importance_v_pow=importance_v_pow,
         ),
         model,

reduced_3dgs/importance/trainer.py

@@ -121,10 +121,11 @@ def prune_gaussians(
         prune_type="comprehensive",
         prune_percent=0.1,
         prune_thr_important_score=None,
-        prune_thr_v_important_score=1.0,
+        prune_thr_v_important_score=None,
         prune_thr_max_v_important_score=None,
-        prune_thr_count=1,
-        prune_thr_T_alpha=0.01,
+        prune_thr_count=None,
+        prune_thr_T_alpha=None,
+        prune_thr_T_alpha_avg=None,
         v_pow=0.1):
     gaussian_list, opacity_imp_list, T_alpha_imp_list = prune_list(gaussians, dataset)
     match prune_type:
@@ -141,6 +142,10 @@ def prune_gaussians(
         case "T_alpha":
             # new importance score defined by doji
             mask = score2mask(prune_percent, T_alpha_imp_list, prune_thr_T_alpha)
+        case "T_alpha_avg":
+            v_list = T_alpha_imp_list / gaussian_list
+            v_list[gaussian_list <= 0] = 0
+            mask = score2mask(prune_percent, v_list, prune_thr_T_alpha_avg)
         case "comprehensive":
             mask = torch.zeros_like(gaussian_list, dtype=torch.bool)
             if prune_thr_important_score is not None:
@@ -155,6 +160,10 @@ def prune_gaussians(
                 mask |= score2mask(prune_percent, gaussian_list, prune_thr_count)
             if prune_thr_T_alpha is not None:
                 mask |= score2mask(prune_percent, T_alpha_imp_list, prune_thr_T_alpha)
+            if prune_thr_T_alpha_avg is not None:
+                v_list = T_alpha_imp_list / gaussian_list
+                v_list[gaussian_list <= 0] = 0
+                mask |= score2mask(prune_percent, v_list, prune_thr_T_alpha_avg)
         case _:
             raise Exception("Unsupportive prunning method")
     return mask
@@ -170,12 +179,12 @@ class ImportancePruner(DensifierWrapper):
             importance_prune_type="comprehensive",
             importance_prune_percent=0.1,
             importance_prune_thr_important_score=None,
-            importance_prune_thr_v_important_score=1.0,
+            importance_prune_thr_v_important_score=3.0,
             importance_prune_thr_max_v_important_score=None,
             importance_prune_thr_count=1,
-            importance_prune_thr_T_alpha=0.01,
-            importance_v_pow=0.1
-    ):
+            importance_prune_thr_T_alpha=1,
+            importance_prune_thr_T_alpha_avg=0.001,
+            importance_v_pow=0.1):
         super().__init__(base_densifier)
         self.dataset = dataset
         self.importance_prune_from_iter = importance_prune_from_iter
@@ -187,6 +196,7 @@ class ImportancePruner(DensifierWrapper):
         self.prune_thr_max_v_important_score = importance_prune_thr_max_v_important_score
         self.prune_thr_count = importance_prune_thr_count
         self.prune_thr_T_alpha = importance_prune_thr_T_alpha
+        self.prune_thr_T_alpha_avg = importance_prune_thr_T_alpha_avg
         self.v_pow = importance_v_pow
         self.prune_type = importance_prune_type
 
@@ -198,7 +208,7 @@ class ImportancePruner(DensifierWrapper):
                 self.prune_type, self.prune_percent,
                 self.prune_thr_important_score, self.prune_thr_v_important_score,
                 self.prune_thr_max_v_important_score, self.prune_thr_count,
-                self.prune_thr_T_alpha, self.v_pow,
+                self.prune_thr_T_alpha, self.prune_thr_T_alpha_avg, self.v_pow,
             )
             ret = ret._replace(remove_mask=remove_mask if ret.remove_mask is None else torch.logical_or(ret.remove_mask, remove_mask))
         return ret
@@ -215,10 +225,11 @@ def BaseImportancePruningTrainer(
         importance_prune_type="comprehensive",
         importance_prune_percent=0.1,
         importance_prune_thr_important_score=None,
-        importance_prune_thr_v_important_score=1.0,
+        importance_prune_thr_v_important_score=3.0,
         importance_prune_thr_max_v_important_score=None,
         importance_prune_thr_count=1,
-        importance_prune_thr_T_alpha=0.01,
+        importance_prune_thr_T_alpha=0.1,
+        importance_prune_thr_T_alpha_avg=0.001,
         importance_v_pow=0.1,
         **kwargs):
     return DensificationTrainer(
@@ -236,6 +247,7 @@ def BaseImportancePruningTrainer(
             importance_prune_thr_max_v_important_score=importance_prune_thr_max_v_important_score,
             importance_prune_thr_count=importance_prune_thr_count,
             importance_prune_thr_T_alpha=importance_prune_thr_T_alpha,
+            importance_prune_thr_T_alpha_avg=importance_prune_thr_T_alpha_avg,
             importance_v_pow=importance_v_pow,
         ), *args, **kwargs
     )

{reduced_3dgs-1.9.1.dist-info → reduced_3dgs-1.9.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reduced_3dgs
-Version: 1.9.1
+Version: 1.9.3
 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.9.1.dist-info → reduced_3dgs-1.9.3.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.cp312-win_amd64.pyd,sha256=jBdMDOB4EBADFssveEKFmexgRdO5DAFIm1edGWOpOSk,1640448
+reduced_3dgs/diff_gaussian_rasterization/_C.cp312-win_amd64.pyd,sha256=mprhB-TBrC7By8CHR6C37Ate1PPldnbuW1P7MMzuu_I,1640448
 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=Q2WqwXNuclPWBsw15aR14xQ72JewVeZo9igMeaSfuf8,2693
-reduced_3dgs/importance/trainer.py,sha256=ozzjPSXIpHNP4H1mkXlyAv5KJXWt1k4NKCikx9E9S1E,11151
-reduced_3dgs/importance/diff_gaussian_rasterization/_C.cp312-win_amd64.pyd,sha256=nt52bzd91cqJ0M5yp2inCHZzTon4M0_7mqQTjsKka9I,1320448
+reduced_3dgs/importance/combinations.py,sha256=RjA1TiTYY3GDKTiFhlNGy4SLOJtVbb1S2T-viVagHMA,2821
+reduced_3dgs/importance/trainer.py,sha256=S0fWvg2kDR5gMO3OWOiJKMv8yCGJyfTrJGCKhSSezac,11908
+reduced_3dgs/importance/diff_gaussian_rasterization/_C.cp312-win_amd64.pyd,sha256=1giDEEQobpaN3MDLH5BlU49QLCvyNie1YTp_ghQ1hf4,1320448
 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.cp312-win_amd64.pyd,sha256=dia5INp_w0nT-yT90oS0tY3kJCAQ0-G7S1bpt5m-oNE,1267712
-reduced_3dgs-1.9.1.dist-info/licenses/LICENSE.md,sha256=LQ4_LAqlncGkg_mQy5ykMAFtQDSPB0eKmIEtBut0yjw,4916
-reduced_3dgs-1.9.1.dist-info/METADATA,sha256=TIrUop8Y3QjsjnVAxNEGADwBMtVArYSbYhhzMYLWMYc,13014
-reduced_3dgs-1.9.1.dist-info/WHEEL,sha256=6TCqs2-PEvxKuqw0d1ghd6tAo1nX2NnQFqKrzb5XC4g,101
-reduced_3dgs-1.9.1.dist-info/top_level.txt,sha256=PpU5aT3-baSCdqCtTaZknoB32H93UeKCkYDkRCCZMEI,13
-reduced_3dgs-1.9.1.dist-info/RECORD,,
+reduced_3dgs/simple_knn/_C.cp312-win_amd64.pyd,sha256=evEEV6-Cd79Dlatq_ivuPowmOjT33QPz1Wfl_O3cQ4c,1267712
+reduced_3dgs-1.9.3.dist-info/licenses/LICENSE.md,sha256=LQ4_LAqlncGkg_mQy5ykMAFtQDSPB0eKmIEtBut0yjw,4916
+reduced_3dgs-1.9.3.dist-info/METADATA,sha256=F5CGIMF7fBLTpRvKqrHtbIhy7H15wEEuvPeSklClMEg,13014
+reduced_3dgs-1.9.3.dist-info/WHEEL,sha256=6TCqs2-PEvxKuqw0d1ghd6tAo1nX2NnQFqKrzb5XC4g,101
+reduced_3dgs-1.9.3.dist-info/top_level.txt,sha256=PpU5aT3-baSCdqCtTaZknoB32H93UeKCkYDkRCCZMEI,13
+reduced_3dgs-1.9.3.dist-info/RECORD,,