gaussian-splatting 1.17.8__cp311-cp311-win_amd64.whl → 1.17.9__cp311-cp311-win_amd64.whl

gaussian_splatting/trainer/base.py

@@ -22,13 +22,11 @@ class BaseTrainer(AbstractTrainer):
             opacity_lr=0.025,
             scaling_lr=0.005,
             rotation_lr=0.001,
-            ignore_out_of_image_mask_loss=False,  # whether to ignore loss for out-of-mask pixels, if True, these pixels will be ignored in loss computation
-            random_out_of_image_mask_color=False,  # if ignore_out_of_mask_loss is False, whether use random color or use camera.bg_color for out-of-mask pixels
+            mask_mode="ignore",  # "ignore", "random", "bg_color"
     ):
         super().__init__()
         self.lambda_dssim = lambda_dssim
-        self.ignore_out_of_image_mask_loss = ignore_out_of_image_mask_loss
-        self.random_out_of_image_mask_color = random_out_of_image_mask_color
+        self.mask_mode = mask_mode
         params = [
             {'params': [model._xyz], 'lr': position_lr_init * scene_extent, "name": "xyz"},
             {'params': [model._features_dc], 'lr': feature_lr, "name": "f_dc"},
@@ -76,13 +74,16 @@ class BaseTrainer(AbstractTrainer):
         gt = camera.ground_truth_image
         mask = camera.ground_truth_image_mask
         if mask is not None:
-            if self.ignore_out_of_image_mask_loss:
-                render = render * mask.unsqueeze(0)
-                gt = gt * mask.unsqueeze(0)
-            elif self.random_out_of_image_mask_color:
-                gt = gt * mask.unsqueeze(0) + (1 - mask.unsqueeze(0)) * torch.rand_like(gt)
-            else:
-                gt = gt * mask.unsqueeze(0) + (1 - mask.unsqueeze(0)) * camera.bg_color.unsqueeze(-1).unsqueeze(-1)
+            match self.mask_mode:
+                case "ignore":
+                    render = render * mask.unsqueeze(0)
+                    gt = gt * mask.unsqueeze(0)
+                case "random":
+                    gt = gt * mask.unsqueeze(0) + (1 - mask.unsqueeze(0)) * torch.rand_like(gt)
+                case "bg_color":
+                    gt = gt * mask.unsqueeze(0) + (1 - mask.unsqueeze(0)) * camera.bg_color.unsqueeze(-1).unsqueeze(-1)
+                case _:
+                    raise ValueError(f"Unknown mask policy: {self.mask_mode}")
         Ll1 = l1_loss(render, gt)
         ssim_value = ssim(render, gt)
         loss = (1.0 - self.lambda_dssim) * Ll1 + self.lambda_dssim * (1.0 - ssim_value)

gaussian_splatting/trainer/depth.py

@@ -75,10 +75,6 @@ class DepthTrainer(TrainerWrapper):
         invdepth = out["depth"].squeeze(0)
         invdepth_gt = camera.ground_truth_depth
         mask = camera.ground_truth_depth_mask
-        if mask is None:
-            mask = camera.ground_truth_image_mask
-        elif camera.ground_truth_image_mask is not None:
-            mask = mask * camera.ground_truth_image_mask
         assert invdepth.shape == invdepth_gt.shape, f"invdepth shape {invdepth.shape} does not match gt depth shape {invdepth_gt.shape}"
         if self.depth_resize is not None:
             height, width = invdepth.shape[-2:]

{gaussian_splatting-1.17.8.dist-info → gaussian_splatting-1.17.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gaussian_splatting
-Version: 1.17.8
+Version: 1.17.9
 Summary: Refactored python training and inference code for 3D Gaussian Splatting
 Home-page: https://github.com/yindaheng98/gaussian-splatting
 Author: yindaheng98

{gaussian_splatting-1.17.8.dist-info → gaussian_splatting-1.17.9.dist-info}/RECORD

@@ -12,15 +12,15 @@ gaussian_splatting/dataset/colmap/__init__.py,sha256=YEYT2k2WJSqrkkZq4KAJYS9UMgq
 gaussian_splatting/dataset/colmap/dataset.py,sha256=Lq2b3hMdtOmdqPjvEjR6CLukAR7dZBEKMz8yzDD2Bgo,4519
 gaussian_splatting/dataset/colmap/params_init.py,sha256=6_6gZ0Wl4aZrps2PJ_U234sxW5D-vOTfwioVa1FWC-E,1802
 gaussian_splatting/dataset/colmap/read_write_model.py,sha256=TenI7ai5UV7Ksg2vAXvJWnYFwOOo1tlS_633RfCLuQU,23137
-gaussian_splatting/diff_gaussian_rasterization/_C.cp311-win_amd64.pyd,sha256=zvtZUC94OYGNWQwhuRuNlexdzGHnPD9_oS2eh7c0UVg,1295360
+gaussian_splatting/diff_gaussian_rasterization/_C.cp311-win_amd64.pyd,sha256=HAKRQZm_kbtdaDJHDsYRMRh-NEMqOA82w_NlzoL9KAs,1295360
 gaussian_splatting/diff_gaussian_rasterization/__init__.py,sha256=a9D0IZiPx-Mk1795hSq54T-NYT4MtEN_MZrxeMhw0Eo,6705
-gaussian_splatting/simple_knn/_C.cp311-win_amd64.pyd,sha256=M41Nx5h9TYuGtbqsUQ7OkCwEZPhHHk_axTnoyzpKFBc,1164288
+gaussian_splatting/simple_knn/_C.cp311-win_amd64.pyd,sha256=iZcwQ6Njb2y2TkBwd6j8PFaJ5nNWuoTnORlX-R7UXLA,1164288
 gaussian_splatting/trainer/__init__.py,sha256=962fEY8A0spSQn5de_d_LkPOjA1PYKrLbuAkxwZo7mI,940
 gaussian_splatting/trainer/abc.py,sha256=kpYnJjLOhsyhE-V2J79EC9nih6MYBcXkmK9cHUA-3ao,4022
-gaussian_splatting/trainer/base.py,sha256=FLm40Ks9I_e9IPn73iW3TpSbyp-kXcJdrMuCn8Aqyt8,3613
+gaussian_splatting/trainer/base.py,sha256=gO1x4m82xrZNl8NZVw2CWYqIvZJIMUWmBtPZQPeyxJ0,3370
 gaussian_splatting/trainer/camera_trainable.py,sha256=TBQXn2f578qeizPz6tgqFm-GRvttv9duuB1xx7_J9TQ,4567
 gaussian_splatting/trainer/combinations.py,sha256=7NX4fXdDOx8ri1_mgAaWNx-YVdo5XsqMlr9qy-Ll2MM,5329
-gaussian_splatting/trainer/depth.py,sha256=EkFfOKdTYUkm4yZwsa9mx0NUf3jYqM74xOYefQvESAA,7252
+gaussian_splatting/trainer/depth.py,sha256=PxWBSNxzoQcRfCFI_yJnJMS6s8qFWn81CXK6O6ffXL0,7059
 gaussian_splatting/trainer/opacity_reset.py,sha256=KfxDyWBNocETGcqCRTdE1n3t63HmjChaAuIP3OTIWtg,2615
 gaussian_splatting/trainer/sh_lift.py,sha256=Hwcn_cRzXZChESpTL83ZmR608ewCR2OzItt-wZtRpak,1220
 gaussian_splatting/trainer/densifier/__init__.py,sha256=cg4aGUolq5ayWtoqQP_BEmHE4NOD5ZuzCluRclJS61I,359
@@ -45,8 +45,8 @@ gaussian_splatting/utils/lpipsPyTorch/modules/__init__.py,sha256=47DEQpj8HBSa-_T
 gaussian_splatting/utils/lpipsPyTorch/modules/lpips.py,sha256=YScu0oXIEstCCjJVRItS_R_csUw70sBMFuP8Syl2UdI,1187
 gaussian_splatting/utils/lpipsPyTorch/modules/networks.py,sha256=kqIebq7dAhHypTXweFVEf_RDbN7_Zv7O3MlD-CfRvpg,2788
 gaussian_splatting/utils/lpipsPyTorch/modules/utils.py,sha256=TDcem3E3HqDNN2MT8qlOL_BKVHeO4HRE77JxF-kOWk8,915
-gaussian_splatting-1.17.8.dist-info/licenses/LICENSE.md,sha256=bMuRQKn0u485mx8JBBTJ5Simc-aWHaQsxmoB6jsg5oE,4752
-gaussian_splatting-1.17.8.dist-info/METADATA,sha256=IPr4HWH0jywGEhZM7CywpdH8sDrBHrpLHrC72TqqPXk,17046
-gaussian_splatting-1.17.8.dist-info/WHEEL,sha256=JLOMsP7F5qtkAkINx5UnzbFguf8CqZeraV8o04b0I8I,101
-gaussian_splatting-1.17.8.dist-info/top_level.txt,sha256=uaYrPYXRHhpybgCnsoazTcdhpzZGnLT_vd5eoRzBWWI,19
-gaussian_splatting-1.17.8.dist-info/RECORD,,
+gaussian_splatting-1.17.9.dist-info/licenses/LICENSE.md,sha256=bMuRQKn0u485mx8JBBTJ5Simc-aWHaQsxmoB6jsg5oE,4752
+gaussian_splatting-1.17.9.dist-info/METADATA,sha256=JtIE2YU2bkhpdJQ-u0TiCWAQXr70eNk7FXSH3d-oS5o,17046
+gaussian_splatting-1.17.9.dist-info/WHEEL,sha256=JLOMsP7F5qtkAkINx5UnzbFguf8CqZeraV8o04b0I8I,101
+gaussian_splatting-1.17.9.dist-info/top_level.txt,sha256=uaYrPYXRHhpybgCnsoazTcdhpzZGnLT_vd5eoRzBWWI,19
+gaussian_splatting-1.17.9.dist-info/RECORD,,