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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reduced_3dgs
-Version: 1.8.14
+Version: 1.8.15
 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.14 → reduced_3dgs-1.8.15}/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 = 500,
+            prune_interval: int = 100,
             box_size=1.,
             lambda_mercy=1.,
             mercy_minimum=3,
@@ -104,7 +104,7 @@ class BasePruner(AbstractDensifier):
         return self._model
 
     def densify_and_prune(self, loss, out, camera, step: int) -> DensificationInstruct:
-        if self.prune_from_iter < step < self.prune_until_iter and step % self.prune_interval == 0:
+        if self.prune_from_iter <= step <= self.prune_until_iter and step % self.prune_interval == 0:
             return DensificationInstruct(remove_mask=mercy_gaussians(self.model, self.dataset, self.box_size, self.lambda_mercy, self.mercy_minimum, self.mercy_type))
         return DensificationInstruct()
 
@@ -115,7 +115,7 @@ def BasePruningTrainer(
         dataset: List[Camera],
         prune_from_iter=1000,
         prune_until_iter=15000,
-        prune_interval: int = 500,
+        prune_interval: int = 100,
         box_size=1.,
         lambda_mercy=1.,
         mercy_minimum=3,
@@ -147,9 +147,7 @@ class PrunerInDensify(Densifier):
         self.mercy_type = mercy_type
 
     def prune(self) -> torch.Tensor:
-        _splatted_num_accum, _ = calculate_redundancy_metric(self.model, self.dataset, pixel_scale=self.box_size)
-        mask = mercy_points(self.model, _splatted_num_accum.squeeze(), self.lambda_mercy, self.mercy_minimum, self.mercy_type)
-        return torch.logical_or(mask, super().prune())
+        return torch.logical_or(mercy_gaussians(self.model, self.dataset, self.box_size, self.lambda_mercy, self.mercy_minimum, self.mercy_type), super().prune())
 
 
 def BasePrunerInDensifyTrainer(
@@ -172,7 +170,7 @@ def BasePrunerInDensifyTrainer(
 
         prune_from_iter=1000,
         prune_until_iter=15000,
-        prune_interval=500,
+        prune_interval=100,
         prune_screensize_threshold=20,
         prune_percent_too_big=1,
 

{reduced_3dgs-1.8.14 → reduced_3dgs-1.8.15}/reduced_3dgs/quantization/abc.py

@@ -42,7 +42,7 @@ class QuantizeTrainerWrapper(TrainerWrapper, metaclass=abc.ABCMeta):
 
     @property
     def model(self) -> GaussianModel:
-        if self.quantize_from_iter <= self.curr_step < self.quantize_until_iter and self.curr_step % self.quantize_interval == 0:
+        if self.quantize_from_iter <= self.curr_step <= self.quantize_until_iter and self.curr_step % self.quantize_interval == 0:
             with torch.no_grad():
                 ids_dict, codebook_dict = self.quantizer.quantize(self.base_trainer.model, update_codebook=True)
                 return self.quantizer.dequantize(self.base_trainer.model, ids_dict, codebook_dict)

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

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

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