MCPcopy Create free account

hub / github.com/JasonLSC/GSCodec_Studio / functions

Functions1,225 in github.com/JasonLSC/GSCodec_Studio

Methodforward
(ctx, input, bitdepth = 8)
gsplat/compression_simulation/simulation.py:469
Methodforward
(ctx, input, Q, input_mean=None)
gsplat/compression_simulation/simulation.py:490
Methodforward
(ctx, input)
gsplat/compression_simulation/simulation.py:500
Methodforward
(self, x, Q=None, **kwargs)
gsplat/compression_simulation/entropy_model.py:72
Methodforward
(self, x, Q=None, **kwargs)
gsplat/compression_simulation/entropy_model.py:154
Methodforward
(self, x, Q=None, **kwargs)
gsplat/compression_simulation/entropy_model.py:201
Methodforward
(self, x, Q=None, pos=None)
gsplat/compression_simulation/entropy_model.py:323
Methodforward
(ctx, x, bound)
gsplat/compression_simulation/entropy_model.py:360
Methodforward
(self, x)
gsplat/compression_simulation/entropy_model.py:387
Methodforward
( ctx, quats: Tensor, # [N, 4], scales: Tensor, # [N, 3], compute_covar: boo
gsplat/cuda/_wrapper.py:650
Methodforward
( ctx, means: Tensor, # [C, N, 3] covars: Tensor, # [C, N, 3, 3] Ks: Tensor,
gsplat/cuda/_wrapper.py:691
Methodforward
( ctx, means: Tensor, # [N, 3] covars: Tensor, # [N, 3, 3] viewmats: Tensor,
gsplat/cuda/_wrapper.py:741
Methodforward
( ctx, means: Tensor, # [N, 3] covars: Tensor, # [N, 6] or None quats: Tenso
gsplat/cuda/_wrapper.py:779
Methodforward
( ctx, means2d: Tensor, # [C, N, 2] conics: Tensor, # [C, N, 3] colors: Tens
gsplat/cuda/_wrapper.py:905
Methodforward
( ctx, means: Tensor, # [N, 3] covars: Tensor, # [N, 6] or None quats: Tenso
gsplat/cuda/_wrapper.py:1035
Methodforward
( ctx, sh_degree: int, dirs: Tensor, coeffs: Tensor, masks: Tensor )
gsplat/cuda/_wrapper.py:1230
Methodforward
( ctx, means: Tensor, quats: Tensor, scales: Tensor, viewmats: Tensor,
gsplat/cuda/_wrapper.py:1368
Methodforward
( ctx, means: Tensor, # [N, 3] quats: Tensor, # [N, 4] scales: Tensor, # [N
gsplat/cuda/_wrapper.py:1475
Methodforward
( ctx, means2d: Tensor, ray_transforms: Tensor, colors: Tensor, opacit
gsplat/cuda/_wrapper.py:1832
Methodforward
Bilateral grid slicing. Supports 2-D, 3-D, 4-D, and 5-D input. For the 2-D, 3-D, and 4-D cases, please refer to `slice`. For the 5-D c
examples/lib_bilagrid.py:310
Methodforward
(self, x)
examples/lib_bilagrid.py:397
Methodforward
Low-rank 4D bilateral grid slicing. Args: xyz (torch.Tensor): The xyz coordinates with shape $(..., 3)$. rgb (torch.T
examples/lib_bilagrid.py:543
Methodforward
Adjust camera pose based on deltas. Args: camtoworlds: (..., 4, 4) embed_ids: (...,) Returns: up
examples/utils.py:30
Methodforward
Adjust appearance based on embeddings. Args: features: (N, feature_dim) embed_ids: (C,) dirs: (C, N, 3)
examples/utils.py:80
Methodforward
(self, input, rays, time=None)
examples/helper/STG/helper_model.py:16
Methodframe_num
(self)
gsplat/compression/post_training/io.py:21
Methodframe_read_fn
(idx_frame)
examples/datasets/INVR.py:87
Functiongenerate_ellipse_path_y
Generate an elliptical render path based on the given poses.
examples/datasets/traj.py:144
Functiongeom_transform_points
(points, transf_matrix)
examples/helper/STG/graphics_utils.py:23
FunctiongetWorld2View
(R, t)
examples/helper/STG/graphics_utils.py:32
Methodget_binary_mask
(self)
gsplat/compression_simulation/ada_mask.py:42
Methodget_binary_mask
(self)
gsplat/compression_simulation/mask.py:160
Methodget_binary_mask
(self)
gsplat/compression_simulation/mask.py:228
Methodget_c2w
(time, v4_w2c, v5_w2c)
examples/simple_trainer_dyngs.py:1049
Methodget_c2w
(time, v4_w2c, v5_w2c)
examples/simple_trainer_STG.py:1128
Methodget_entropy_models
(self)
gsplat/compression_simulation/runtime.py:198
Methodget_entropy_models
(self)
gsplat/compression_simulation/runtime.py:278
Functionget_expon_lr_func
Copied from Plenoxels Continuous learning rate decay function. Adapted from JaxNeRF The returned rate is lr_init when step=0 and lr_fina
examples/helper/STG/general_utils.py:90
Methodget_likelihood
(self, x, Q=None, **kwargs)
gsplat/compression_simulation/entropy_model.py:256
Methodget_mask_binary
(self)
gsplat/compression_simulation/runtime.py:195
Methodget_mask_binary
(self)
gsplat/compression_simulation/runtime.py:275
Methodget_profiler
(self, tb_writer)
examples/ply_loader_renderer.py:554
Functionhelper
(step)
examples/helper/STG/general_utils.py:108
Methodhook
(grad: Tensor)
gsplat/compression_simulation/mask.py:187
Methodinitialize_state
Initialize and return the running state for this strategy. The returned state should be passed to the `step_pre_backward()` and `step
gsplat/strategy/modified_stg.py:36
Methodinitialize_state
Initialize and return the running state for this strategy. The returned state should be passed to the `step_pre_backward()` and `step
gsplat/strategy/STG_Strategy.py:35
Methodinitialize_state
Initialize and return the running state for this strategy. The returned state should be passed to the `step_pre_backward()` and `step
gsplat/strategy/default.py:96
Functioninterpolate_camera_poses
(qvec0, tvec0, qvec1, tvec1, t)
examples/helper/STG/posetrace_utils.py:28
Functioninverse_log_transform
(y)
gsplat/compression_simulation/ops.py:11
Functioninverse_sigmoid
(x)
examples/helper/STG/general_utils.py:23
Methodinverse_sigmoid
(x)
examples/simple_trainer_dyngs.py:1233
Functioninverse_sigmoida
(x)
examples/helper/STG/general_utils.py:40
Functioninverse_sigmoidc
(x)
examples/helper/STG/general_utils.py:46
Functioninverse_sigmoidk
(x, k)
examples/helper/STG/general_utils.py:34
Functioninverse_sigmoidv2
(x)
examples/helper/STG/general_utils.py:26
Functioninverse_sigmoidv3
(x)
examples/helper/STG/general_utils.py:50
FunctionloadCamv2timing
(args, id, cam_info, resolution_scale)
examples/helper/STG/camera_utils.py:126
Methodload_entropy_model_from_ckpt
(self, ckpt: Dict, entropy_model_type: str)
examples/ply_loader_renderer.py:857
Methodload_entropy_model_from_ckpt
(self, ckpt: Dict, entropy_model_type: str)
examples/simple_trainer.py:1412
Functionload_image_sequence_to_tensors
Load image sequence and convert to tensor sequence Args: file_pattern: Complete file path pattern, e.g. "results/mpeg151/video_a
examples/helper/mpeg_gsc/gsc_metric.py:291
Functionload_ply
(path: str)
examples/simple_trainer_old.py:654
Functionload_ply_sequence
(pattern: str, frame_num: int = 1)
gsplat/compression/post_training/io.py:115
Methodload_state_dict
(self, state: MutableMapping[str, Any])
gsplat/compression_simulation/runtime.py:185
Methodload_state_dict
(self, state: MutableMapping[str, Any])
gsplat/compression_simulation/runtime.py:262
Methodload_state_dict
(self, state: Mapping[str, Any])
gsplat/compression_simulation/mask.py:149
Methodload_state_dict
(self, state: Mapping[str, Any])
gsplat/compression_simulation/mask.py:221
Methodload_state_dict
(self, state: Dict[str, Any])
gsplat/compression_simulation/entropy.py:172
Functionlog_transform
(x)
gsplat/compression_simulation/ops.py:8
Functionmain
(local_rank: int, world_rank, world_size: int, cfg: Config)
examples/compress_ply_sequence.py:951
Functionmain
(local_rank: int, world_rank, world_size: int, cfg: Config)
examples/codec_ply_sequence.py:1108
Functionmain
(local_rank: int, world_rank, world_size: int, cfg: Config)
examples/simple_trainer_old.py:1725
Functionmain
(local_rank: int, world_rank, world_size: int, cfg: Config)
examples/ply_loader_renderer.py:910
Functionmain
(local_rank: int, world_rank, world_size: int, args)
examples/simple_viewer_dyn.py:244
Functionmain
( height: int = 256, width: int = 256, num_points: int = 100000, save_imgs: bool = True, i
examples/image_fitting.py:161
Functionmain
(local_rank: int, world_rank, world_size: int, args)
examples/simple_viewer.py:27
Functionmain
(local_rank: int, world_rank, world_size: int, cfg: Config)
examples/simple_trainer.py:1465
Functionmain
(results_dir: str, rps: List[str])
examples/benchmarks/mpeg/zip_and_summarize_stats.py:11
Functionmain
(results_dir: str, scenes: List[str], num_frame: int)
examples/benchmarks/stg/summarize_GOP50_stats.py:13
Functionmain
(results_dir: str, scenes: List[str])
examples/benchmarks/stg/summarize_stats.py:71
Functionmain
( exp_dir: Annotated[str, "Path to the experiment directory, e.g. results/mpeg151/video_anchor/bartender"]
examples/benchmarks/mpeg152/rate_distortion_stats_to_csv.py:7
Functionmain
(results_dir: str, scenes: List[str])
examples/benchmarks/compression/summarize_stats.py:11
Methodmaybe_update
(self, step: int, splats: Mapping[str, Tensor])
gsplat/compression_simulation/mask.py:49
Methodmaybe_update
(self, step: int, splats: Mapping[str, Tensor])
gsplat/compression_simulation/mask.py:108
Methodmaybe_update
(self, step: int, splats: Mapping[str, Tensor])
gsplat/compression_simulation/mask.py:204
Methodmaybe_update
(self, step: int, splats: Dict[str, Tensor])
gsplat/compression_simulation/entropy.py:75
Functionnormalize
(camtoworlds, points=None)
examples/datasets/normalize.py:133
Methodoptimizer_fn
(key: str, v: Tensor)
gsplat/strategy/modified_stg.py:338
Methodoptimizer_fn
(key: str, v: Tensor)
gsplat/strategy/STG_Strategy.py:353
Functionpair_id_to_image_ids
(pair_id)
mpeg_gsc_utils/pre_colmap.py:88
Methodparam_fn
(name: str, p: Tensor)
gsplat/strategy/modified_stg.py:333
Methodparam_fn
(name: str, p: Tensor)
gsplat/strategy/STG_Strategy.py:348
Methodparam_inverse_transform
Inverse transform the splats parameters to original domain Args: splats_list (List[Dict]): splats parameters of all frame
gsplat/compression/seq_hevc_compression.py:230
Methodparam_inverse_transform
Inverse transform the splats parameters to original domain Args: splats_list (List[Dict]): splats parameters of all frame
gsplat/compression/yuv_compression.py:257
Methodparam_transform
Transform the splats parameters to new domain e.g. for SH coefficients, transform them from RGB domain to YCbCr domain for q
gsplat/compression/seq_hevc_compression.py:144
Methodparam_transform
Transform the splats parameters to new domain e.g. for SH coefficients, transform them from RGB domain to YCbCr domain for q
gsplat/compression/yuv_compression.py:158
Functionpath_serializer
(obj: Any, _serialize: Callable)
cfg_sys/config_manager.py:222
Functionpersp_proj
Perspective projection on Gaussians. DEPRECATED: please use `proj` with `ortho=False` instead. Args: means: Gaussian means. [C, N, 3]
gsplat/cuda/_wrapper.py:110
Methodplot_distribution
(value, param_name, save_dir)
examples/simple_trainer_old.py:1622
Methodplot_distribution
(value, param_name, save_dir)
examples/simple_trainer.py:1362
Methodquantize
Quantize input splats and return quantized tensors with context.
gsplat/compression/post_training/components/quant/base.py:12
← previousnext →1,001–1,100 of 1,225, ranked by callers