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
Function
parse_args
Parse command-line options for the batch runner.
utils/batch_training/config.py:24
↓ 1 callers
Function
parse_cfg_args_namespace
Safely parse the legacy `cfg_args` Namespace(...) text format.
arguments/__init__.py:301
↓ 1 callers
Function
parse_scene_filter
Convert a comma-separated scene filter into a set of scene names.
utils/batch_training/config.py:56
↓ 1 callers
Function
parse_training_arguments
Build the training CLI parser, register all option groups, and parse argv.
scene/training_runtime.py:162
↓ 1 callers
Function
prepare_edge_maps
Precompute one edge map per training camera for later scoring calls.
scene/methods/densification_methods.py:58
↓ 1 callers
Function
prepare_output_and_logger
Prepare the experiment folder before training starts. Existing output is archived unless this run resumes in place. The current
scene/training_runtime.py:77
↓ 1 callers
Function
print_split_summary
Print one split's aggregate metrics in a compact readable block.
utils/evaluation/render_metric.py:224
↓ 1 callers
Function
processTiles
submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h:199
↓ 1 callers
Function
process_exists
Return whether a local process id still exists.
utils/batch_training/gpu.py:46
↓ 1 callers
Function
process_splits
Import and run the heavy post-processing implementation on demand.
postprocess.py:14
↓ 1 callers
Function
query_gpu_stats
Read GPU memory and utilization numbers from `nvidia-smi`.
utils/batch_training/gpu.py:20
↓ 1 callers
Function
rasterize_gaussians
( means3D: torch.Tensor, means2D: torch.Tensor, dc: torch.Tensor, sh: torch.Tensor, opacit
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:70
↓ 1 callers
Function
readColmapCameras
Read Blender/NeRF transform JSON and convert frames into CameraInfo records.
scene/dataset_readers.py:83
↓ 1 callers
Function
read_extrinsics_binary
see: src/base/reconstruction.cc void Reconstruction::ReadImagesBinary(const std::string& path) void Reconstruction::WriteImagesBinary(const s
scene/colmap_loader.py:169
↓ 1 callers
Function
read_extrinsics_text
Taken from https://github.com/colmap/colmap/blob/dev/scripts/python/read_write_model.py
scene/colmap_loader.py:228
↓ 1 callers
Function
read_intrinsics_binary
see: src/base/reconstruction.cc void Reconstruction::WriteCamerasBinary(const std::string& path) void Reconstruction::ReadCamerasBinary(const
scene/colmap_loader.py:201
↓ 1 callers
Function
read_intrinsics_text
Taken from https://github.com/colmap/colmap/blob/dev/scripts/python/read_write_model.py
scene/colmap_loader.py:145
↓ 1 callers
Function
read_points3D_binary
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DBinary(const std::string& path) void Reconstruction::WritePoints3DBinary(con
scene/colmap_loader.py:116
↓ 1 callers
Function
read_points3D_text
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DText(const std::string& path) void Reconstruction::WritePoints3DText(const s
scene/colmap_loader.py:76
↓ 1 callers
Function
read_training_elapsed_minutes
Read pure training time saved by the training script. A legacy metadata file is checked as a fallback so old outputs can still
utils/evaluation/render_metric.py:45
↓ 1 callers
Method
reinitial_pts
Rebuild the Gaussian set from MiniGS periodic back-sampling results. Keep the current SH settings and exposure parameters so `training_se
scene/gaussian_model_densification.py:365
↓ 1 callers
Function
release_gpu_lock
Remove a lock file after the scene job finishes.
utils/batch_training/gpu.py:112
↓ 1 callers
Method
relocate_gs
Delegate MCMC relocation of low-opacity Gaussians to the method helper.
scene/gaussian_model_densification.py:240
↓ 1 callers
Function
render_3dgs
Render one camera view with the standard 3DGS parameter tensors. The function always passes DC color, higher-order SH, scale, rotati
gaussian_renderer/__init__.py:169
↓ 1 callers
Function
render_minigs_aux
Render MiniGS auxiliary statistics for the current view. The CUDA path returns projected area and accumulated weights. MiniGS uses
gaussian_renderer/__init__.py:268
↓ 1 callers
Function
render_minigs_depth
Render MiniGS depth back-sampling data for one view. The returned 3D points, alpha map, and Gaussian indices are used when M
gaussian_renderer/__init__.py:314
↓ 1 callers
Function
render_path_for
Return the output folder for one split and model iteration.
utils/evaluation/render_metric.py:38
↓ 1 callers
Method
replace_tensor_to_optimizer
Replace one optimizer parameter tensor and reset its momentum buffers.
scene/gaussian_model_optimizer.py:225
↓ 1 callers
Function
resolve_config_paths
Resolve either a single config file or all runnable configs in a folder.
utils/batch_training/config.py:77
↓ 1 callers
Function
resolve_initial_sh_degree_3dgs
Choose the starting SH degree for the Gaussian model. MiniGS starts from degree 0 and restores higher SH capacity later; other
scene/methods/initialization_3dgs.py:13
↓ 1 callers
Function
resolve_start_checkpoint_path
Return the checkpoint path to load, or None when resume is disabled. If the checkpoint directory is omitted, the current model outpu
scene/training_runtime.py:188
↓ 1 callers
Method
restore
Restore model tensors, gradient accumulators, and optimizer state.
scene/gaussian_model_optimizer.py:167
↓ 1 callers
Function
run_3dgs_densification_method
Dispatch one iteration's densification work for the active method. The caller already produced `render_state` during optimization. T
scene/methods/densification_stage.py:380
↓ 1 callers
Function
run_3dgs_optimization_method
Render one camera and create gradients for this iteration. The returned `render_state` keeps only the render outputs needed by later
scene/methods/optimization_methods.py:53
↓ 1 callers
Function
run_3dgs_parameter_update_method
Apply optimizer updates for one iteration. Exposure is always stepped with its own Adam optimizer. Dense Gaussian tensors fo
scene/training_loop.py:203
↓ 1 callers
Function
run_3dgs_pruning_method
Dispatch RAP and GNS pruning actions for one training iteration.
scene/methods/pruning_methods.py:186
↓ 1 callers
Function
run_log_and_save_stage
Run scheduled evaluation, model saving, and checkpoint saving. The return value is the time spent in reporting, which the caller
scene/training_loop.py:54
↓ 1 callers
Function
run_single_config
Execute or rebuild one batch config. The function owns the top-level order for one config: validate scenes, prepare output r
utils/batch_training/main.py:39
↓ 1 callers
Function
sample_training_viewpoint
Draw one random training camera without replacement. When the local pool becomes empty, it is refilled from the scene so each
scene/training_runtime.py:149
↓ 1 callers
Method
save_ply
Write current raw Gaussian tensors to a PLY point-cloud file.
scene/gaussian_model_io.py:86
↓ 1 callers
Function
save_training_checkpoint
Save Gaussian model state and iteration in the current checkpoint format.
scene/training_runtime.py:204
↓ 1 callers
Function
searchForMaxIteration
Return the largest saved iteration number from an iteration folder.
utils/general_utils.py:150
↓ 1 callers
Function
select_best_repeat_outputs_by_psnr
Choose the winning repeat output name for each scene by PSNR. Test rows are preferred when available; train rows are used as a fallb
utils/batch_training/results.py:110
↓ 1 callers
Function
select_scene_gpu
Return the GPU id to expose to the child process plus its lock path.
utils/batch_training/gpu.py:124
↓ 1 callers
Method
setup_functions
Register activation and inverse-activation functions for raw parameters.
scene/gaussian_model.py:34
↓ 1 callers
Function
should_run_parameter_update
Decide whether this iteration should step trainable parameters. ImprovedGS/GNS can use MU, which skips dense optimizer steps later i
scene/training_loop.py:155
↓ 1 callers
Function
summarize_opacity_intervals
Count Gaussians in opacity intervals to make model sparsity easy to inspect.
utils/evaluation/render_metric.py:152
↓ 1 callers
Function
training
Run one complete training job and return pure training time in seconds. The function keeps the high-level training order in one plac
train.py:39
↓ 1 callers
Function
training_report
Evaluate train and test cameras and write a compact report. The function averages metrics over each split, prints them through tqdm,
scene/training_runtime.py:320
↓ 1 callers
Function
transformPoint4x3
submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h:75
↓ 1 callers
Function
transformPoint4x4
submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h:85
↓ 1 callers
Function
update_3dgs_training_schedule
Update learning rate and SH degree before rendering the current iteration.
scene/training_loop.py:145
↓ 1 callers
Method
update_learning_rate
Update exposure and xyz learning rates for the current iteration.
scene/gaussian_model_optimizer.py:211
↓ 1 callers
Function
update_training_progress_bar
Update the tqdm postfix with smoothed loss and current Gaussian count.
scene/training_loop.py:37
↓ 1 callers
Function
write_opacity_report
Save opacity interval statistics next to the experiment output.
utils/evaluation/render_metric.py:207
↓ 1 callers
Function
write_training_parameters
Write the parameters used by this run into the experiment folder for reproduction and post-processing.
utils/experiment_utils.py:63
↓ 1 callers
Method
z_score
(self, x: torch.Tensor)
lpipsPyTorch/modules/networks.py:50
Function
PYBIND11_MODULE
submodules/fused-ssim/ext.cpp:4
Function
PYBIND11_MODULE
submodules/diff-gaussian-rasterization/ext.cpp:15
Function
PYBIND11_MODULE
submodules/simple-knn/ext.cpp:15
Function
Pix2ndc
submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h:46
Method
__init__
Create an empty Gaussian model with the requested SH capacity.
scene/gaussian_model.py:47
Method
__init__
Load cameras, scene bounds, and Gaussian state for one experiment.
scene/__init__.py:103
Method
__init__
Load image/depth tensors and precompute transforms for rendering.
scene/cameras.py:65
Method
__init__
Store projection fields and derive the camera center from the view matrix.
scene/cameras.py:150
Method
__init__
(self, silent: bool)
utils/general_utils.py:110
Method
__init__
(self, parser: ArgumentParser, name: str, fill_none: bool = False)
arguments/__init__.py:47
Method
__init__
(self, parser: ArgumentParser, sentinel: bool = False)
arguments/__init__.py:112
Method
__init__
(self, parser: ArgumentParser, sentinel: bool = False)
arguments/__init__.py:141
Method
__init__
(self, parser: ArgumentParser)
arguments/__init__.py:153
Method
__init__
(self, parser: ArgumentParser)
arguments/__init__.py:189
Method
__init__
(self, parser: ArgumentParser)
arguments/__init__.py:203
Method
__init__
(self, parser: ArgumentParser)
arguments/__init__.py:217
Method
__init__
(self, parser: ArgumentParser)
arguments/__init__.py:240
Method
__init__
(self, parser: ArgumentParser)
arguments/__init__.py:264
Method
__init__
(self, parser: ArgumentParser)
arguments/__init__.py:278
Method
__init__
(self, raster_settings: GaussianRasterizationSettings)
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:304
Method
__init__
(self, n_channels_list: Sequence[int])
lpipsPyTorch/modules/networks.py:24
Method
__init__
(self)
lpipsPyTorch/modules/networks.py:37
Method
__init__
(self)
lpipsPyTorch/modules/networks.py:67
Method
__init__
(self)
lpipsPyTorch/modules/networks.py:78
Method
__init__
(self, net_type: str = 'alex', version: str = '0.1')
lpipsPyTorch/modules/lpips.py:17
Function
add_new_gs
Add new Gaussians by sampling existing ones until the budget cap is reached.
scene/methods/mcmc_ops.py:113
Function
apply_mcmc_noise
Add covariance-shaped position noise scaled by opacity and xyz learning rate.
scene/methods/mcmc_ops.py:140
Method
backward
( ctx: Any, grad_out_color: torch.Tensor, _: torch.Tensor, grad_out_depth: tor
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:217
Function
dnormvdv
submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h:124
Function
dnormvdz
submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h:116
Function
duplicateToTilesTouched
submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h:280
Method
extract
Extract model options and normalize the source path to an absolute path.
arguments/__init__.py:128
Method
extract
Merge all optimization parameter groups into one object for training.
arguments/__init__.py:289
Method
forward
( ctx: Any, C1: float, C2: float, img1: torch.Tensor, img2: torch.Tens
submodules/fused-ssim/fused_ssim/__init__.py:13
Method
forward
( ctx: Any, means3D: torch.Tensor, means2D: torch.Tensor, dc: torch.Tensor,
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:174
Method
forward
( self, means3D: torch.Tensor, means2D: torch.Tensor, opacities: torch.Tensor,
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:308
Method
forward
(self, x: torch.Tensor)
lpipsPyTorch/modules/networks.py:53
Method
forward
(self, x: torch.Tensor, y: torch.Tensor)
lpipsPyTorch/modules/lpips.py:30
Function
getRect
submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h:51
Method
get_features_dc
Return DC color SH coefficients.
scene/gaussian_model.py:127
Method
get_features_rest
Return higher-order SH color coefficients.
scene/gaussian_model.py:134
Method
get_opacity
Return activated opacity values in `[0, 1]`.
scene/gaussian_model.py:141
Method
get_rotation
Return normalized quaternion rotations.
scene/gaussian_model.py:113
← previous
next →
201–300 of 319, ranked by callers