reduced-3dgs 1.10.4__cp310-cp310-win_amd64.whl → 1.10.6__cp310-cp310-win_amd64.whl

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/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/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/quantization/quantizer.py

@@ -4,7 +4,15 @@ from typing import Dict, Tuple
 import torch
 import torch.nn as nn
 import numpy as np
-from sklearn.cluster import MiniBatchKMeans as KMeans
+try:
+    from cuml.cluster import KMeans
+    kmeans_init = 'k-means||'
+except ImportError:
+    print("Cuml not found, using sklearn's MiniBatchKMeans for quantization.")
+    from sklearn.cluster import MiniBatchKMeans
+    from functools import partial
+    KMeans = partial(MiniBatchKMeans, batch_size=256 * os.cpu_count())
+    kmeans_init = 'k-means++'
 from gaussian_splatting import GaussianModel
 from plyfile import PlyData, PlyElement
 import numpy as np
@@ -65,9 +73,8 @@ class VectorQuantizer(AbstractQuantizer):
     def generate_codebook(self, values: torch.Tensor, num_clusters, init_codebook=None):
         kmeans = KMeans(
             n_clusters=num_clusters, tol=self.tol, max_iter=self.max_iter,
-            init='k-means++' if init_codebook is None else init_codebook.cpu().numpy(),
-            random_state=0, n_init="auto", verbose=0,
-            batch_size=256 * os.cpu_count()
+            init=kmeans_init if init_codebook is None else init_codebook.cpu().numpy(),
+            random_state=0, n_init="auto", verbose=1,
         )
         ids = torch.tensor(kmeans.fit_predict(values.cpu().numpy()), device=values.device)
         centers = torch.tensor(kmeans.cluster_centers_, dtype=values.dtype, device=values.device)

{reduced_3dgs-1.10.4.dist-info → reduced_3dgs-1.10.6.dist-info}/METADATA

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

@@ -1,29 +1,29 @@
 reduced_3dgs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-reduced_3dgs/combinations.py,sha256=k4ErxpAscCqJMtVXZ29KGQjw2VoZMV0W3V4u3nj0e-Y,9401
+reduced_3dgs/combinations.py,sha256=dRrtTIQLggUP22Olp0DVSlD0C8YDgnrpTlrCcGZyWM0,9441
 reduced_3dgs/prepare.py,sha256=MFUUckRHKfgcva4ZOBxfPFyE95N-OlCQLplpmEPuzOk,4440
 reduced_3dgs/quantize.py,sha256=BVqBb2tQgiP3hap5-OByD8VELtJJGfEeFzaVFyzCJZU,2572
 reduced_3dgs/train.py,sha256=yRQPQv-hTOBQN-tqGHYs2aIZ0AbWq158CXpthYN2cfw,5666
-reduced_3dgs/diff_gaussian_rasterization/_C.cp310-win_amd64.pyd,sha256=LHHSwD5jDEkO_0PH91Xzbsykr72WQQ7Wq_q3dvQ7Fss,1618432
+reduced_3dgs/diff_gaussian_rasterization/_C.cp310-win_amd64.pyd,sha256=QnHil663TLhHihMOWPGK_tpogJ0wRLO66wzGfXne2QQ,1610752
 reduced_3dgs/diff_gaussian_rasterization/__init__.py,sha256=oV6JjTc-50MscX4XHeIWSgLr3l8Y25knBIs-0gRbJr4,7932
 reduced_3dgs/importance/__init__.py,sha256=neJsbY5cLikEGBQGdR4MjwCQ5VWVikT1357DwL0EtWU,289
-reduced_3dgs/importance/combinations.py,sha256=eAdykeTdvRGCHxskjILQnZVaqQVvwC-0wMxdgYMeeDs,2922
+reduced_3dgs/importance/combinations.py,sha256=F9IYDZ6iquZw5Djn2An88dlp2MUGieAyuZyVaDZdSts,2962
 reduced_3dgs/importance/trainer.py,sha256=Sj4ORvoYtFT7z3hifzFZDfhFyqumHraXyk3vMVtk0AU,12661
-reduced_3dgs/importance/diff_gaussian_rasterization/_C.cp310-win_amd64.pyd,sha256=FJlv_ytBpqI0i2lXSc2iWPPMM0lC8j8HLfsP_pTPQOU,1308160
+reduced_3dgs/importance/diff_gaussian_rasterization/_C.cp310-win_amd64.pyd,sha256=h0_JG45D8yuy8-_AgY8-V7DE4_50CuWHOq5gPdEkNQ8,1302016
 reduced_3dgs/importance/diff_gaussian_rasterization/__init__.py,sha256=Tix8auyXBb_QFQtXrV3sLE9kdnl5zgHH0BbqcFzDp84,12850
 reduced_3dgs/pruning/__init__.py,sha256=E_YxJ9cDV_B6EJbYUBEcuRYMIht_C72rI1VJUXFCLpM,201
-reduced_3dgs/pruning/combinations.py,sha256=QhXt2C7pTXhwzp9hPL9dVdiQzz0cUQpm5qljqytPEsM,2345
+reduced_3dgs/pruning/combinations.py,sha256=xO39d08DEsDnKI5np_1xQ4H29O486Q_cThLfWhlH810,2385
 reduced_3dgs/pruning/trainer.py,sha256=JJml-uYfDfUpbsjRNZbIvnUYYslVgFXkhejbkYSo0s4,6542
 reduced_3dgs/quantization/__init__.py,sha256=1z1xMn3yj9u7cR9JizGrI3WSyIES_Tqq6oDquvglSeo,225
 reduced_3dgs/quantization/abc.py,sha256=rsi8HFRwQCltWTYiJ3BpygtQDT7hK6J01jKMOboOY8w,1910
 reduced_3dgs/quantization/exclude_zeros.py,sha256=fKSgjHous4OpdI6mQi9z23if9jnbB79w2jChpxkCJWw,2381
-reduced_3dgs/quantization/quantizer.py,sha256=lsaJ4yjxquC9XCDZPTGIKe8EgHAEwmVw9rt73oezjmc,17686
+reduced_3dgs/quantization/quantizer.py,sha256=2myyBsYPGbzjZhjpKnYMYcvk6INqeds7wRdBRbIPki4,17948
 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.cp310-win_amd64.pyd,sha256=cknfimJJGEUL7vCBdTYuTNhHZdhGTeKJr8KVQJdRD0M,1255424
-reduced_3dgs-1.10.4.dist-info/licenses/LICENSE.md,sha256=LQ4_LAqlncGkg_mQy5ykMAFtQDSPB0eKmIEtBut0yjw,4916
-reduced_3dgs-1.10.4.dist-info/METADATA,sha256=TfBJNYmRSs23HhMOj15mrTFzryN015ZgGTeqUMKza7w,13015
-reduced_3dgs-1.10.4.dist-info/WHEEL,sha256=KUuBC6lxAbHCKilKua8R9W_TM71_-9Sg5uEP3uDWcoU,101
-reduced_3dgs-1.10.4.dist-info/top_level.txt,sha256=PpU5aT3-baSCdqCtTaZknoB32H93UeKCkYDkRCCZMEI,13
-reduced_3dgs-1.10.4.dist-info/RECORD,,
+reduced_3dgs/simple_knn/_C.cp310-win_amd64.pyd,sha256=w7v44ndobRqldV1LuioMCAI3KgEZT0F0En7IKQGdj-s,1248768
+reduced_3dgs-1.10.6.dist-info/licenses/LICENSE.md,sha256=LQ4_LAqlncGkg_mQy5ykMAFtQDSPB0eKmIEtBut0yjw,4916
+reduced_3dgs-1.10.6.dist-info/METADATA,sha256=Sn6qPfw9clEy74_ULrx2uR-C6VmJvE4eSlwUHmv53go,13015
+reduced_3dgs-1.10.6.dist-info/WHEEL,sha256=KUuBC6lxAbHCKilKua8R9W_TM71_-9Sg5uEP3uDWcoU,101
+reduced_3dgs-1.10.6.dist-info/top_level.txt,sha256=PpU5aT3-baSCdqCtTaZknoB32H93UeKCkYDkRCCZMEI,13
+reduced_3dgs-1.10.6.dist-info/RECORD,,