reduced-3dgs 1.10.3__tar.gz → 1.10.5__tar.gz

{reduced_3dgs-1.10.3 → reduced_3dgs-1.10.5}/PKG-INFO

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

@@ -97,11 +97,17 @@ def BaseFullPrunerInDensifyTrainer(
 
 
 def DepthFullPruningTrainer(model: GaussianModel, scene_extent: float, dataset: TrainableCameraDataset, *args, **kwargs):
-    return DepthTrainerWrapper(BaseFullPruningTrainer, model, scene_extent, *args, dataset=dataset, **kwargs)
+    return DepthTrainerWrapper(
+        BaseFullPruningTrainer,
+        model, scene_extent, dataset,
+        *args, **kwargs)
 
 
 def DepthFullPrunerInDensifyTrainer(model: GaussianModel, scene_extent: float, dataset: TrainableCameraDataset, *args, **kwargs):
-    return DepthTrainerWrapper(BaseFullPrunerInDensifyTrainer, model, scene_extent, *args, dataset=dataset, **kwargs)
+    return DepthTrainerWrapper(
+        BaseFullPrunerInDensifyTrainer,
+        model, scene_extent, dataset,
+        *args, **kwargs)
 
 
 def OpacityResetPruningTrainer(

{reduced_3dgs-1.10.3 → reduced_3dgs-1.10.5}/reduced_3dgs/importance/combinations.py

@@ -52,11 +52,17 @@ def BaseImportancePrunerInDensifyTrainer(
 
 
 def DepthImportancePruningTrainer(model: GaussianModel, scene_extent: float, dataset: TrainableCameraDataset, *args, **kwargs):
-    return DepthTrainerWrapper(BaseImportancePruningTrainer, model, scene_extent, *args, dataset=dataset, **kwargs)
+    return DepthTrainerWrapper(
+        BaseImportancePruningTrainer,
+        model, scene_extent, dataset,
+        *args, **kwargs)
 
 
 def DepthImportancePrunerInDensifyTrainer(model: GaussianModel, scene_extent: float, dataset: TrainableCameraDataset, *args, **kwargs):
-    return DepthTrainerWrapper(BaseImportancePrunerInDensifyTrainer, model, scene_extent, *args, dataset=dataset, **kwargs)
+    return DepthTrainerWrapper(
+        BaseImportancePrunerInDensifyTrainer,
+        model, scene_extent, dataset,
+        *args, **kwargs)
 
 
 ImportancePruningTrainer = DepthImportancePruningTrainer

{reduced_3dgs-1.10.3 → reduced_3dgs-1.10.5}/reduced_3dgs/pruning/combinations.py

@@ -54,11 +54,17 @@ def BasePrunerInDensifyTrainer(
 
 
 def DepthPruningTrainer(model: GaussianModel, scene_extent: float, dataset: TrainableCameraDataset, *args, **kwargs):
-    return DepthTrainerWrapper(BasePruningTrainer, model, scene_extent, *args, dataset=dataset, **kwargs)
+    return DepthTrainerWrapper(
+        BasePruningTrainer,
+        model, scene_extent, dataset,
+        *args, **kwargs)
 
 
 def DepthPrunerInDensifyTrainer(model: GaussianModel, scene_extent: float, dataset: TrainableCameraDataset, *args, **kwargs):
-    return DepthTrainerWrapper(BasePrunerInDensifyTrainer, model, scene_extent, *args, dataset=dataset, **kwargs)
+    return DepthTrainerWrapper(
+        BasePrunerInDensifyTrainer,
+        model, scene_extent, dataset,
+        *args, **kwargs)
 
 
 PruningTrainer = DepthPruningTrainer

{reduced_3dgs-1.10.3 → reduced_3dgs-1.10.5}/reduced_3dgs/quantization/quantizer.py

@@ -262,36 +262,46 @@ class VectorQuantizer(AbstractQuantizer):
 
         PlyData([el, *cb]).write(ply_path)
 
-    def load_quantized(self, model: GaussianModel, ply_path: str) -> GaussianModel:
-        plydata = PlyData.read(ply_path)
-
+    def parse_ids(self, plydata: PlyData, max_sh_degree: int, device: torch.device) -> Dict[str, torch.Tensor]:
         ids_dict = {}
         elements = plydata['vertex']
-        kwargs = dict(dtype=torch.long, device=model._xyz.device)
+        kwargs = dict(dtype=torch.long, device=device)
         ids_dict["rotation_re"] = torch.tensor(elements["rot_re"].copy(), **kwargs)
         ids_dict["rotation_im"] = torch.tensor(elements["rot_im"].copy(), **kwargs)
         ids_dict["opacity"] = torch.tensor(elements["opacity"].copy(), **kwargs)
         ids_dict["scaling"] = torch.tensor(elements["scale"].copy(), **kwargs)
         ids_dict["features_dc"] = torch.tensor(elements["f_dc"].copy(), **kwargs).unsqueeze(-1)
-        for sh_degree in range(model.max_sh_degree):
+        for sh_degree in range(max_sh_degree):
             ids_dict[f'features_rest_{sh_degree}'] = torch.tensor(np.stack([elements[f'f_rest_{sh_degree}_{ch}'] for ch in range(3)], axis=1), **kwargs)
+        return ids_dict
 
+    def parse_codebook(self, plydata: PlyData, max_sh_degree: int, device: torch.device) -> Dict[str, torch.Tensor]:
         codebook_dict = {}
-        kwargs = dict(dtype=torch.float32, device=model._xyz.device)
+        kwargs = dict(dtype=torch.float32, device=device)
         codebook_dict["rotation_re"] = torch.tensor(plydata["codebook_rot_re"]["rot_re"], **kwargs).unsqueeze(-1)
         codebook_dict["rotation_im"] = torch.tensor(np.stack([plydata["codebook_rot_im"][f'rot_im_{ch}'] for ch in range(3)], axis=1), **kwargs)
         codebook_dict["opacity"] = torch.tensor(plydata["codebook_opacity"]["opacity"], **kwargs).unsqueeze(-1)
         codebook_dict["scaling"] = torch.tensor(np.stack([plydata["codebook_scaling"][f'scaling_{ch}'] for ch in range(3)], axis=1), **kwargs)
         codebook_dict["features_dc"] = torch.tensor(np.stack([plydata["codebook_f_dc"][f'f_dc_{ch}'] for ch in range(3)], axis=1), **kwargs)
-        for sh_degree in range(model.max_sh_degree):
+        for sh_degree in range(max_sh_degree):
             n_channels = (sh_degree + 2) ** 2 - (sh_degree + 1) ** 2
             codebook_dict[f'features_rest_{sh_degree}'] = torch.tensor(np.stack([plydata[f"codebook_f_rest_{sh_degree}"][f'f_rest_{sh_degree}_{ch}'] for ch in range(n_channels)], axis=1), **kwargs)
+        return codebook_dict
 
-        self._codebook_dict = codebook_dict
-
+    def parse_xyz(self, plydata: PlyData, device: torch.device) -> torch.Tensor:
+        elements = plydata['vertex']
+        kwargs = dict(dtype=torch.float32, device=device)
         xyz = torch.stack([
             torch.tensor(elements["x"].copy(), **kwargs),
             torch.tensor(elements["y"].copy(), **kwargs),
             torch.tensor(elements["z"].copy(), **kwargs),
         ], dim=1)
+        return xyz
+
+    def load_quantized(self, model: GaussianModel, ply_path: str) -> GaussianModel:
+        plydata = PlyData.read(ply_path)
+        ids_dict = self.parse_ids(plydata, model.max_sh_degree, model._xyz.device)
+        codebook_dict = self.parse_codebook(plydata, model.max_sh_degree, model._xyz.device)
+        xyz = self.parse_xyz(plydata, model._xyz.device)
+        self._codebook_dict = codebook_dict
         return self.dequantize(model, ids_dict, codebook_dict, xyz=xyz, replace=True)

{reduced_3dgs-1.10.3 → reduced_3dgs-1.10.5}/reduced_3dgs.egg-info/PKG-INFO

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

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