reduced-3dgs 1.8.13__cp311-cp311-win_amd64.whl → 1.8.15__cp311-cp311-win_amd64.whl

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()
 
@@ -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(

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.13.dist-info → reduced_3dgs-1.8.15.dist-info}/METADATA

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

@@ -2,22 +2,22 @@ reduced_3dgs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reduced_3dgs/combinations.py,sha256=YDDNRGQlOuLsXnOOMiZ6mqG0cx1ICtG7jz6ObVx4OUo,4794
 reduced_3dgs/quantize.py,sha256=Y44qHyFdOIqke7NoeqXmyKloS43j-al74ZiNsuZZHbM,2527
 reduced_3dgs/train.py,sha256=WdxDg2MC4HCwIaGta3c7oYVPA9gb6CMqHLDD6WeOwEk,9829
-reduced_3dgs/diff_gaussian_rasterization/_C.cp311-win_amd64.pyd,sha256=lMjTMQsxU6Mamjp8u1p4W_s-Syp9pqeKw_zhP93IAf8,1632256
+reduced_3dgs/diff_gaussian_rasterization/_C.cp311-win_amd64.pyd,sha256=ob7KO4nwHWZQDRwEd_VNht2olUqX-Fmsn-rKhSKfW7Y,1632256
 reduced_3dgs/diff_gaussian_rasterization/__init__.py,sha256=oV6JjTc-50MscX4XHeIWSgLr3l8Y25knBIs-0gRbJr4,7932
 reduced_3dgs/pruning/__init__.py,sha256=T6vvO5RFMFUfHl2k7kleZ_AGgDwyMBfc7jxNJ-CRULA,164
 reduced_3dgs/pruning/combinations.py,sha256=ZaqvmmWsl1pJJZbA_yJOJVbv-MoQHLO8Lts3KT-5r_Q,833
-reduced_3dgs/pruning/trainer.py,sha256=MhzzMnnZTAj8mNCV_3_WopU_15V0016JqxjSm2ISd0U,8663
+reduced_3dgs/pruning/trainer.py,sha256=5mOfYPEO6ZOW5ZSeI33LoN4EmGYxWKz2pMQw8RMRiEU,8530
 reduced_3dgs/quantization/__init__.py,sha256=1z1xMn3yj9u7cR9JizGrI3WSyIES_Tqq6oDquvglSeo,225
-reduced_3dgs/quantization/abc.py,sha256=DELX8tbADosnlqSjn-zIQMYDduuCPWRrZbM-iEet5jw,1909
+reduced_3dgs/quantization/abc.py,sha256=rsi8HFRwQCltWTYiJ3BpygtQDT7hK6J01jKMOboOY8w,1910
 reduced_3dgs/quantization/exclude_zeros.py,sha256=fKSgjHous4OpdI6mQi9z23if9jnbB79w2jChpxkCJWw,2381
 reduced_3dgs/quantization/quantizer.py,sha256=-B07X1VrV7E1fT_NXXlZ0FLAb-c_jBUlZ8274KwHyVw,16614
 reduced_3dgs/quantization/wrapper.py,sha256=cyXqfJgo9b3fS7DYXxOk5LmQudvrEhweOebFsjRnXiQ,2549
 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=JW0nyeUvJ3kYYmD1U1ptlt9tURi_dwzxU2xhZyi_OZw,1263616
-reduced_3dgs-1.8.13.dist-info/licenses/LICENSE.md,sha256=LQ4_LAqlncGkg_mQy5ykMAFtQDSPB0eKmIEtBut0yjw,4916
-reduced_3dgs-1.8.13.dist-info/METADATA,sha256=SQt1RzUaj3ESrgbQQCvV4ehBpDX6RlLdDmHYaVPQKCQ,13015
-reduced_3dgs-1.8.13.dist-info/WHEEL,sha256=oEjOS__eymZOlbVKA4cO_63xUK8u3m8eSPjzCdMe61I,101
-reduced_3dgs-1.8.13.dist-info/top_level.txt,sha256=PpU5aT3-baSCdqCtTaZknoB32H93UeKCkYDkRCCZMEI,13
-reduced_3dgs-1.8.13.dist-info/RECORD,,
+reduced_3dgs/simple_knn/_C.cp311-win_amd64.pyd,sha256=aIz8AsSTKVNFiZ8gv24AIF-JhSSsavNKHHiWNH0MO5Y,1263616
+reduced_3dgs-1.8.15.dist-info/licenses/LICENSE.md,sha256=LQ4_LAqlncGkg_mQy5ykMAFtQDSPB0eKmIEtBut0yjw,4916
+reduced_3dgs-1.8.15.dist-info/METADATA,sha256=0O3KJo4RTyBSVD47eD6gGTGg3MgxDSYAJLDs_-vMm3U,13015
+reduced_3dgs-1.8.15.dist-info/WHEEL,sha256=ZhKniLpVlRfVU2a9ty7N0kTvsYDNee6K-jhM1zk8cMw,101
+reduced_3dgs-1.8.15.dist-info/top_level.txt,sha256=PpU5aT3-baSCdqCtTaZknoB32H93UeKCkYDkRCCZMEI,13
+reduced_3dgs-1.8.15.dist-info/RECORD,,

{reduced_3dgs-1.8.13.dist-info → reduced_3dgs-1.8.15.dist-info}/WHEEL

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