reduced-3dgs 1.10.9__cp311-cp311-win_amd64.whl → 1.10.10__cp311-cp311-win_amd64.whl

reduced_3dgs/train.py

@@ -27,7 +27,7 @@ def prepare_training(
 
 def training(dataset: CameraDataset, gaussians: GaussianModel, trainer: AbstractTrainer, quantizer: AbstractQuantizer, destination: str, iteration: int, save_iterations: List[int], device: str, empty_cache_every_step=False):
     shutil.rmtree(os.path.join(destination, "point_cloud"), ignore_errors=True)  # remove the previous point cloud
-    pbar = tqdm(range(1, iteration+1))
+    pbar = tqdm(range(1, iteration+1), dynamic_ncols=True, desc="Training")
     epoch = list(range(len(dataset)))
     epoch_psnr = torch.empty(3, 0, device=device)
     ema_loss_for_log = 0.0
@@ -43,8 +43,13 @@ def training(dataset: CameraDataset, gaussians: GaussianModel, trainer: Abstract
         if empty_cache_every_step:
             torch.cuda.empty_cache()
         with torch.no_grad():
+            ground_truth_image = dataset[idx].ground_truth_image
+            rendered_image = out["render"]
+            if dataset[idx].ground_truth_image_mask is not None:
+                ground_truth_image *= dataset[idx].ground_truth_image_mask
+                rendered_image *= dataset[idx].ground_truth_image_mask
             ema_loss_for_log = 0.4 * loss.item() + 0.6 * ema_loss_for_log
-            epoch_psnr = torch.concat([epoch_psnr, psnr(out["render"], dataset[idx].ground_truth_image)], dim=1)
+            epoch_psnr = torch.concat([epoch_psnr, psnr(rendered_image, ground_truth_image)], dim=1)
             if step % 10 == 0:
                 pbar.set_postfix({'epoch': step // len(dataset), 'loss': ema_loss_for_log, 'psnr': avg_psnr_for_log, 'n': gaussians._xyz.shape[0]})
         if step in save_iterations:

{reduced_3dgs-1.10.9.dist-info → reduced_3dgs-1.10.10.dist-info}/METADATA

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

@@ -2,13 +2,13 @@ reduced_3dgs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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=yyo1gmdwM5EW3_asAD1uhioJICCPKrFdIucIB-K6iSc,5856
-reduced_3dgs/diff_gaussian_rasterization/_C.cp311-win_amd64.pyd,sha256=pU3iAWEIGdMLLzAKITTYUNOJYptPP2v6OsxFClXo9-k,1623552
+reduced_3dgs/train.py,sha256=0An36u1xTXAQYn2_y85C84YdfJHkno387O5nlxaKYZI,6205
+reduced_3dgs/diff_gaussian_rasterization/_C.cp311-win_amd64.pyd,sha256=qUOQ0KkQ2bgIpbNPBO6Hh--S6hzYVVeJi9ZJHJs4zsQ,1623552
 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=F9IYDZ6iquZw5Djn2An88dlp2MUGieAyuZyVaDZdSts,2962
 reduced_3dgs/importance/trainer.py,sha256=Sj4ORvoYtFT7z3hifzFZDfhFyqumHraXyk3vMVtk0AU,12661
-reduced_3dgs/importance/diff_gaussian_rasterization/_C.cp311-win_amd64.pyd,sha256=RPSU4Z1HToiQRDyBsQE2KDqdkHyMvxIeX0rJ-L27iP0,1309696
+reduced_3dgs/importance/diff_gaussian_rasterization/_C.cp311-win_amd64.pyd,sha256=-D0QjVI_2gbrCW7sBZ-tPIyAdIvdFYbIL6A5F6_7U8c,1309696
 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=xO39d08DEsDnKI5np_1xQ4H29O486Q_cThLfWhlH810,2385
@@ -21,9 +21,9 @@ reduced_3dgs/quantization/wrapper.py,sha256=cyXqfJgo9b3fS7DYXxOk5LmQudvrEhweOebF
 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.cp311-win_amd64.pyd,sha256=ecvkYxpxmkZXcAbYSRXhNxQOghv4T_irRavqSfufx74,1256448
-reduced_3dgs-1.10.9.dist-info/licenses/LICENSE.md,sha256=LQ4_LAqlncGkg_mQy5ykMAFtQDSPB0eKmIEtBut0yjw,4916
-reduced_3dgs-1.10.9.dist-info/METADATA,sha256=Wk8Ps_Lgg5dut6dcmaGlw4jht2YZnfhXBk-BoqW_8M0,12403
-reduced_3dgs-1.10.9.dist-info/WHEEL,sha256=JLOMsP7F5qtkAkINx5UnzbFguf8CqZeraV8o04b0I8I,101
-reduced_3dgs-1.10.9.dist-info/top_level.txt,sha256=PpU5aT3-baSCdqCtTaZknoB32H93UeKCkYDkRCCZMEI,13
-reduced_3dgs-1.10.9.dist-info/RECORD,,
+reduced_3dgs/simple_knn/_C.cp311-win_amd64.pyd,sha256=BAMJEpLSpzeC3xcI8nbXRjaU7UJPwb9I3BfzyRdMlWs,1256448
+reduced_3dgs-1.10.10.dist-info/licenses/LICENSE.md,sha256=LQ4_LAqlncGkg_mQy5ykMAFtQDSPB0eKmIEtBut0yjw,4916
+reduced_3dgs-1.10.10.dist-info/METADATA,sha256=c75WD2Y-fK04loLm_fTuX3O1uIjJs5Xia0byAebU_co,12404
+reduced_3dgs-1.10.10.dist-info/WHEEL,sha256=JLOMsP7F5qtkAkINx5UnzbFguf8CqZeraV8o04b0I8I,101
+reduced_3dgs-1.10.10.dist-info/top_level.txt,sha256=PpU5aT3-baSCdqCtTaZknoB32H93UeKCkYDkRCCZMEI,13
+reduced_3dgs-1.10.10.dist-info/RECORD,,