gaussian-splatting 1.17.0__cp311-cp311-win_amd64.whl → 1.17.2__cp311-cp311-win_amd64.whl

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/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/dataset/camera_trainable.py

@@ -104,7 +104,7 @@ class FixedTrainableCameraDataset(JSONCameraDataset):
             postprocess=exposure_postprocess,
             custom_data={
                 **camera.custom_data,
-                'exposures': (torch.tensor(json_camera['exposure'], dtype=torch.float, device=device) if 'exposure' in json_camera else torch.eye(3, 4))
+                'exposures': (torch.tensor(json_camera['exposure'], dtype=torch.float, device=device) if 'exposure' in json_camera else torch.eye(3, 4, dtype=torch.float, device=device))
             }
         ) for camera, json_camera in zip(self.cameras, self.json_cameras)]
         return self

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/render.py

@@ -48,6 +48,7 @@ def build_pcd_rescale(
 def rendering(
         dataset: CameraDataset, gaussians: GaussianModel, save: str, save_pcd: bool = False,
         rescale_depth_gt: bool = True) -> None:
+    os.makedirs(save, exist_ok=True)
     dataset.save_cameras(os.path.join(save, "cameras.json"))
     render_path = os.path.join(save, "renders")
     gt_path = os.path.join(save, "gt")

{gaussian_splatting-1.17.0.dist-info → gaussian_splatting-1.17.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gaussian_splatting
-Version: 1.17.0
+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.0.dist-info → gaussian_splatting-1.17.2.dist-info}/RECORD

@@ -1,20 +1,20 @@
 gaussian_splatting/__init__.py,sha256=CiOZMcyPTAaKtEuMZUhEda_Ad4_RUhmIstB-A3iuOJY,131
-gaussian_splatting/camera.py,sha256=jQdJZaHJABXshZ9LH9U9FC_-fi0bEUZo4m3b71PImbA,8218
-gaussian_splatting/camera_trainable.py,sha256=XHynfc-9Z3d8N00u4rUlyAglwTadT_BDoV4ffOBLcLU,3862
-gaussian_splatting/gaussian_model.py,sha256=Jg0RJXNeCMNjQpZStag2H8JpvTeukjgqnrHVHJRSDcY,13296
+gaussian_splatting/camera.py,sha256=u-U99RW9dCPVqUZxVzrUW7nxHsXgKZ7NXHnSqCNOfKk,8365
+gaussian_splatting/camera_trainable.py,sha256=nI6hFFRV2ev7VwLlKUbzEdN9zUmngYZAANGLr1p1yBA,3841
+gaussian_splatting/gaussian_model.py,sha256=_Dy_dDa2prALhVgg428a-O8-8PODg3c_JPkOJJ8X4o8,13275
 gaussian_splatting/prepare.py,sha256=SdljFF2TU-ETkIY2ceHEz9tGA-o4HfHxGsFKIhyB1Ek,3042
-gaussian_splatting/render.py,sha256=SyvRmiZ1OzYTZf7lQTRPjncQ0vL9297oWF3SgxEfJnA,5892
+gaussian_splatting/render.py,sha256=2dP7Yr5e5uLJAs9MwRJSBY9eZQjaDtf9XB-HdtPmxPc,5930
 gaussian_splatting/train.py,sha256=VAFlPHB5ETH8BiYhe0LWY2UWXyB16tSRPKUaV5VPp5A,5196
 gaussian_splatting/dataset/__init__.py,sha256=-runuT-61P0YVpfV_WXqwUZM1oY0N012YH13Bt3rzSU,138
-gaussian_splatting/dataset/camera_trainable.py,sha256=pHtjXXCAlNbbtTvj56-UogM8BHPjc_CNCwFWxixDYkY,4935
+gaussian_splatting/dataset/camera_trainable.py,sha256=D8bqeLVcnQ4qrsy3U1s6BVzdG4KCCPGR-Tygj6nLp-Q,4969
 gaussian_splatting/dataset/dataset.py,sha256=mlcIS0pJNdUIT-RObcQNYwgxxTOFJ0OYg3AlhoQ4Mww,2315
 gaussian_splatting/dataset/colmap/__init__.py,sha256=YEYT2k2WJSqrkkZq4KAJYS9UMgqU6W6TJaeHLRc1CM4,213
 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=W6MQVsrnBMJ_E8yQfGKvDgvWPto8gaMtGASs85gK0yg,1295360
+gaussian_splatting/diff_gaussian_rasterization/_C.cp311-win_amd64.pyd,sha256=syOz4eY6SNVetWOKQVDDnHMNP4Zm-b0oLswrKqqKYeI,1295360
 gaussian_splatting/diff_gaussian_rasterization/__init__.py,sha256=a9D0IZiPx-Mk1795hSq54T-NYT4MtEN_MZrxeMhw0Eo,6705
-gaussian_splatting/simple_knn/_C.cp311-win_amd64.pyd,sha256=6k02KfkxVFbLQ76zKw5tpwJo0ZEI3ca51jcA3CJv6EM,1164288
+gaussian_splatting/simple_knn/_C.cp311-win_amd64.pyd,sha256=Q_1N0DkOzxSa4GZn9_Az32Qfyx8ScPZgVcCulQHZdBA,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=GrenIGWMXtF-2-b3b1WCA-gQV30TpMQMxBZmPPVKFSg,2795
@@ -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.0.dist-info/licenses/LICENSE.md,sha256=bMuRQKn0u485mx8JBBTJ5Simc-aWHaQsxmoB6jsg5oE,4752
-gaussian_splatting-1.17.0.dist-info/METADATA,sha256=h5nRgV0Ew4gcpFQ9OuD3lb2K8iA2spi7P6o0VYH6BCo,17046
-gaussian_splatting-1.17.0.dist-info/WHEEL,sha256=JLOMsP7F5qtkAkINx5UnzbFguf8CqZeraV8o04b0I8I,101
-gaussian_splatting-1.17.0.dist-info/top_level.txt,sha256=uaYrPYXRHhpybgCnsoazTcdhpzZGnLT_vd5eoRzBWWI,19
-gaussian_splatting-1.17.0.dist-info/RECORD,,
+gaussian_splatting-1.17.2.dist-info/licenses/LICENSE.md,sha256=bMuRQKn0u485mx8JBBTJ5Simc-aWHaQsxmoB6jsg5oE,4752
+gaussian_splatting-1.17.2.dist-info/METADATA,sha256=G245M6QmHj-eOXE9nGsi6cgmi563tdLUbeuE6VDBcYY,17046
+gaussian_splatting-1.17.2.dist-info/WHEEL,sha256=JLOMsP7F5qtkAkINx5UnzbFguf8CqZeraV8o04b0I8I,101
+gaussian_splatting-1.17.2.dist-info/top_level.txt,sha256=uaYrPYXRHhpybgCnsoazTcdhpzZGnLT_vd5eoRzBWWI,19
+gaussian_splatting-1.17.2.dist-info/RECORD,,