reduced-3dgs 1.9.5__cp312-cp312-win_amd64.whl → 1.10.0__cp312-cp312-win_amd64.whl

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

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

@@ -2,12 +2,12 @@ reduced_3dgs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reduced_3dgs/combinations.py,sha256=k4ErxpAscCqJMtVXZ29KGQjw2VoZMV0W3V4u3nj0e-Y,9401
 reduced_3dgs/quantize.py,sha256=Y44qHyFdOIqke7NoeqXmyKloS43j-al74ZiNsuZZHbM,2527
 reduced_3dgs/train.py,sha256=jXHdXk05o_ebHjx_VBzcY6fRNn9EdKve6Tf5YC5an0o,9803
-reduced_3dgs/diff_gaussian_rasterization/_C.cp312-win_amd64.pyd,sha256=D7ZCIZZAsv4ybmBpLOqxk_ciPlpO5TrxpHMejYYmKmc,1640448
+reduced_3dgs/diff_gaussian_rasterization/_C.cp312-win_amd64.pyd,sha256=vxO1tZDmYDx1RCVZg88T1wJ3zFVkVrY1tun0y1hW_Lk,1640448
 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/trainer.py,sha256=Sj4ORvoYtFT7z3hifzFZDfhFyqumHraXyk3vMVtk0AU,12661
-reduced_3dgs/importance/diff_gaussian_rasterization/_C.cp312-win_amd64.pyd,sha256=CKNzRxNhntqRRGnvMYtFjxzFqVIGcPZ92XMW59o0uBc,1320448
+reduced_3dgs/importance/diff_gaussian_rasterization/_C.cp312-win_amd64.pyd,sha256=oMwEz5ip6T2ch_qEOcCH8r2vm2MY7wvKaURjsgdS1D4,1320448
 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
@@ -15,14 +15,14 @@ reduced_3dgs/pruning/trainer.py,sha256=JJml-uYfDfUpbsjRNZbIvnUYYslVgFXkhejbkYSo0
 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=-B07X1VrV7E1fT_NXXlZ0FLAb-c_jBUlZ8274KwHyVw,16614
+reduced_3dgs/quantization/quantizer.py,sha256=VL96ZTY7G8jUt-fLBJ3gw1MQV46NkHOBJyfUPsmDQxE,16712
 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.cp312-win_amd64.pyd,sha256=nW0YFZFsAzZu92nWX3SLB7II3V0ivfkT2O9M0L9PmrM,1267712
-reduced_3dgs-1.9.5.dist-info/licenses/LICENSE.md,sha256=LQ4_LAqlncGkg_mQy5ykMAFtQDSPB0eKmIEtBut0yjw,4916
-reduced_3dgs-1.9.5.dist-info/METADATA,sha256=VkgFVSczzZCk6RpIMZ_fEA18ZVyf6tSc7cEJhLUtixY,13014
-reduced_3dgs-1.9.5.dist-info/WHEEL,sha256=t5QiC5vd2hOdU0bFY5fyqZIc3wMdCuUPTCilUZToJT4,101
-reduced_3dgs-1.9.5.dist-info/top_level.txt,sha256=PpU5aT3-baSCdqCtTaZknoB32H93UeKCkYDkRCCZMEI,13
-reduced_3dgs-1.9.5.dist-info/RECORD,,
+reduced_3dgs/simple_knn/_C.cp312-win_amd64.pyd,sha256=wGxp2MJfzY2BUNrkVqJwdjQIvyom6peboTpMWw38vAg,1267712
+reduced_3dgs-1.10.0.dist-info/licenses/LICENSE.md,sha256=LQ4_LAqlncGkg_mQy5ykMAFtQDSPB0eKmIEtBut0yjw,4916
+reduced_3dgs-1.10.0.dist-info/METADATA,sha256=5FMe1oopipogQu_qsW7-GnO6KVVXrRXdsCKcVtbqIF8,13015
+reduced_3dgs-1.10.0.dist-info/WHEEL,sha256=8UP9x9puWI0P1V_d7K2oMTBqfeLNm21CTzZ_Ptr0NXU,101
+reduced_3dgs-1.10.0.dist-info/top_level.txt,sha256=PpU5aT3-baSCdqCtTaZknoB32H93UeKCkYDkRCCZMEI,13
+reduced_3dgs-1.10.0.dist-info/RECORD,,

{reduced_3dgs-1.9.5.dist-info → reduced_3dgs-1.10.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.0.1)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: false
 Tag: cp312-cp312-win_amd64