Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/XiaoBin2001/Improved-GS
/ functions
Functions
319 in github.com/XiaoBin2001/Improved-GS
⨍
Functions
319
◇
Types & classes
45
↓ 1 callers
Method
_ensure_optimizer_parameter_layouts
Make optimizer-updated parameters contiguous before optimizer creation.
scene/gaussian_model_optimizer.py:29
↓ 1 callers
Function
_evaluate_report_view
Render one validation view and compute all requested image metrics. The train/test exposure mode compares only the held-out half of
scene/training_runtime.py:282
↓ 1 callers
Method
_get_optimizer_step_value
Read an optimizer step counter whether it is stored as a tensor or int.
scene/gaussian_model_optimizer.py:53
↓ 1 callers
Function
_load_colmap_initial_point_cloud
Read or generate the initial point cloud for a Colmap scene. `sfm` converts COLMAP points to PLY on first use. `random` creates the
scene/dataset_readers.py:264
↓ 1 callers
Function
_load_or_create_gaussians
Fill the Gaussian model either from a saved iteration or the initial point cloud. Signature checks keep compatibility with related G
scene/__init__.py:61
↓ 1 callers
Function
_load_scene_info
Detect the dataset type from files on disk and load matching scene metadata.
scene/__init__.py:24
↓ 1 callers
Function
_parse_metric_float
Parse a metric for sorting, treating missing values as negative infinity.
utils/batch_training/results.py:93
↓ 1 callers
Method
_prune_optimizer
Apply a keep mask to all optimizer parameters and matching state tensors.
scene/gaussian_model_optimizer.py:287
↓ 1 callers
Function
_read_colmap_camera_metadata
Read Colmap camera extrinsics and intrinsics. Binary files are preferred because they are the normal COLMAP output; text fil
scene/dataset_readers.py:199
↓ 1 callers
Function
_read_depth_params
Read depth-scale metadata and attach the median valid scale to each entry.
scene/dataset_readers.py:216
↓ 1 callers
Method
_register_group
Convert fields in `self._fields` into argparse options. A leading underscore means the option also gets a one-letter short
arguments/__init__.py:54
↓ 1 callers
Method
_replace_feature_rest_with_optimizer_state
Replace `f_rest` and copy compatible optimizer momentum into the new tensor.
scene/gaussian_model_sh.py:19
↓ 1 callers
Function
_resolve_camera_device
Convert a device string to torch.device, falling back to CUDA on failure.
scene/cameras.py:21
↓ 1 callers
Function
_resolve_colmap_test_camera_names
Build the Colmap test-view name list from evaluation settings. LLFF-style scenes use every `llffhold`-th sorted image as test. Other
scene/dataset_readers.py:244
↓ 1 callers
Function
_resolve_gaussian_adam_feature_width
Resolve how many contiguous scalar values each Gaussian owns in one GaussianAdam parameter group.
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:506
↓ 1 callers
Method
_restore_optimizer_state
Restore checkpoint optimizer state by parameter group name. Tensor states are copied only when their shape still matches the
scene/gaussian_model_optimizer.py:118
↓ 1 callers
Function
_run_3dgs_improvedgs_budget_densification
Run ImprovedGS budgeted densification on densification intervals. Edge-aware scores are computed when enabled. The Gaussian model th
scene/methods/densification_stage.py:295
↓ 1 callers
Function
_run_3dgs_mcmc_interval_budget_densification
Apply the MCMC update by relocating weak Gaussians and adding new ones.
scene/methods/densification_stage.py:361
↓ 1 callers
Function
_run_3dgs_minigs_rule_densification
Apply MiniGS structure updates. Most intervals split blurred Gaussians with the MiniGS mask. At the reinitialization interva
scene/methods/densification_stage.py:321
↓ 1 callers
Function
_run_3dgs_original_interval_densification
Run the original 3DGS clone/split/prune rule at configured intervals.
scene/methods/densification_stage.py:277
↓ 1 callers
Function
_run_minigs_reinitialization_3dgs
Rebuild MiniGS Gaussians from depth back-sampled training pixels. Each view contributes samples from pixels with low accumulated alp
scene/methods/densification_stage.py:188
↓ 1 callers
Function
_sample_improvedgs_views
Pick the training views used for ImprovedGS edge-aware scoring. The function keeps a small rotating pool in runtime state so repeate
scene/methods/densification_stage.py:37
↓ 1 callers
Function
_schedule_next_rap_prune_if_needed
After an opacity reset, schedule RAP's delayed pruning step. The delayed step removes a configured low-opacity percentage after the
scene/methods/pruning_methods.py:34
↓ 1 callers
Function
_select_training_background
Choose the background color for this render. Random background is a training augmentation. Otherwise the dataset's fixed whi
scene/methods/optimization_methods.py:34
↓ 1 callers
Function
_should_reset_opacity
Check whether the current iteration lands on an opacity-reset interval.
scene/methods/densification_stage.py:30
↓ 1 callers
Function
_should_run_rap_opacity_reset
Check whether RAP should reset opacity on this iteration. RAP resets only inside the densification window and only on the normal
scene/methods/pruning_methods.py:19
↓ 1 callers
Function
_ssim
Apply the SSIM formula using prebuilt convolution windows.
utils/loss_utils.py:54
↓ 1 callers
Function
_step_dense_optimizer
Step either the custom GaussianAdam optimizer or the default Adam optimizer.
scene/training_loop.py:187
↓ 1 callers
Function
_validate_dense_optimizer_type
Ensure the model optimizer type matches the current config.
scene/training_loop.py:126
↓ 1 callers
Function
_write_initial_scene_files
Copy the starting point cloud and camera metadata into the experiment folder.
scene/__init__.py:43
↓ 1 callers
Method
add_new_gs
Delegate MCMC Gaussian insertion and return the number of added points.
scene/gaussian_model_densification.py:246
↓ 1 callers
Function
apply_best_repeat_selection_by_psnr
Keep only best-repeat outputs on disk and return rows for the winners. Non-winning repeat folders are removed. The winning folder is
utils/batch_training/results.py:158
↓ 1 callers
Function
apply_gns_opacity_regularization
Add GNS opacity regularization during the pruning window. The strength is adjusted every few iterations by comparing the current
scene/methods/regularization_methods.py:45
↓ 1 callers
Function
apply_gns_pruning
Run GNS pruning logic for the current iteration. The first pruning iteration scales the opacity learning rate. During the pr
scene/methods/pruning_methods.py:148
↓ 1 callers
Function
apply_gns_threshold_prune
Remove Gaussians below the opacity threshold without crossing the budget. If threshold pruning would remove too many Gaussians, the
scene/methods/pruning_methods.py:82
↓ 1 callers
Method
apply_mcmc_noise
Add MCMC SGLD noise to positions through the method helper.
scene/gaussian_model_densification.py:252
↓ 1 callers
Function
apply_mcmc_regularization
Add MCMC's fixed opacity and scale penalties to the current loss.
scene/methods/regularization_methods.py:17
↓ 1 callers
Function
apply_regularization_method
Select the regularization rule for the active training method.
scene/methods/regularization_methods.py:84
↓ 1 callers
Function
attach_scene_and_gaussians_to_context
Attach scene-dependent objects and initialize the runtime-state cache.
scene/training_context.py:130
↓ 1 callers
Function
backup_batch_training_config
When one run contains multiple training jobs, back up the batch config into the output root.
utils/experiment_utils.py:155
↓ 1 callers
Method
backward
( ctx: Any, opt_grad: torch.Tensor, )
submodules/fused-ssim/fused_ssim/__init__.py:35
↓ 1 callers
Function
benchmark_fps
Measure average FPS by repeatedly rendering all views. Warmup rounds let CUDA kernels and caches settle before the timed rounds.
utils/evaluation/render_metric.py:112
↓ 1 callers
Function
build_background
Create the CUDA background tensor that matches the dataset setting.
utils/evaluation/render_metric.py:63
↓ 1 callers
Function
build_gaussian_model_3dgs
Create a GaussianModel and attach method-specific initialization flags.
scene/methods/initialization_3dgs.py:25
↓ 1 callers
Function
build_opacity_report
Build the text content for the standalone opacity report.
utils/evaluation/render_metric.py:175
↓ 1 callers
Function
build_postprocess_payload
Build post-processing arguments from config and matching train settings. Dataset-related train options are copied so post-processing
utils/batch_training/config.py:164
↓ 1 callers
Function
build_random_colmap_point_cloud
Build a random initial point cloud for Colmap scenes. The random points are sampled inside a cube scaled by the scene radius,
scene/dataset_readers.py:177
↓ 1 callers
Function
build_split_specs
Build the train/test split plan requested by CLI flags.
utils/evaluation/render_metric.py:92
↓ 1 callers
Function
build_training_context
Build the context skeleton before scene creation. Method settings are parsed early and written back to `opt`, so later modul
scene/training_context.py:107
↓ 1 callers
Function
build_training_loop_state
Build shared loop state after scene creation. This attaches scene-related objects into `context.runtime_state`, creates the
scene/training_runtime.py:107
↓ 1 callers
Function
build_training_method_config
Normalize the selected training method and resolve method component flags.
scene/methods/training_config.py:23
↓ 1 callers
Function
build_training_parameters_payload
Build the training-parameter record saved in one experiment folder and add the start time.
utils/experiment_utils.py:53
↓ 1 callers
Function
camera_to_JSON
Convert a Camera object into the compact JSON format saved with outputs.
utils/camera_utils.py:99
↓ 1 callers
Method
capture
Pack model tensors and optimizer state into the checkpoint tuple format.
scene/gaussian_model.py:77
↓ 1 callers
Method
cat_tensors_to_optimizer
Append new Gaussian tensors to optimizer parameters and extend state buffers.
scene/gaussian_model_optimizer.py:312
↓ 1 callers
Function
cleanup_stale_lock
Remove a GPU lock file when the process that created it is gone.
utils/batch_training/gpu.py:57
↓ 1 callers
Function
collect_existing_total_rows
Read result rows from existing output folders without launching jobs.
utils/batch_training/results.py:224
↓ 1 callers
Function
compute_edge_map
Compute a grayscale edge map from a training image. The image is converted to an 8-bit grayscale approximation and filtered
scene/methods/densification_methods.py:35
↓ 1 callers
Function
compute_gns_pre_l1_regularization
Compute GNS pre-activation opacity regularization. GNS regularizes raw opacity logits instead of activated opacity values, w
scene/methods/regularization_methods.py:33
↓ 1 callers
Function
compute_relocation
( opacity_old: torch.Tensor, scale_old: torch.Tensor, N: torch.Tensor, binoms: torch.Tensor,
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:160
↓ 1 callers
Function
compute_relocation_cuda
Clamp sample counts and call the CUDA relocation kernel.
utils/reloc_utils.py:24
↓ 1 callers
Method
construct_list_of_attributes
Build the ordered PLY attribute list for all saved Gaussian fields.
scene/gaussian_model_io.py:69
↓ 1 callers
Function
count_total_runs
Check whether the current config contains batch training, which decides whether to back up the batch config.
utils/experiment_utils.py:146
↓ 1 callers
Function
create_window
Build a channel-wise 2D Gaussian window for SSIM convolution.
utils/loss_utils.py:27
↓ 1 callers
Method
densify_and_prune
Run original 3DGS densification, then prune Gaussians that meet the pruning rules.
scene/gaussian_model_densification.py:139
↓ 1 callers
Method
densify_and_prune_improved
Select candidate Gaussians with absolute screen-space gradients, then run long-axis split under the budget. In late training, fall back t
scene/gaussian_model_densification.py:273
↓ 1 callers
Method
densify_and_prune_split
Reuse the original clone branch, then merge the MiniGS blur mask during split to match the reference densification behavior.
scene/gaussian_model_densification.py:159
↓ 1 callers
Method
densify_and_split_mask
Add the MiniGS blur mask to the original split candidates, so highly blurred areas can split even when gradients are weak.
scene/gaussian_model_densification.py:179
↓ 1 callers
Function
dict_to_cli_args
Convert a payload dictionary into argparse-style `--key value` tokens.
utils/batch_training/config.py:100
↓ 1 callers
Function
evaluate_split
Evaluate one split from start to finish. For each view, the function renders the image, optionally saves render and ground t
utils/evaluation/render_metric.py:238
↓ 1 callers
Function
filter_config_scenes
Return only scenes requested by the filter and report unknown names early.
utils/batch_training/config.py:63
↓ 1 callers
Function
finalize_training_parameters
After training ends, add the end time and pure training duration to the parameter record.
utils/experiment_utils.py:72
↓ 1 callers
Method
flush
Flush the wrapped stdout stream.
utils/general_utils.py:123
↓ 1 callers
Function
fused_ssim
( img1: torch.Tensor, img2: torch.Tensor, padding: str = "same", train: bool = True, )
submodules/fused-ssim/fused_ssim/__init__.py:48
↓ 1 callers
Function
gaussian
Build a normalized 1D Gaussian kernel used by SSIM.
utils/loss_utils.py:20
↓ 1 callers
Function
getProjectionMatrix
Build the perspective projection matrix used by the main renderer.
utils/graphics_utils.py:46
↓ 1 callers
Function
get_center_and_diag
Return the average camera center and the largest distance from it.
scene/dataset_readers.py:58
↓ 1 callers
Function
get_combined_args
Merge command-line arguments with the saved `cfg_args` file in model_path. Values provided on the command line take priority, while
arguments/__init__.py:324
↓ 1 callers
Function
get_empty_tensor
(device: torch.device, dtype: torch.dtype = torch.float32)
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:23
↓ 1 callers
Function
get_network
(net_type: str)
lpipsPyTorch/modules/networks.py:12
↓ 1 callers
Function
get_state_dict
(net_type: str = 'alex', version: str = '0.1')
lpipsPyTorch/modules/utils.py:11
↓ 1 callers
Function
initialize_runtime_state
Create method-specific runtime state after the scene and model exist. ImprovedGS/GNS precompute edge maps for edge-aware scoring. Mi
scene/training_context.py:63
↓ 1 callers
Function
install_signal_handlers
Register cleanup handlers for SIGINT and SIGTERM.
utils/batch_training/processes.py:60
↓ 1 callers
Function
inverse_sigmoid
Convert opacity values from sigmoid space back to raw logits.
utils/general_utils.py:23
↓ 1 callers
Function
is_inplace_checkpoint_resume
Check whether resume should reuse the current output folder. An empty checkpoint directory means the checkpoint is expected inside
scene/training_runtime.py:270
↓ 1 callers
Function
l1_loss
Return mean absolute pixel error between prediction and ground truth.
utils/loss_utils.py:14
↓ 1 callers
Function
loadCam
Load one image/depth pair and convert a CameraInfo record into a Camera. Resolution handling follows the original 3DGS behavior: fix
utils/camera_utils.py:20
↓ 1 callers
Function
load_batch_config
Load one batch JSON config, returning an empty dict for missing content.
utils/batch_training/config.py:49
↓ 1 callers
Function
load_result_summary
Load one `result_<split>.json` file if it exists and contains data.
utils/batch_training/results.py:25
↓ 1 callers
Function
load_training_checkpoint
Load checkpoint payload and validate that it matches the supported model type. The loader accepts both the current dictionary format
scene/training_runtime.py:222
↓ 1 callers
Method
long_axis_split
Sample candidates by importance, create two child Gaussians by moving along the longest axis in both directions, then adjust scale and op
scene/gaussian_model_densification.py:312
↓ 1 callers
Function
main
Parse batch arguments and process every resolved config path.
utils/batch_training/main.py:126
↓ 1 callers
Method
minigs_forward_aux
( self, means3D: torch.Tensor, means2D: torch.Tensor, opacities: torch.Tensor,
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:383
↓ 1 callers
Function
minigs_rasterize_gaussians_aux
( means3D: torch.Tensor, means2D: torch.Tensor, dc: torch.Tensor, sh: torch.Tensor, opacit
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:92
↓ 1 callers
Function
minigs_rasterize_gaussians_depth
( means3D: torch.Tensor, means2D: torch.Tensor, dc: torch.Tensor, sh: torch.Tensor, opacit
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:126
↓ 1 callers
Method
minigs_render_depth
( self, means3D: torch.Tensor, means2D: torch.Tensor, opacities: torch.Tensor,
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:415
↓ 1 callers
Function
mkdir_p
Create a folder and ignore the error when it already exists.
utils/general_utils.py:137
↓ 1 callers
Function
namespace_to_dict
Convert a parameter object into a plain dictionary that can be written to JSON.
utils/experiment_utils.py:23
↓ 1 callers
Function
normalize_activation
(x: torch.Tensor, eps: float = 1e-10)
lpipsPyTorch/modules/utils.py:6
↓ 1 callers
Function
opacity_sigmoid
Gate noise so low-opacity Gaussians receive stronger perturbations.
scene/methods/mcmc_ops.py:147
← previous
next →
101–200 of 319, ranked by callers