reduced-3dgs 1.9.4__tar.gz → 1.10.0__tar.gz

{reduced_3dgs-1.9.4 → reduced_3dgs-1.10.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reduced_3dgs
-Version: 1.9.4
+Version: 1.10.0
 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.4 → reduced_3dgs-1.10.0}/reduced_3dgs/pruning/combinations.py

@@ -23,7 +23,7 @@ def PrunerInDensifyTrainerWrapper(
     return SplitCloneDensifierTrainerWrapper(
         lambda model, scene_extent: BasePruner(
             noargs_base_densifier_constructor(model, scene_extent, dataset),
-            dataset,
+            scene_extent, dataset,
             prune_from_iter=prune_from_iter,
             prune_until_iter=prune_until_iter,
             prune_interval=prune_interval,

{reduced_3dgs-1.9.4 → reduced_3dgs-1.10.0}/reduced_3dgs/pruning/trainer.py

@@ -1,7 +1,7 @@
 from typing import Callable, List
 import torch
 from gaussian_splatting import GaussianModel, Camera
-from gaussian_splatting.trainer import AbstractDensifier, DensifierWrapper, DensificationTrainer, NoopDensifier
+from gaussian_splatting.trainer import AbstractDensifier, OpacityPruner, DensificationTrainer, NoopDensifier
 from reduced_3dgs.diff_gaussian_rasterization._C import sphere_ellipsoid_intersection, allocate_minimum_redundancy_value, find_minimum_projected_pixel_size
 from reduced_3dgs.simple_knn._C import distIndex2
 
@@ -79,33 +79,28 @@ def mercy_gaussians(
     return mask
 
 
-class BasePruner(DensifierWrapper):
+class BasePruner(OpacityPruner):
     def __init__(
             self, base_densifier: AbstractDensifier,
+            scene_extent,
             dataset: List[Camera],
-            prune_from_iter=1000,
-            prune_until_iter=15000,
-            prune_interval: int = 100,
+            *args,
             box_size=1.,
             lambda_mercy=1.,
             mercy_minimum=3,
-            mercy_type='redundancy_opacity'):
-        super().__init__(base_densifier)
+            mercy_type='redundancy_opacity',
+            **kwargs):
+        super().__init__(base_densifier, scene_extent, *args, **kwargs)
         self.dataset = dataset
-        self.prune_from_iter = prune_from_iter
-        self.prune_until_iter = prune_until_iter
-        self.prune_interval = prune_interval
         self.box_size = box_size
         self.lambda_mercy = lambda_mercy
         self.mercy_minimum = mercy_minimum
         self.mercy_type = mercy_type
 
-    def densify_and_prune(self, loss, out, camera, step: int):
-        ret = super().densify_and_prune(loss, out, camera, step)
-        if self.prune_from_iter <= step <= self.prune_until_iter and step % self.prune_interval == 0:
-            remove_mask = mercy_gaussians(self.model, self.dataset, self.box_size, self.lambda_mercy, self.mercy_minimum, self.mercy_type)
-            ret = ret._replace(remove_mask=remove_mask if ret.remove_mask is None else torch.logical_or(remove_mask, ret.remove_mask))
-        return ret
+    def prune(self) -> torch.Tensor:
+        remove_mask = mercy_gaussians(self.model, self.dataset, self.box_size, self.lambda_mercy, self.mercy_minimum, self.mercy_type)
+        prune_mask = torch.logical_or(super().prune(), remove_mask)
+        return prune_mask
 
 
 def PruningTrainerWrapper(
@@ -126,7 +121,7 @@ def PruningTrainerWrapper(
         model, scene_extent,
         BasePruner(
             noargs_base_densifier_constructor(model, scene_extent, dataset),
-            dataset,
+            scene_extent, dataset,
             prune_from_iter=prune_from_iter,
             prune_until_iter=prune_until_iter,
             prune_interval=prune_interval,

{reduced_3dgs-1.9.4 → reduced_3dgs-1.10.0}/reduced_3dgs/quantization/quantizer.py

@@ -119,10 +119,12 @@ class VectorQuantizer(AbstractQuantizer):
         return self.one_nearst(model._opacity.detach(), codebook)
 
     def produce_clusters_scaling(self, model: GaussianModel, *args, **kwargs):
-        return self.generate_codebook(model.get_scaling.detach(), self.num_clusters_scaling, *args, **kwargs)
+        centers, ids = self.generate_codebook(model.get_scaling.detach(), self.num_clusters_scaling, *args, **kwargs)
+        centers_log = model.scaling_inverse_activation(centers)
+        return centers_log, ids
 
     def find_nearest_cluster_id_scaling(self, model: GaussianModel, codebook: torch.Tensor):
-        return self.one_nearst(model.get_scaling.detach(), codebook)
+        return self.one_nearst(model.get_scaling.detach(), model.scaling_activation(codebook))
 
     def produce_clusters(self, model: GaussianModel, init_codebook_dict={}):
         codebook_dict: Dict[str, torch.Tensor] = {}
@@ -163,7 +165,7 @@ class VectorQuantizer(AbstractQuantizer):
 
     def dequantize(self, model: GaussianModel, ids_dict: Dict[str, torch.Tensor], codebook_dict: Dict[str, torch.Tensor], xyz: torch.Tensor = None, replace=False) -> GaussianModel:
         opacity = codebook_dict["opacity"][ids_dict["opacity"], ...]
-        scaling = model.scaling_inverse_activation(codebook_dict["scaling"][ids_dict["scaling"], ...])
+        scaling = codebook_dict["scaling"][ids_dict["scaling"], ...]
 
         rotation = torch.cat((
             codebook_dict["rotation_re"][ids_dict["rotation_re"], ...],

{reduced_3dgs-1.9.4 → reduced_3dgs-1.10.0}/reduced_3dgs.egg-info/PKG-INFO

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

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