reduced-3dgs 1.9.2__cp311-cp311-win_amd64.whl → 1.9.4__cp311-cp311-win_amd64.whl

reduced_3dgs/combinations.py

@@ -17,9 +17,19 @@ def BaseFullPruningTrainer(
         scene_extent: float,
         dataset: List[Camera],
         *args,
-        importance_prune_from_iter=1000,
-        importance_prune_until_iter=15000,
-        importance_prune_interval: int = 100,
+        importance_prune_from_iter=15000,
+        importance_prune_until_iter=20000,
+        importance_prune_interval: int = 1000,
+        importance_score_resize=None,
+        importance_prune_type="comprehensive",
+        importance_prune_percent=0.1,
+        importance_prune_thr_important_score=None,
+        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=1.0,
+        importance_prune_thr_T_alpha_avg=0.001,
+        importance_v_pow=0.1,
         **kwargs):
     return PruningTrainerWrapper(
         lambda model, scene_extent, dataset: ImportancePruner(
@@ -28,6 +38,16 @@ def BaseFullPruningTrainer(
             importance_prune_from_iter=importance_prune_from_iter,
             importance_prune_until_iter=importance_prune_until_iter,
             importance_prune_interval=importance_prune_interval,
+            importance_score_resize=importance_score_resize,
+            importance_prune_type=importance_prune_type,
+            importance_prune_percent=importance_prune_percent,
+            importance_prune_thr_important_score=importance_prune_thr_important_score,
+            importance_prune_thr_v_important_score=importance_prune_thr_v_important_score,
+            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, scene_extent, dataset,
         *args, **kwargs
@@ -39,9 +59,19 @@ def BaseFullPrunerInDensifyTrainer(
         scene_extent: float,
         dataset: List[Camera],
         *args,
-        importance_prune_from_iter=1000,
-        importance_prune_until_iter=15000,
-        importance_prune_interval: int = 100,
+        importance_prune_from_iter=15000,
+        importance_prune_until_iter=20000,
+        importance_prune_interval: int = 1000,
+        importance_score_resize=None,
+        importance_prune_type="comprehensive",
+        importance_prune_percent=0.1,
+        importance_prune_thr_important_score=None,
+        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=1.0,
+        importance_prune_thr_T_alpha_avg=0.001,
+        importance_v_pow=0.1,
         **kwargs):
     return PrunerInDensifyTrainerWrapper(
         lambda model, scene_extent, dataset: ImportancePruner(
@@ -50,6 +80,16 @@ def BaseFullPrunerInDensifyTrainer(
             importance_prune_from_iter=importance_prune_from_iter,
             importance_prune_until_iter=importance_prune_until_iter,
             importance_prune_interval=importance_prune_interval,
+            importance_score_resize=importance_score_resize,
+            importance_prune_type=importance_prune_type,
+            importance_prune_percent=importance_prune_percent,
+            importance_prune_thr_important_score=importance_prune_thr_important_score,
+            importance_prune_thr_v_important_score=importance_prune_thr_v_important_score,
+            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, scene_extent, dataset,
         *args, **kwargs

reduced_3dgs/importance/combinations.py

@@ -13,13 +13,15 @@ def BaseImportancePrunerInDensifyTrainer(
         importance_prune_from_iter=15000,
         importance_prune_until_iter=20000,
         importance_prune_interval: int = 1000,
+        importance_score_resize=None,
         importance_prune_type="comprehensive",
         importance_prune_percent=0.1,
         importance_prune_thr_important_score=None,
         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.1,
+        importance_prune_thr_T_alpha=1.0,
+        importance_prune_thr_T_alpha_avg=0.001,
         importance_v_pow=0.1,
         **kwargs):
     return DensificationTrainerWrapper(
@@ -29,6 +31,7 @@ def BaseImportancePrunerInDensifyTrainer(
             importance_prune_from_iter=importance_prune_from_iter,
             importance_prune_until_iter=importance_prune_until_iter,
             importance_prune_interval=importance_prune_interval,
+            importance_score_resize=importance_score_resize,
             importance_prune_type=importance_prune_type,
             importance_prune_percent=importance_prune_percent,
             importance_prune_thr_important_score=importance_prune_thr_important_score,
@@ -36,6 +39,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

@@ -1,8 +1,9 @@
 import math
-from typing import Callable, List
+from typing import List
 import torch
 
 from gaussian_splatting import Camera, GaussianModel
+from gaussian_splatting.camera import build_camera
 from gaussian_splatting.trainer import AbstractDensifier, DensifierWrapper, DensificationTrainer, NoopDensifier
 from gaussian_splatting.dataset import CameraDataset
 from .diff_gaussian_rasterization import GaussianRasterizationSettings, GaussianRasterizer
@@ -75,11 +76,20 @@ def count_render(self: GaussianModel, viewpoint_camera: Camera):
     }
 
 
-def prune_list(model: GaussianModel, dataset: CameraDataset):
+def prune_list(model: GaussianModel, dataset: CameraDataset, resize=None):
     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)
     for camera in dataset:
+        if resize is not None:
+            height, width = camera.image_height, camera.image_width
+            scale = resize / max(height, width)
+            height, width = int(height * scale), int(width * scale)
+            camera = build_camera(
+                image_height=height, image_width=width,
+                FoVx=camera.FoVx, FoVy=camera.FoVy,
+                R=camera.R, T=camera.T,
+                device=camera.R.device)
         out = count_render(model, camera)
         gaussian_count += out["gaussians_count"]
         opacity_important_score += out["opacity_important_score"]
@@ -118,6 +128,7 @@ def score2mask(percent, import_score: list, threshold=None):
 
 def prune_gaussians(
         gaussians: GaussianModel, dataset: CameraDataset,
+        resize=None,
         prune_type="comprehensive",
         prune_percent=0.1,
         prune_thr_important_score=None,
@@ -125,8 +136,9 @@ def prune_gaussians(
         prune_thr_max_v_important_score=None,
         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)
+    gaussian_list, opacity_imp_list, T_alpha_imp_list = prune_list(gaussians, dataset, resize)
     match prune_type:
         case "important_score":
             mask = score2mask(prune_percent, opacity_imp_list, prune_thr_important_score)
@@ -141,6 +153,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 +171,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
@@ -167,25 +187,29 @@ class ImportancePruner(DensifierWrapper):
             importance_prune_from_iter=15000,
             importance_prune_until_iter=20000,
             importance_prune_interval: int = 1000,
+            importance_score_resize=None,
             importance_prune_type="comprehensive",
             importance_prune_percent=0.1,
             importance_prune_thr_important_score=None,
             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.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
         self.importance_prune_until_iter = importance_prune_until_iter
         self.importance_prune_interval = importance_prune_interval
+        self.resize = importance_score_resize
         self.prune_percent = importance_prune_percent
         self.prune_thr_important_score = importance_prune_thr_important_score
         self.prune_thr_v_important_score = importance_prune_thr_v_important_score
         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
 
@@ -194,10 +218,11 @@ class ImportancePruner(DensifierWrapper):
         if self.importance_prune_from_iter <= step <= self.importance_prune_until_iter and step % self.importance_prune_interval == 0:
             remove_mask = prune_gaussians(
                 self.model, self.dataset,
+                self.resize,
                 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
@@ -211,13 +236,15 @@ def BaseImportancePruningTrainer(
         importance_prune_from_iter=15000,
         importance_prune_until_iter=20000,
         importance_prune_interval: int = 1000,
+        importance_score_resize=None,
         importance_prune_type="comprehensive",
         importance_prune_percent=0.1,
         importance_prune_thr_important_score=None,
         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.1,
+        importance_prune_thr_T_alpha=1.0,
+        importance_prune_thr_T_alpha_avg=0.001,
         importance_v_pow=0.1,
         **kwargs):
     return DensificationTrainer(
@@ -228,6 +255,7 @@ def BaseImportancePruningTrainer(
             importance_prune_from_iter=importance_prune_from_iter,
             importance_prune_until_iter=importance_prune_until_iter,
             importance_prune_interval=importance_prune_interval,
+            importance_score_resize=importance_score_resize,
             importance_prune_type=importance_prune_type,
             importance_prune_percent=importance_prune_percent,
             importance_prune_thr_important_score=importance_prune_thr_important_score,
@@ -235,6 +263,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.2.dist-info → reduced_3dgs-1.9.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reduced_3dgs
-Version: 1.9.2
+Version: 1.9.4
 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.2.dist-info → reduced_3dgs-1.9.4.dist-info}/RECORD

@@ -1,13 +1,13 @@
 reduced_3dgs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-reduced_3dgs/combinations.py,sha256=FrZLxd3AZlHPSq_WJeEdGWH5zh40rAuV5txxr8HsSPY,7031
+reduced_3dgs/combinations.py,sha256=k4ErxpAscCqJMtVXZ29KGQjw2VoZMV0W3V4u3nj0e-Y,9401
 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=8anBqusRxxIydY2ZUbYybjAx40x520eHwBroXS_x5Cc,1632256
+reduced_3dgs/diff_gaussian_rasterization/_C.cp311-win_amd64.pyd,sha256=mNCYXaE2Dq-zOfK19nMCcdVx2WvXTZqx574PqK9canY,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=PFo1X1nx2BG53HJjWC8xUWwHTjNipt--I3Rw1fhcEHw,2692
-reduced_3dgs/importance/trainer.py,sha256=gk14lGbVOmtlqBR5uu1UF9K669RZ1VY05E0g4qxQeng,11147
-reduced_3dgs/importance/diff_gaussian_rasterization/_C.cp311-win_amd64.pyd,sha256=mX7f5wO1lvsmglO_aqTtyPv2LwOUf8fqPvDNfoKjTxY,1315840
+reduced_3dgs/importance/combinations.py,sha256=eAdykeTdvRGCHxskjILQnZVaqQVvwC-0wMxdgYMeeDs,2922
+reduced_3dgs/importance/trainer.py,sha256=Sj4ORvoYtFT7z3hifzFZDfhFyqumHraXyk3vMVtk0AU,12661
+reduced_3dgs/importance/diff_gaussian_rasterization/_C.cp311-win_amd64.pyd,sha256=OvadjumWCD_wdPZuwU90YR1G11p1tNOjGsVlAsIfjDg,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=o2Td6A6t-GL_m93kXhzJxSyAYdMMCgvpXbFXbMNviic,1263616
-reduced_3dgs-1.9.2.dist-info/licenses/LICENSE.md,sha256=LQ4_LAqlncGkg_mQy5ykMAFtQDSPB0eKmIEtBut0yjw,4916
-reduced_3dgs-1.9.2.dist-info/METADATA,sha256=CMTy9CnY3qY318sKtQeM4usFKoD_RbTa1Qq0NKkCd5Y,13014
-reduced_3dgs-1.9.2.dist-info/WHEEL,sha256=ZhKniLpVlRfVU2a9ty7N0kTvsYDNee6K-jhM1zk8cMw,101
-reduced_3dgs-1.9.2.dist-info/top_level.txt,sha256=PpU5aT3-baSCdqCtTaZknoB32H93UeKCkYDkRCCZMEI,13
-reduced_3dgs-1.9.2.dist-info/RECORD,,
+reduced_3dgs/simple_knn/_C.cp311-win_amd64.pyd,sha256=ZeG0NTTKvaG1CKRE0o9rE_2uuVfMiYkVogeuReFW8dI,1263616
+reduced_3dgs-1.9.4.dist-info/licenses/LICENSE.md,sha256=LQ4_LAqlncGkg_mQy5ykMAFtQDSPB0eKmIEtBut0yjw,4916
+reduced_3dgs-1.9.4.dist-info/METADATA,sha256=AxCPAYi-0sHw1PwzbbNWsK5GL1pdc8VoCnzSYXcjFik,13014
+reduced_3dgs-1.9.4.dist-info/WHEEL,sha256=ge07nwgZNrHM7F-h6ntB4NDhu5ar8XLYRt0vb6q09YM,101
+reduced_3dgs-1.9.4.dist-info/top_level.txt,sha256=PpU5aT3-baSCdqCtTaZknoB32H93UeKCkYDkRCCZMEI,13
+reduced_3dgs-1.9.4.dist-info/RECORD,,

{reduced_3dgs-1.9.2.dist-info → reduced_3dgs-1.9.4.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.0.0)
+Generator: setuptools (80.0.1)
 Root-Is-Purelib: false
 Tag: cp311-cp311-win_amd64