reduced-3dgs 1.8.17__tar.gz → 1.8.19__tar.gz

{reduced_3dgs-1.8.17 → reduced_3dgs-1.8.19}/PKG-INFO

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

@@ -1,13 +1,67 @@
+from typing import List
 from gaussian_splatting import GaussianModel, CameraTrainableGaussianModel, Camera
 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 OpacityResetTrainerWrapper, CameraTrainerWrapper
+from gaussian_splatting.trainer import OpacityResetTrainerWrapper, CameraTrainerWrapper, NoopDensifier, DepthTrainerWrapper
 from .shculling import VariableSHGaussianModel, SHCullingTrainerWrapper
 from .shculling import SHCullingTrainer
 # from .shculling import BaseSHCullingTrainer as SHCullingTrainer
-from .pruning import PruningTrainer, PrunerInDensifyTrainer
+from .pruning import PruningTrainerWrapper, PrunerInDensifyTrainerWrapper
 # from .pruning import BasePruningTrainer as PruningTrainer, BasePrunerInDensifyTrainer as PrunerInDensifyTrainer
+from .importance import ImportancePruner
+
+
+def BaseFullPruningTrainer(
+        model: GaussianModel,
+        scene_extent: float,
+        dataset: List[Camera],
+        *args,
+        importance_prune_from_iter=1000,
+        importance_prune_until_iter=15000,
+        importance_prune_interval: int = 100,
+        **kwargs):
+    return PruningTrainerWrapper(
+        lambda model, scene_extent, dataset: ImportancePruner(
+            NoopDensifier(model),
+            dataset,
+            importance_prune_from_iter=importance_prune_from_iter,
+            importance_prune_until_iter=importance_prune_until_iter,
+            importance_prune_interval=importance_prune_interval,
+        ),
+        model, scene_extent, dataset,
+        *args, **kwargs
+    )
+
+
+def BaseFullPrunerInDensifyTrainer(
+        model: GaussianModel,
+        scene_extent: float,
+        dataset: List[Camera],
+        *args,
+        importance_prune_from_iter=1000,
+        importance_prune_until_iter=15000,
+        importance_prune_interval: int = 100,
+        **kwargs):
+    return PrunerInDensifyTrainerWrapper(
+        lambda model, scene_extent, dataset: ImportancePruner(
+            NoopDensifier(model),
+            dataset,
+            importance_prune_from_iter=importance_prune_from_iter,
+            importance_prune_until_iter=importance_prune_until_iter,
+            importance_prune_interval=importance_prune_interval,
+        ),
+        model, scene_extent, dataset,
+        *args, **kwargs
+    )
+
+
+def DepthFullPruningTrainer(model: GaussianModel, scene_extent: float, dataset: TrainableCameraDataset, *args, **kwargs):
+    return DepthTrainerWrapper(BaseFullPruningTrainer, model, scene_extent, *args, dataset=dataset, **kwargs)
+
+
+def DepthFullPrunerInDensifyTrainer(model: GaussianModel, scene_extent: float, dataset: TrainableCameraDataset, *args, **kwargs):
+    return DepthTrainerWrapper(BaseFullPrunerInDensifyTrainer, model, scene_extent, *args, dataset=dataset, **kwargs)
 
 
 def OpacityResetPruningTrainer(
@@ -16,7 +70,7 @@ def OpacityResetPruningTrainer(
         dataset: CameraDataset,
         *args, **kwargs):
     return OpacityResetTrainerWrapper(
-        lambda model, scene_extent, *args, **kwargs: PruningTrainer(model, scene_extent, dataset, *args, **kwargs),
+        lambda model, scene_extent, *args, **kwargs: DepthFullPruningTrainer(model, scene_extent, dataset, *args, **kwargs),
         model, scene_extent,
         *args, **kwargs
     )
@@ -28,13 +82,17 @@ def OpacityResetPrunerInDensifyTrainer(
         dataset: CameraDataset,
         *args, **kwargs):
     return OpacityResetTrainerWrapper(
-        lambda model, scene_extent, *args, **kwargs: PrunerInDensifyTrainer(model, scene_extent, dataset, *args, **kwargs),
+        lambda model, scene_extent, *args, **kwargs: DepthFullPrunerInDensifyTrainer(model, scene_extent, dataset, *args, **kwargs),
         model, scene_extent,
         *args, **kwargs
     )
 
 
-def SHCullingDensifyTrainer(
+PruningTrainer = OpacityResetPruningTrainer
+PrunerInDensifyTrainer = OpacityResetPrunerInDensifyTrainer
+
+
+def SHCullingDensificationTrainer(
     model: VariableSHGaussianModel,
         scene_extent: float,
         dataset: CameraDataset,
@@ -46,7 +104,7 @@ def SHCullingDensifyTrainer(
     )
 
 
-def SHCullingPruneTrainer(
+def SHCullingPruningTrainer(
     model: VariableSHGaussianModel,
         scene_extent: float,
         dataset: CameraDataset,
@@ -58,7 +116,7 @@ def SHCullingPruneTrainer(
     )
 
 
-def SHCullingPruningDensifyTrainer(
+def SHCullingPrunerInDensifyTrainer(
     model: VariableSHGaussianModel,
         scene_extent: float,
         dataset: CameraDataset,
@@ -99,7 +157,7 @@ def CameraPruningTrainer(
     )
 
 
-def CameraOpacityResetPrunerInDensifyTrainer(
+def CameraPrunerInDensifyTrainer(
         model: CameraTrainableVariableSHGaussianModel,
         scene_extent: float,
         dataset: TrainableCameraDataset,
@@ -117,19 +175,19 @@ def CameraSHCullingDensifyTrainer(
         dataset: TrainableCameraDataset,
         *args, **kwargs):
     return CameraTrainerWrapper(
-        SHCullingDensifyTrainer,
+        SHCullingDensificationTrainer,
         model, scene_extent, dataset,
         *args, **kwargs
     )
 
 
-def CameraSHCullingPruneTrainer(
+def CameraSHCullingPruningTrainer(
         model: CameraTrainableVariableSHGaussianModel,
         scene_extent: float,
         dataset: TrainableCameraDataset,
         *args, **kwargs):
     return CameraTrainerWrapper(
-        SHCullingPruneTrainer,
+        SHCullingPruningTrainer,
         model, scene_extent, dataset,
         *args, **kwargs
     )
@@ -141,7 +199,7 @@ def CameraSHCullingPruningDensifyTrainer(
         dataset: TrainableCameraDataset,
         *args, **kwargs):
     return CameraTrainerWrapper(
-        SHCullingPruningDensifyTrainer,
+        SHCullingPrunerInDensifyTrainer,
         model, scene_extent, dataset,
         *args, **kwargs
     )

reduced_3dgs-1.8.19/reduced_3dgs/importance/__init__.py

@@ -0,0 +1,3 @@
+from .trainer import ImportancePruner, BaseImportancePruningTrainer
+from .combinations import BaseImportancePrunerInDensifyTrainer, DepthImportancePruningTrainer, DepthImportancePrunerInDensifyTrainer
+from .combinations import ImportancePruningTrainer, ImportancePrunerInDensifyTrainer

reduced_3dgs-1.8.19/reduced_3dgs/importance/combinations.py

@@ -0,0 +1,43 @@
+from typing import List
+from gaussian_splatting import Camera, GaussianModel
+from gaussian_splatting.dataset import TrainableCameraDataset
+from gaussian_splatting.trainer import DepthTrainerWrapper, NoopDensifier, DensificationTrainerWrapper
+from .trainer import ImportancePruner, BaseImportancePruningTrainer
+
+
+def BaseImportancePrunerInDensifyTrainer(
+        model: GaussianModel,
+        scene_extent: float,
+        dataset: List[Camera],
+        *args,
+        importance_prune_from_iter=1000,
+        importance_prune_until_iter=15000,
+        importance_prune_interval=100,
+        **kwargs):
+    return DensificationTrainerWrapper(
+        lambda model, scene_extent: ImportancePruner(
+            NoopDensifier(model),
+            dataset,
+            importance_prune_from_iter=importance_prune_from_iter,
+            importance_prune_until_iter=importance_prune_until_iter,
+            importance_prune_interval=importance_prune_interval,
+        ),
+        model,
+        scene_extent,
+        *args, **kwargs
+    )
+
+
+# Depth trainer
+
+
+def DepthImportancePruningTrainer(model: GaussianModel, scene_extent: float, dataset: TrainableCameraDataset, *args, **kwargs):
+    return DepthTrainerWrapper(BaseImportancePruningTrainer, model, scene_extent, *args, dataset=dataset, **kwargs)
+
+
+def DepthImportancePrunerInDensifyTrainer(model: GaussianModel, scene_extent: float, dataset: TrainableCameraDataset, *args, **kwargs):
+    return DepthTrainerWrapper(BaseImportancePrunerInDensifyTrainer, model, scene_extent, *args, dataset=dataset, **kwargs)
+
+
+ImportancePruningTrainer = DepthImportancePruningTrainer
+ImportancePrunerInDensifyTrainer = DepthImportancePrunerInDensifyTrainer

{reduced_3dgs-1.8.17/reduced_3dgs/pruning → reduced_3dgs-1.8.19/reduced_3dgs}/importance/trainer.py

@@ -1,8 +1,9 @@
 import math
+from typing import Callable, List
 import torch
 
 from gaussian_splatting import Camera, GaussianModel
-from gaussian_splatting.trainer import AbstractTrainer, TrainerWrapper, BaseTrainer, Trainer
+from gaussian_splatting.trainer import AbstractDensifier, DensifierWrapper, DensificationTrainer, NoopDensifier
 from gaussian_splatting.dataset import CameraDataset
 from .diff_gaussian_rasterization import GaussianRasterizationSettings, GaussianRasterizer
 
@@ -74,68 +75,56 @@ def count_render(self: GaussianModel, viewpoint_camera: Camera):
     }
 
 
-class ImportancePruner(TrainerWrapper):
+def prune_gaussians(model: GaussianModel, dataset: CameraDataset):
+    gaussian_count = torch.zeros(model.get_xyz.shape[0], device=model.get_xyz.device, dtype=torch.int)
+    opacity_important_score = torch.zeros(model.get_xyz.shape[0], device=model.get_xyz.device, dtype=torch.float)
+    T_alpha_important_score = torch.zeros(model.get_xyz.shape[0], device=model.get_xyz.device, dtype=torch.float)
+    for camera in dataset:
+        out = count_render(model, camera)
+        gaussian_count += out["gaussians_count"]
+        opacity_important_score += out["opacity_important_score"]
+        T_alpha_important_score += out["T_alpha_important_score"]
+    return None
+
+
+class ImportancePruner(DensifierWrapper):
     def __init__(
-            self, base_trainer: AbstractTrainer,
+            self, base_densifier: AbstractDensifier,
             dataset: CameraDataset,
-            importance_prune_at_steps=[15000],
+            importance_prune_from_iter=15000,
+            importance_prune_until_iter=20000,
+            importance_prune_interval: int = 1000,
     ):
-        super().__init__(base_trainer)
+        super().__init__(base_densifier)
         self.dataset = dataset
-        self.importance_prune_at_steps = importance_prune_at_steps
-
-    def optim_step(self):
-        ret = super().optim_step()
-        if self.curr_step in self.importance_prune_at_steps:
-            gaussian_count = torch.zeros(self.model.get_xyz.shape[0], device=self.model.get_xyz.device, dtype=torch.int)
-            opacity_important_score = torch.zeros(self.model.get_xyz.shape[0], device=self.model.get_xyz.device, dtype=torch.float)
-            T_alpha_important_score = torch.zeros(self.model.get_xyz.shape[0], device=self.model.get_xyz.device, dtype=torch.float)
-            for camera in self.dataset:
-                out = count_render(self.model, camera)
-                gaussian_count += out["gaussians_count"]
-                opacity_important_score += out["opacity_important_score"]
-                T_alpha_important_score += out["T_alpha_important_score"]
-            pass
+        self.importance_prune_from_iter = importance_prune_from_iter
+        self.importance_prune_until_iter = importance_prune_until_iter
+        self.importance_prune_interval = importance_prune_interval
+
+    def densify_and_prune(self, loss, out, camera, step: int):
+        ret = super().densify_and_prune(loss, out, camera, step)
+        if self.importance_prune_from_iter <= step <= self.importance_prune_until_iter and step % self.importance_prune_interval == 0:
+            remove_mask = prune_gaussians(self.model, self.dataset)
+            ret = ret._replace(remove_mask=remove_mask if ret.remove_mask is None else torch.logical_or(ret.remove_mask, remove_mask))
         return ret
 
 
-def ImportancePruningTrainerWrapper(
-    base_trainer_constructor,
-        model: GaussianModel,
-        scene_extent: float,
-        dataset: CameraDataset,
-        importance_prune_at_steps=[15000],
-        *args, **kwargs):
-    return ImportancePruner(
-        base_trainer_constructor(model, scene_extent, dataset, *args, **kwargs),
-        dataset,
-        importance_prune_at_steps=importance_prune_at_steps,
-    )
-
-
 def BaseImportancePruningTrainer(
-    model: GaussianModel,
-        scene_extent: float,
-        dataset: CameraDataset,
-        importance_prune_at_steps=[15000],
-        *args, **kwargs):
-    return ImportancePruningTrainerWrapper(
-        lambda model, scene_extent, dataset, *args, **kwargs: BaseTrainer(model, scene_extent, *args, **kwargs),
-        model, scene_extent, dataset,
-        importance_prune_at_steps=importance_prune_at_steps,
-        *args, **kwargs,
-    )
-
-
-def ImportancePruningTrainer(
-    model: GaussianModel,
+        model: GaussianModel,
         scene_extent: float,
-        dataset: CameraDataset,
-        importance_prune_at_steps=[15000],
-        *args, **kwargs):
-    return ImportancePruningTrainerWrapper(
-        lambda model, scene_extent, dataset, *args, **kwargs: Trainer(model, scene_extent, *args, **kwargs),
-        model, scene_extent, dataset,
-        importance_prune_at_steps=importance_prune_at_steps,
-        *args, **kwargs,
+        dataset: List[Camera],
+        *args,
+        importance_prune_from_iter=1000,
+        importance_prune_until_iter=15000,
+        importance_prune_interval: int = 100,
+        **kwargs):
+    return DensificationTrainer(
+        model, scene_extent,
+        ImportancePruner(
+            NoopDensifier(model),
+            dataset,
+            importance_prune_from_iter=importance_prune_from_iter,
+            importance_prune_until_iter=importance_prune_until_iter,
+            importance_prune_interval=importance_prune_interval,
+        ), *args, **kwargs
     )

reduced_3dgs-1.8.19/reduced_3dgs/pruning/__init__.py

@@ -0,0 +1,2 @@
+from .trainer import BasePruner, BasePruningTrainer, PruningTrainerWrapper
+from .combinations import BasePrunerInDensifyTrainer, PruningTrainer, PrunerInDensifyTrainer, PrunerInDensifyTrainerWrapper

{reduced_3dgs-1.8.17 → reduced_3dgs-1.8.19}/reduced_3dgs/pruning/combinations.py

@@ -1,15 +1,17 @@
 
-from typing import List
+from typing import Callable, List
 from gaussian_splatting import Camera, GaussianModel
 from gaussian_splatting.dataset import TrainableCameraDataset
-from gaussian_splatting.trainer import DepthTrainerWrapper, NoopDensifier, DensificationTrainerWrapper
+from gaussian_splatting.trainer import AbstractDensifier, DepthTrainerWrapper, NoopDensifier, SplitCloneDensifierTrainerWrapper
 from .trainer import BasePruner, BasePruningTrainer
 
 
-def BasePrunerInDensifyTrainer(
+def PrunerInDensifyTrainerWrapper(
+        noargs_base_densifier_constructor: Callable[[GaussianModel, float, List[Camera]], AbstractDensifier],
         model: GaussianModel,
         scene_extent: float,
         dataset: List[Camera],
+        *args,
         prune_from_iter=1000,
         prune_until_iter=15000,
         prune_interval: int = 100,
@@ -17,10 +19,10 @@ def BasePrunerInDensifyTrainer(
         lambda_mercy=1.,
         mercy_minimum=3,
         mercy_type='redundancy_opacity',
-        *args, **kwargs):
-    return DensificationTrainerWrapper(
+        **kwargs):
+    return SplitCloneDensifierTrainerWrapper(
         lambda model, scene_extent: BasePruner(
-            NoopDensifier(model),
+            noargs_base_densifier_constructor(model, scene_extent, dataset),
             dataset,
             prune_from_iter=prune_from_iter,
             prune_until_iter=prune_until_iter,
@@ -36,6 +38,18 @@ def BasePrunerInDensifyTrainer(
     )
 
 
+def BasePrunerInDensifyTrainer(
+        model: GaussianModel,
+        scene_extent: float,
+        dataset: List[Camera],
+        *args, **kwargs):
+    return PrunerInDensifyTrainerWrapper(
+        lambda model, scene_extent, dataset: NoopDensifier(model),
+        model, scene_extent, dataset,
+        *args, **kwargs
+    )
+
+
 # Depth trainer
 
 

{reduced_3dgs-1.8.17 → reduced_3dgs-1.8.19}/reduced_3dgs/pruning/trainer.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import Callable, List
 import torch
 from gaussian_splatting import GaussianModel, Camera
 from gaussian_splatting.trainer import AbstractDensifier, DensifierWrapper, DensificationTrainer, NoopDensifier
@@ -108,10 +108,12 @@ class BasePruner(DensifierWrapper):
         return ret
 
 
-def BasePruningTrainer(
+def PruningTrainerWrapper(
+        noargs_base_densifier_constructor: Callable[[GaussianModel, float, List[Camera]], AbstractDensifier],
         model: GaussianModel,
         scene_extent: float,
         dataset: List[Camera],
+        *args,
         prune_from_iter=1000,
         prune_until_iter=15000,
         prune_interval: int = 100,
@@ -119,11 +121,11 @@ def BasePruningTrainer(
         lambda_mercy=1.,
         mercy_minimum=3,
         mercy_type='redundancy_opacity',
-        *args, **kwargs):
+        **kwargs):
     return DensificationTrainer(
         model, scene_extent,
         BasePruner(
-            NoopDensifier(model),
+            noargs_base_densifier_constructor(model, scene_extent, dataset),
             dataset,
             prune_from_iter=prune_from_iter,
             prune_until_iter=prune_until_iter,
@@ -134,3 +136,15 @@ def BasePruningTrainer(
             mercy_type=mercy_type,
         ), *args, **kwargs
     )
+
+
+def BasePruningTrainer(
+        model: GaussianModel,
+        scene_extent: float,
+        dataset: List[Camera],
+        *args, **kwargs):
+    return DensificationTrainer(
+        lambda model, scene_extent, dataset: NoopDensifier(model),
+        model, scene_extent, dataset,
+        *args, **kwargs
+    )

{reduced_3dgs-1.8.17 → reduced_3dgs-1.8.19}/reduced_3dgs/train.py

@@ -14,30 +14,27 @@ from gaussian_splatting.trainer.extensions import ScaleRegularizeTrainerWrapper
 from reduced_3dgs.quantization import AbstractQuantizer, VectorQuantizeTrainerWrapper
 from reduced_3dgs.shculling import VariableSHGaussianModel, SHCullingTrainer
 from reduced_3dgs.pruning import PruningTrainer
-from reduced_3dgs.pruning.importance import ImportancePruningTrainerWrapper
-from reduced_3dgs.combinations import OpacityResetPrunerInDensifyTrainer, SHCullingDensifyTrainer, SHCullingPruneTrainer, SHCullingPruningDensifyTrainer
+from reduced_3dgs.combinations import PrunerInDensifyTrainer, SHCullingDensificationTrainer, SHCullingPruningTrainer, SHCullingPrunerInDensifyTrainer
 from reduced_3dgs.combinations import CameraTrainableVariableSHGaussianModel, CameraSHCullingTrainer, CameraPruningTrainer
-from reduced_3dgs.combinations import CameraOpacityResetPrunerInDensifyTrainer, CameraSHCullingDensifyTrainer, CameraSHCullingPruneTrainer, CameraSHCullingPruningDensifyTrainer
+from reduced_3dgs.combinations import CameraPrunerInDensifyTrainer, CameraSHCullingDensifyTrainer, CameraSHCullingPruningTrainer, CameraSHCullingPruningDensifyTrainer
 
 
 basemodes = {
     "shculling": SHCullingTrainer,
     "pruning": PruningTrainer,
-    "densify-pruning": OpacityResetPrunerInDensifyTrainer,
-    "densify-shculling": SHCullingDensifyTrainer,
-    "prune-shculling": SHCullingPruneTrainer,
-    "densify-prune-shculling": SHCullingPruningDensifyTrainer,
+    "densify-pruning": PrunerInDensifyTrainer,
+    "densify-shculling": SHCullingDensificationTrainer,
+    "prune-shculling": SHCullingPruningTrainer,
+    "densify-prune-shculling": SHCullingPrunerInDensifyTrainer,
 }
-basemodes = {k: lambda *args, **kwargs: ImportancePruningTrainerWrapper(v, *args, **kwargs) for k, v in basemodes.items()}
 cameramodes = {
     "camera-shculling": CameraSHCullingTrainer,
     "camera-pruning": CameraPruningTrainer,
-    "camera-densify-pruning": CameraOpacityResetPrunerInDensifyTrainer,
+    "camera-densify-pruning": CameraPrunerInDensifyTrainer,
     "camera-densify-shculling": CameraSHCullingDensifyTrainer,
-    "camera-prune-shculling": CameraSHCullingPruneTrainer,
+    "camera-prune-shculling": CameraSHCullingPruningTrainer,
     "camera-densify-prune-shculling": CameraSHCullingPruningDensifyTrainer,
 }
-cameramodes = {k: lambda *args, **kwargs: ImportancePruningTrainerWrapper(v, *args, **kwargs) for k, v in cameramodes.items()}
 
 
 def prepare_quantizer(

{reduced_3dgs-1.8.17 → reduced_3dgs-1.8.19}/reduced_3dgs.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reduced_3dgs
-Version: 1.8.17
+Version: 1.8.19
 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.17 → reduced_3dgs-1.8.19}/reduced_3dgs.egg-info/SOURCES.txt

@@ -10,11 +10,12 @@ reduced_3dgs.egg-info/SOURCES.txt
 reduced_3dgs.egg-info/dependency_links.txt
 reduced_3dgs.egg-info/requires.txt
 reduced_3dgs.egg-info/top_level.txt
+reduced_3dgs/importance/__init__.py
+reduced_3dgs/importance/combinations.py
+reduced_3dgs/importance/trainer.py
 reduced_3dgs/pruning/__init__.py
 reduced_3dgs/pruning/combinations.py
 reduced_3dgs/pruning/trainer.py
-reduced_3dgs/pruning/importance/__init__.py
-reduced_3dgs/pruning/importance/trainer.py
 reduced_3dgs/quantization/__init__.py
 reduced_3dgs/quantization/abc.py
 reduced_3dgs/quantization/exclude_zeros.py

{reduced_3dgs-1.8.17 → reduced_3dgs-1.8.19}/setup.py

@@ -48,7 +48,7 @@ importance_sources = [
     "rasterize_points.cu",
     "ext.cpp"]
 importance_packages = {
-    'reduced_3dgs.pruning.importance.diff_gaussian_rasterization': 'submodules/gaussian-importance/diff_gaussian_rasterization',
+    'reduced_3dgs.importance.diff_gaussian_rasterization': 'submodules/gaussian-importance/diff_gaussian_rasterization',
 }
 
 cxx_compiler_flags = []
@@ -60,7 +60,7 @@ if os.name == 'nt':
 
 setup(
     name="reduced_3dgs",
-    version='1.8.17',
+    version='1.8.19',
     author='yindaheng98',
     author_email='yindaheng98@gmail.com',
     url='https://github.com/yindaheng98/reduced-3dgs',
@@ -83,7 +83,7 @@ setup(
             extra_compile_args={"nvcc": nvcc_compiler_flags + ["-I" + os.path.join(os.path.abspath(rasterizor_root), "third_party/glm/")]}
         ),
         CUDAExtension(
-            name="reduced_3dgs.pruning.importance.diff_gaussian_rasterization._C",
+            name="reduced_3dgs.importance.diff_gaussian_rasterization._C",
             sources=[os.path.join(importance_root, source) for source in importance_sources],
             extra_compile_args={"nvcc": nvcc_compiler_flags + ["-I" + os.path.join(os.path.abspath(importance_root), "third_party/glm/")]}
         ),

reduced_3dgs-1.8.17/reduced_3dgs/pruning/__init__.py

@@ -1,2 +0,0 @@
-from .trainer import BasePruner, BasePruningTrainer
-from .combinations import BasePrunerInDensifyTrainer, PruningTrainer, PrunerInDensifyTrainer

reduced_3dgs-1.8.17/reduced_3dgs/pruning/importance/__init__.py

@@ -1 +0,0 @@
-from .trainer import ImportancePruningTrainerWrapper, BaseImportancePruningTrainer, ImportancePruningTrainer