gaussian-splatting 1.17.1__tar.gz → 1.17.2__tar.gz

{gaussian_splatting-1.17.1/gaussian_splatting.egg-info → gaussian_splatting-1.17.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gaussian_splatting
-Version: 1.17.1
+Version: 1.17.2
 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.1 → gaussian_splatting-1.17.2}/gaussian_splatting/camera.py

@@ -1,5 +1,5 @@
 import os
-from typing import NamedTuple, Callable
+from typing import NamedTuple, Callable, Tuple
 import torch
 import logging
 from .utils import fov2focal, focal2fov, getProjectionMatrix, getWorld2View2, read_image, read_image_mask, read_depth, read_depth_mask, matrix_to_quaternion
@@ -62,6 +62,7 @@ def build_camera(
         R: torch.Tensor, T: torch.Tensor,
         image_path: str = None, image_mask_path: str = None,
         depth_path: str = None, depth_mask_path: str = None,
+        bg_color: Tuple[float, float, float] = (0., 0., 0.),
         device="cuda", custom_data: dict = {}
 ):
     R, T = R.to(device=device, dtype=torch.float), T.to(device=device, dtype=torch.float)
@@ -126,6 +127,7 @@ def build_camera(
         ground_truth_depth=gt_depth,
         ground_truth_depth_mask_path=depth_mask_path,
         ground_truth_depth_mask=gt_depth_mask,
+        bg_color=torch.tensor(bg_color, dtype=torch.float32, device=device),
         custom_data=custom_data,
     )
 

{gaussian_splatting-1.17.1 → gaussian_splatting-1.17.2}/gaussian_splatting/camera_trainable.py

@@ -27,7 +27,7 @@ class CameraTrainableGaussianModel(GaussianModel):
             image_width=int(viewpoint_camera.image_width),
             tanfovx=tanfovx,
             tanfovy=tanfovy,
-            bg=viewpoint_camera.bg_color.to(self._xyz.device),
+            bg=viewpoint_camera.bg_color,
             scale_modifier=self.scale_modifier,
             viewmatrix=viewpoint_camera.world_view_transform,
             projmatrix=viewpoint_camera.full_proj_transform,

{gaussian_splatting-1.17.1 → gaussian_splatting-1.17.2}/gaussian_splatting/gaussian_model.py

@@ -106,7 +106,7 @@ class GaussianModel(nn.Module):
             image_width=int(viewpoint_camera.image_width),
             tanfovx=tanfovx,
             tanfovy=tanfovy,
-            bg=viewpoint_camera.bg_color.to(self._xyz.device),
+            bg=viewpoint_camera.bg_color,
             scale_modifier=self.scale_modifier,
             viewmatrix=viewpoint_camera.world_view_transform,
             projmatrix=viewpoint_camera.full_proj_transform,

{gaussian_splatting-1.17.1 → gaussian_splatting-1.17.2/gaussian_splatting.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gaussian_splatting
-Version: 1.17.1
+Version: 1.17.2
 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.1 → gaussian_splatting-1.17.2}/setup.py

@@ -44,7 +44,7 @@ if os.name == 'nt':
 
 setup(
     name="gaussian_splatting",
-    version='1.17.1',
+    version='1.17.2',
     author='yindaheng98',
     author_email='yindaheng98@gmail.com',
     url='https://github.com/yindaheng98/gaussian-splatting',