Rename utils files

gaussian_splatting/gaussian_renderer/__init__.py

@@ -12,10 +12,8 @@
 import math
 
 import torch
-from diff_gaussian_rasterization import (
-    GaussianRasterizationSettings,
-    GaussianRasterizer,
-)
+from diff_gaussian_rasterization import (GaussianRasterizationSettings,
+                                         GaussianRasterizer)
 
 from gaussian_splatting.scene.gaussian_model import GaussianModel
 

gaussian_splatting/lpipsPyTorch/modules/lpips.py

@@ -1,7 +1,8 @@
 import torch
 import torch.nn as nn
 
-from gaussian_splatting.lpipsPyTorch.modules.networks import LinLayers, get_network
+from gaussian_splatting.lpipsPyTorch.modules.networks import (LinLayers,
+                                                              get_network)
 from gaussian_splatting.lpipsPyTorch.modules.utils import get_state_dict
 
 

gaussian_splatting/scene/__init__.py

@@ -16,11 +16,9 @@
 from gaussian_splatting.arguments import ModelParams
 from gaussian_splatting.scene.dataset_readers import readColmapSceneInfo
 from gaussian_splatting.scene.gaussian_model import GaussianModel
-from gaussian_splatting.utils.camera_utils import (
-    camera_to_JSON,
-    cameraList_from_camInfos,
-)
-from gaussian_splatting.utils.system_utils import searchForMaxIteration
+from gaussian_splatting.utils.camera import (camera_to_JSON,
+                                             cameraList_from_camInfos)
+from gaussian_splatting.utils.system import searchForMaxIteration
 
 
 class Scene:

gaussian_splatting/scene/cameras.py

@@ -13,7 +13,8 @@
 import torch
 from torch import nn
 
-from gaussian_splatting.utils.graphics_utils import getProjectionMatrix, getWorld2View2
+from gaussian_splatting.utils.graphics import (getProjectionMatrix,
+                                               getWorld2View2)
 
 
 class Camera(nn.Module):

gaussian_splatting/scene/dataset_readers.py

@@ -17,17 +17,15 @@
 from PIL import Image
 from plyfile import PlyData, PlyElement
 
-from gaussian_splatting.scene.colmap_loader import (
-    qvec2rotmat,
-    read_extrinsics_binary,
-    read_extrinsics_text,
-    read_intrinsics_binary,
-    read_intrinsics_text,
-    read_points3D_binary,
-    read_points3D_text,
-)
+from gaussian_splatting.scene.colmap_loader import (qvec2rotmat,
+                                                    read_extrinsics_binary,
+                                                    read_extrinsics_text,
+                                                    read_intrinsics_binary,
+                                                    read_intrinsics_text,
+                                                    read_points3D_binary,
+                                                    read_points3D_text)
 from gaussian_splatting.scene.gaussian_model import BasicPointCloud
-from gaussian_splatting.utils.graphics_utils import focal2fov, getWorld2View2
+from gaussian_splatting.utils.graphics import focal2fov, getWorld2View2
 
 
 class CameraInfo(NamedTuple):

gaussian_splatting/scene/gaussian_model.py

@@ -17,16 +17,13 @@
 from simple_knn._C import distCUDA2
 from torch import nn
 
-from gaussian_splatting.utils.general_utils import (
-    build_rotation,
-    build_scaling_rotation,
-    get_expon_lr_func,
-    inverse_sigmoid,
-    strip_symmetric,
-)
-from gaussian_splatting.utils.graphics_utils import BasicPointCloud
-from gaussian_splatting.utils.sh_utils import RGB2SH
-from gaussian_splatting.utils.system_utils import mkdir_p
+from gaussian_splatting.utils.general import (build_rotation,
+                                              build_scaling_rotation,
+                                              get_expon_lr_func,
+                                              inverse_sigmoid, strip_symmetric)
+from gaussian_splatting.utils.graphics import BasicPointCloud
+from gaussian_splatting.utils.sh import RGB2SH
+from gaussian_splatting.utils.system import mkdir_p
 
 
 class GaussianModel:

gaussian_splatting/training.py

@@ -8,9 +8,9 @@
 
 from gaussian_splatting.gaussian_renderer import render
 from gaussian_splatting.scene import GaussianModel, Scene
-from gaussian_splatting.utils.general_utils import safe_state
-from gaussian_splatting.utils.image_utils import psnr
-from gaussian_splatting.utils.loss_utils import l1_loss, ssim
+from gaussian_splatting.utils.general import safe_state
+from gaussian_splatting.utils.image import psnr
+from gaussian_splatting.utils.loss import l1_loss, ssim
 
 
 class Trainer:

gaussian_splatting/utils/camera_utils.py → gaussian_splatting/utils/camera.py

@@ -12,8 +12,8 @@
 import numpy as np
 
 from gaussian_splatting.scene.cameras import Camera
-from gaussian_splatting.utils.general_utils import PILtoTorch
-from gaussian_splatting.utils.graphics_utils import fov2focal
+from gaussian_splatting.utils.general import PILtoTorch
+from gaussian_splatting.utils.graphics import fov2focal
 
 WARNED = False
 

scripts/metrics.py

@@ -20,8 +20,8 @@
 from tqdm import tqdm
 
 from gaussian_splatting.lpipsPyTorch import lpips
-from gaussian_splatting.utils.image_utils import psnr
-from gaussian_splatting.utils.loss_utils import ssim
+from gaussian_splatting.utils.image import psnr
+from gaussian_splatting.utils.loss import ssim
 
 
 def readImages(renders_dir, gt_dir):

scripts/render.py

@@ -20,7 +20,7 @@
 from gaussian_splatting.arguments import ModelParams, get_combined_args
 from gaussian_splatting.gaussian_renderer import GaussianModel, render
 from gaussian_splatting.scene import Scene
-from gaussian_splatting.utils.general_utils import safe_state
+from gaussian_splatting.utils.general import safe_state
 
 
 def render_set(model_path, name, iteration, views, gaussians):