MCPcopy Create free account

hub / github.com/XiaoBin2001/Improved-GS / functions

Functions319 in github.com/XiaoBin2001/Improved-GS

↓ 13 callersFunctionread_next_bytes
Read and unpack the next bytes from a binary file. :param fid: :param num_bytes: Sum of combination of {2, 4, 8}, e.g. 2, 6, 16, 30, etc. :pa
scene/colmap_loader.py:69
↓ 13 callersMethodwrite
Write stdout text, appending a timestamp to completed lines.
utils/general_utils.py:113
↓ 9 callersMethodgetTrainCameras
Return training cameras for the requested resolution scale.
scene/__init__.py:170
↓ 9 callersFunctionload_json
Read a JSON config or result file in one common way.
utils/experiment_utils.py:43
↓ 9 callersMethodprune_points
Remove Gaussians selected by `mask` and prune matching optimizer state.
scene/gaussian_model.py:170
↓ 8 callersMethod__init__
(self, parser: ArgumentParser)
arguments/__init__.py:253
↓ 7 callersMethodextract
Copy only this group's parsed fields from the shared argparse namespace.
arguments/__init__.py:94
↓ 6 callersFunctioncomputeEllipseIntersection
submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h:183
↓ 6 callersFunctionformat_metric
Format a metric for JSON/CSV summaries, leaving missing values blank.
utils/evaluation/render_metric.py:29
↓ 5 callersFunction_new_gaussian_cuda_bool_mask
Create an all-false CUDA mask with one entry per current Gaussian.
scene/methods/densification_stage.py:23
↓ 5 callersMethoddensification_postfix
Append newly created Gaussians to all parameter tensors. The optimizer receives the new tensors first, then the model fields
scene/gaussian_model_densification.py:36
↓ 5 callersFunctionfocal2fov
Convert focal length in pixels to field of view in radians.
utils/graphics_utils.py:77
↓ 5 callersMethodsave
Save the current point cloud and exposure table for one iteration.
scene/__init__.py:156
↓ 4 callersMethod__init__
(self)
lpipsPyTorch/modules/networks.py:89
↓ 4 callersFunction_read_bool
Read a boolean option and reject non-boolean values after argparse parsing.
scene/methods/training_config.py:13
↓ 4 callersFunctionbuild_scene_output_stem
Build the base output name from scene name plus global and scene suffixes.
utils/batch_training/config.py:120
↓ 4 callersFunctionmerge_dict
Deep-merge two dictionaries without mutating either input.
utils/batch_training/config.py:36
↓ 4 callersFunctionresolve_regularized_prune_target_budget
Resolve and cache the final Gaussian count used by GNS pruning. `num` mode uses a fixed count. `rate` mode converts a keep percentag
scene/methods/pruning_methods.py:46
↓ 4 callersFunctionsave_json
Save a JSON file with the shared format and create parent folders automatically.
utils/experiment_utils.py:34
↓ 4 callersFunctionsynchronized_timestamp
Return a timer value after pending CUDA work has finished. CUDA kernels are asynchronous, so synchronization is needed before timing
scene/training_runtime.py:44
↓ 3 callersFunctionSH2RGB
Convert a DC spherical-harmonic coefficient back to RGB.
utils/general_utils.py:168
↓ 3 callersFunction_build_screenspace_points
Create the 2D placeholder tensor passed to the rasterizer. The CUDA backward pass writes gradients into this tensor. Channels 0-1
gaussian_renderer/__init__.py:37
↓ 3 callersMethodadd_densification_stats
Accumulate normal screen-space gradient magnitudes for visible Gaussians.
scene/gaussian_model_densification.py:218
↓ 3 callersFunctionbuild_output_name
Add a repeat index to the output name when the config runs repeats.
utils/batch_training/config.py:131
↓ 3 callersFunctionbuild_rotation
Convert normalized quaternions into 3x3 rotation matrices.
utils/general_utils.py:75
↓ 3 callersFunctionbuild_train_payload
Merge train arguments and inject source/model paths for one scene run.
utils/batch_training/config.py:150
↓ 3 callersFunctioncollect_scene_result_rows
Convert one scene output directory into train/test CSV rows.
utils/batch_training/results.py:36
↓ 3 callersFunctioncpu_deep_copy_tuple
(input_tuple: tuple[Any, ...])
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:32
↓ 3 callersFunctionextend_result_rows
Append newly collected rows into the split-indexed row accumulator.
utils/batch_training/results.py:213
↓ 3 callersFunctionfetchPly
Read a PLY point cloud into the BasicPointCloud container.
scene/dataset_readers.py:145
↓ 3 callersFunctionfov2focal
Convert field of view in radians to focal length in pixels.
utils/graphics_utils.py:71
↓ 3 callersFunctionget_expon_lr_func
Return an exponential learning-rate schedule function. The returned helper starts near `lr_init`, ends at `lr_final`, and can
utils/general_utils.py:41
↓ 3 callersFunctionrender
Common render entry point used by training and evaluation code. It currently forwards directly to `render_3dgs`, which keeps call si
gaussian_renderer/__init__.py:238
↓ 3 callersFunctionrender_inference_image
Render one view in inference mode without keeping training gradients.
utils/evaluation/render_metric.py:71
↓ 3 callersMethodset_requires_grad
(self, state: bool)
lpipsPyTorch/modules/networks.py:46
↓ 3 callersMethodstep
(self, N: int, global_step: int)
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:527
↓ 3 callersFunctionstorePly
Write xyz and RGB arrays as a PLY point cloud with zero normals.
scene/dataset_readers.py:156
↓ 3 callersMethodtraining_setup
Initialize accumulators and optimizers for the current Gaussian tensors.
scene/gaussian_model_optimizer.py:196
↓ 2 callersFunctionRGB2SH
Convert RGB values in `[0, 1]` to the DC spherical-harmonic coefficient.
utils/general_utils.py:161
↓ 2 callersFunction_apply_trained_exposure
Apply the per-image exposure matrix learned during training. Exposure is stored by image name. The helper multiplies RGB by the
gaussian_renderer/__init__.py:23
↓ 2 callersFunction_build_forward_args
( means3D: torch.Tensor, dc: torch.Tensor, sh: torch.Tensor, opacities: torch.Tensor, scal
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:37
↓ 2 callersFunction_build_minigs_raster_settings
Build rasterizer settings shared by the MiniGS auxiliary and depth paths. The settings mirror the main renderer, but use the MiniGS
gaussian_renderer/__init__.py:99
↓ 2 callersFunction_resolve_dense_optimizer_type
Validate and return the configured optimizer type for dense Gaussian tensors.
scene/training_loop.py:22
↓ 2 callersFunction_rotate_vector_by_quaternion
Rotate vectors by normalized quaternions stored as `[w, x, y, z]`.
scene/methods/mcmc_ops.py:35
↓ 2 callersFunction_sample_alives
Sample alive Gaussian indices by opacity and count how often each one is used.
scene/methods/mcmc_ops.py:18
↓ 2 callersFunction_update_params
Build relocated child parameters from selected source Gaussians.
scene/methods/mcmc_ops.py:59
↓ 2 callersFunctionacquire_gpu_lock
Choose a candidate GPU and create its lock file atomically. The atomic file creation step is what prevents another process from
utils/batch_training/gpu.py:76
↓ 2 callersMethodadd_densification_stats_abs
Accumulate `abs(dx)` and `abs(dy)` written by CUDA backward for ImprovedGS densification scoring. When `sync_default_accumulator=True`, a
scene/gaussian_model_densification.py:225
↓ 2 callersFunctionappend_log_lines
Copy terminal log lines into a log file and add a timestamp to each non-empty line.
utils/experiment_utils.py:84
↓ 2 callersFunctionapply_gns_final_prune
Keep the top-opacity Gaussians until the exact final budget is reached.
scene/methods/pruning_methods.py:108
↓ 2 callersFunctionarchive_existing_output
When an output folder with the same name already exists, rename the old folder by its training timestamp to avoid overwriting results.
utils/experiment_utils.py:98
↓ 2 callersFunctionbeijing_now_iso
Create the Beijing-time timestamp used in experiment records.
utils/experiment_utils.py:16
↓ 2 callersFunctionbuild_python_command
Build the final Python command list passed to subprocess.
utils/batch_training/config.py:192
↓ 2 callersFunctionbuild_scene_paths
Resolve the input data folder and output folder for one scene run.
utils/batch_training/config.py:141
↓ 2 callersFunctioncameraList_from_camInfos
Build Camera objects for every CameraInfo entry in a scene split.
utils/camera_utils.py:82
↓ 2 callersMethodcreate_from_pcd
Initialize Gaussian tensors from an input point cloud. Positions and colors come from the point cloud, scales are estimated
scene/gaussian_model_io.py:29
↓ 2 callersMethoddensify_and_clone
Clone small high-gradient Gaussians using the original 3DGS rule. Unlike split, cloning keeps the original Gaussian and appe
scene/gaussian_model_densification.py:111
↓ 2 callersMethoddensify_and_split
Split large high-gradient Gaussians using the original 3DGS rule. Selected parents generate `N` children sampled from their
scene/gaussian_model_densification.py:73
↓ 2 callersMethodensure_sh_degree_capacity
Expand stored SH coefficients when the target degree needs more channels. Returns True only when the tensor was actually exp
scene/gaussian_model_sh.py:52
↓ 2 callersFunctionfinish_gns_pruning
Mark GNS pruning as complete and restore the opacity learning rate. The completion message is written both to tqdm and to the experi
scene/methods/pruning_methods.py:122
↓ 2 callersFunctiongetNerfppNorm
Compute scene translation and radius from training camera centers.
scene/dataset_readers.py:53
↓ 2 callersMethodgetTestCameras
Return test cameras for the requested resolution scale.
scene/__init__.py:176
↓ 2 callersFunctiongetWorld2View2
Build a world-to-view matrix and apply optional scene translation/scale.
utils/graphics_utils.py:25
↓ 2 callersMethodget_exposure_from_name
Return the learned or loaded exposure matrix for one image name.
scene/gaussian_model.py:147
↓ 2 callersMethodload_ply
Load raw Gaussian tensors from a saved PLY file. When train/test exposure is enabled, exposure matrices are loaded
scene/gaussian_model_io.py:108
↓ 2 callersFunctionnormalize_to_unit_range
Normalize a tensor into `[0, 1]` while handling NaN and infinite values.
scene/methods/densification_methods.py:20
↓ 2 callersMethodoneupSHdegree
Increase the active SH degree by one, expanding capacity first if needed.
scene/gaussian_model_sh.py:79
↓ 2 callersMethodonly_prune
Provide a lightweight pruning entry point that RAP and GNS can call without running densification.
scene/gaussian_model_densification.py:258
↓ 2 callersFunctionpsnr
Compute PSNR for each image in a batch.
utils/loss_utils.py:86
↓ 2 callersFunctionqvec2rotmat
Convert a COLMAP quaternion `[w, x, y, z]` into a rotation matrix.
scene/colmap_loader.py:43
↓ 2 callersFunctionreadCamerasFromTransforms
Read Blender/NeRF transform JSON and convert frames into CameraInfo records.
scene/dataset_readers.py:336
↓ 2 callersFunctionrebuild_result_total_csv
Recreate total CSV files from existing per-scene result summaries.
utils/batch_training/results.py:248
↓ 2 callersMethodreplace_tensors_to_optimizer
Rebind all optimizer parameter objects to current model tensors. When indices are provided, only those momentum entries are
scene/gaussian_model_optimizer.py:245
↓ 2 callersMethodreset_opacity
Clamp all activated opacities to a maximum value and update optimizer state.
scene/gaussian_model_densification.py:28
↓ 2 callersFunctionrun_command
Run one command with an optional CUDA_VISIBLE_DEVICES override.
utils/batch_training/processes.py:68
↓ 2 callersFunctionsafe_state
Set deterministic seeds, select CUDA device 0, and optionally timestamp stdout.
utils/general_utils.py:101
↓ 2 callersFunctionshould_select_best_repeat_by_psnr
Check whether repeat outputs should be collapsed to the best PSNR run.
utils/batch_training/results.py:103
↓ 2 callersFunctionssim
Compute structural similarity between two image batches.
utils/loss_utils.py:36
↓ 2 callersMethodupdate_opacity_lr
Let GNS raise the opacity learning rate during regularized pruning and restore it afterward.
scene/gaussian_model_optimizer.py:347
↓ 2 callersFunctionwrite_result_total_csv
Write rows for one split into the config-level total CSV file.
utils/batch_training/results.py:66
↓ 1 callersFunctionPILtoTorch
Resize a PIL image and convert it to a channel-first float tensor.
utils/general_utils.py:30
↓ 1 callersMethod__init__
(self, params: Any, lr: float, eps: float, max_steps: int)
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:522
↓ 1 callersFunction_apply_mcmc_covariance_to_noise
Shape isotropic noise by each Gaussian's local covariance.
scene/methods/mcmc_ops.py:44
↓ 1 callersFunction_apply_minigs_sh_schedule
Apply MiniGS's delayed SH-degree schedule. MiniGS keeps low SH capacity early, then restores the dataset target degree and r
scene/training_loop.py:105
↓ 1 callersFunction_build_3dgs_raster_settings
Build rasterizer settings for the main 3DGS render call. These settings choose image size, camera transforms, active SH degree,
gaussian_renderer/__init__.py:133
↓ 1 callersFunction_build_alpha_mask
Build the alpha mask used to ignore invalid image regions. RGBA images provide the base mask. In train/test exposure experiments,
scene/cameras.py:33
↓ 1 callersFunction_build_minigs_full_proj_transform
Build the projection matrix expected by the MiniGS reference kernels. MiniGS auxiliary and depth outputs depend on the exact project
gaussian_renderer/__init__.py:59
↓ 1 callersMethod_build_optimizer_param_groups
Build one optimizer parameter group per Gaussian tensor family. Separate groups keep learning rates independent for xyz, col
scene/gaussian_model_optimizer.py:66
↓ 1 callersFunction_build_training_background
Build the fixed white or black CUDA background tensor for training renders.
scene/training_runtime.py:69
↓ 1 callersFunction_collect_3dgs_densification_stats
Collect the statistics needed by the selected densification method. Standard 3DGS uses normal screen-space gradients, ABSGS/Improved
scene/methods/densification_stage.py:244
↓ 1 callersFunction_collect_minigs_densification_stats_3dgs
Update MiniGS gradient stats and return the blur mask for split decisions. The normal 3DGS gradient accumulator is updated first. A
scene/methods/densification_stage.py:153
↓ 1 callersFunction_compute_current_budget
Compute the active Gaussian budget for the current densification step. Early iterations use a square-root warmup toward the configur
scene/methods/densification_stage.py:127
↓ 1 callersFunction_compute_improvedgs_scores
Compute edge-aware Gaussian importance scores for ImprovedGS. It renders sampled training views with edge maps as pixel weights, the
scene/methods/densification_stage.py:69
↓ 1 callersFunction_compute_reconstruction_loss
Combine pixel L1 loss and fused SSIM into the standard 3DGS image loss.
scene/methods/optimization_methods.py:44
↓ 1 callersMethod_configure_optimizers
Recreate optimizers and learning-rate schedules from current tensors. This is called both at initial setup and after methods
scene/gaussian_model_optimizer.py:84
↓ 1 callersFunction_enable_debug_if_needed
Enable renderer debug mode one iteration before the requested debug step. The rasterizer uses the flag during the next render call,
scene/methods/optimization_methods.py:22
↓ 1 callersFunction_ensure_gaussian_adam_state
( state: dict[str, torch.Tensor], N: int, param: torch.Tensor, max_steps: int, bootstrap_s
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:466
↓ 1 callersFunction_ensure_last_seen_state
( state: dict[str, torch.Tensor], N: int, param: torch.Tensor, bootstrap_step: int, )
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:449
↓ 1 callersFunction_ensure_lr_history_capacity
(state: dict[str, torch.Tensor], required_size: int, device: torch.device)
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:494
↓ 1 callersFunction_ensure_model_path
Fill `args.model_path` with a unique output folder when it is empty.
scene/training_runtime.py:56
next →1–100 of 319, ranked by callers