reduced-3dgs 1.8.12__tar.gz → 1.8.14__tar.gz

{reduced_3dgs-1.8.12 → reduced_3dgs-1.8.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reduced_3dgs
-Version: 1.8.12
+Version: 1.8.14
 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.12 → reduced_3dgs-1.8.14}/reduced_3dgs/combinations.py

@@ -2,7 +2,7 @@ from gaussian_splatting import GaussianModel, CameraTrainableGaussianModel, Came
 from gaussian_splatting.dataset import CameraDataset, TrainableCameraDataset
 from gaussian_splatting.trainer import OpacityResetDensificationTrainer
 # from gaussian_splatting.trainer import BaseOpacityResetDensificationTrainer as OpacityResetDensificationTrainer
-from gaussian_splatting.trainer import OpacityResetter, CameraTrainerWrapper
+from gaussian_splatting.trainer import OpacityResetTrainerWrapper, CameraTrainerWrapper
 from .shculling import VariableSHGaussianModel, SHCullingTrainerWrapper
 from .shculling import SHCullingTrainer
 # from .shculling import BaseSHCullingTrainer as SHCullingTrainer
@@ -14,24 +14,11 @@ def OpacityResetPruningTrainer(
         model: GaussianModel,
         scene_extent: float,
         dataset: CameraDataset,
-        opacity_reset_from_iter=3000,
-        opacity_reset_until_iter=15000,
-        # these params are better than default ones
-        mercy_type='redundancy_opacity_opacity',  # replace the default 'redundancy_opacity'. therefore work with opacity reset (prune low-opacity points)
-        opacity_reset_interval=500,  # replace the default 100. since prune by opacity, donot prune too fast (train enough after opacity reset)
-        prune_interval=500,  # replace the default 100. since prune by opacity, donot prune too fast (train enough after opacity reset)
         *args, **kwargs):
-    return OpacityResetter(
-        PruningTrainer(
-            model, scene_extent, dataset,
-            *args,
-            prune_interval=prune_interval,
-            mercy_type=mercy_type,
-            **kwargs
-        ),
-        opacity_reset_from_iter=opacity_reset_from_iter,
-        opacity_reset_until_iter=opacity_reset_until_iter,
-        opacity_reset_interval=opacity_reset_interval
+    return OpacityResetTrainerWrapper(
+        lambda model, scene_extent, *args, **kwargs: PruningTrainer(model, scene_extent, dataset, *args, **kwargs),
+        model, scene_extent,
+        *args, **kwargs
     )
 
 
@@ -39,24 +26,11 @@ def OpacityResetPrunerInDensifyTrainer(
         model: GaussianModel,
         scene_extent: float,
         dataset: CameraDataset,
-        opacity_reset_from_iter=3000,
-        opacity_reset_until_iter=15000,
-        # these params are better than default ones
-        mercy_type='redundancy_opacity_opacity',  # replace the default 'redundancy_opacity'. therefore work with opacity reset (prune low-opacity points)
-        opacity_reset_interval=500,  # replace the default 100. since prune by opacity, donot prune too fast (train enough after opacity reset)
-        prune_interval=500,  # replace the default 100. since prune by opacity, donot prune too fast (train enough after opacity reset)
         *args, **kwargs):
-    return OpacityResetter(
-        PrunerInDensifyTrainer(
-            model, scene_extent, dataset,
-            *args,
-            prune_interval=prune_interval,
-            mercy_type=mercy_type,
-            **kwargs
-        ),
-        opacity_reset_from_iter=opacity_reset_from_iter,
-        opacity_reset_until_iter=opacity_reset_until_iter,
-        opacity_reset_interval=opacity_reset_interval
+    return OpacityResetTrainerWrapper(
+        lambda model, scene_extent, *args, **kwargs: PrunerInDensifyTrainer(model, scene_extent, dataset, *args, **kwargs),
+        model, scene_extent,
+        *args, **kwargs
     )
 
 

{reduced_3dgs-1.8.12 → reduced_3dgs-1.8.14}/reduced_3dgs/pruning/trainer.py

@@ -84,7 +84,7 @@ class BasePruner(AbstractDensifier):
             self, model: GaussianModel, dataset: List[Camera],
             prune_from_iter=1000,
             prune_until_iter=15000,
-            prune_interval: int = 100,
+            prune_interval: int = 500,
             box_size=1.,
             lambda_mercy=1.,
             mercy_minimum=3,
@@ -115,7 +115,7 @@ def BasePruningTrainer(
         dataset: List[Camera],
         prune_from_iter=1000,
         prune_until_iter=15000,
-        prune_interval: int = 100,
+        prune_interval: int = 500,
         box_size=1.,
         lambda_mercy=1.,
         mercy_minimum=3,
@@ -172,7 +172,7 @@ def BasePrunerInDensifyTrainer(
 
         prune_from_iter=1000,
         prune_until_iter=15000,
-        prune_interval=100,
+        prune_interval=500,
         prune_screensize_threshold=20,
         prune_percent_too_big=1,
 

{reduced_3dgs-1.8.12 → reduced_3dgs-1.8.14}/reduced_3dgs.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reduced_3dgs
-Version: 1.8.12
+Version: 1.8.14
 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.12 → reduced_3dgs-1.8.14}/setup.py

@@ -49,7 +49,7 @@ if os.name == 'nt':
 
 setup(
     name="reduced_3dgs",
-    version='1.8.12',
+    version='1.8.14',
     author='yindaheng98',
     author_email='yindaheng98@gmail.com',
     url='https://github.com/yindaheng98/reduced-3dgs',