MCPcopy Create free account

hub / github.com/DSL-Lab/StreamSplat / functions

Functions474 in github.com/DSL-Lab/StreamSplat

↓ 1 callersMethod_transform_inputs
Transform inputs for decoder. Args: inputs (list[Tensor]): List of multi-level img features. Returns: Tensor:
encoders/dinov2/hub/depth/decode_heads.py:237
↓ 1 callersMethodactivate_mean
(self, means, mean_activation='tanh')
model/mixture_model_utils.py:20
↓ 1 callersFunctionadd_prefix
Add prefix for dict. Args: inputs (dict): The input dict with str keys. prefix (str): The prefix to add. Returns: d
encoders/dinov2/hub/depth/encoder_decoder.py:15
↓ 1 callersFunctionadd_residual
(x, brange, residual, residual_scale_factor, scaling_vector=None)
model/depth_anything/depth_anything_v2/dinov2_layers/block.py:142
↓ 1 callersFunctionadd_residual
(x, brange, residual, residual_scale_factor, scaling_vector=None)
encoders/dinov2/layers/block.py:180
↓ 1 callersMethodattention
(self, x, index)
model/transformer_utils.py:272
↓ 1 callersMethodaug_test
Test with augmentations. Only rescale=True is supported.
encoders/dinov2/hub/depth/encoder_decoder.py:200
↓ 1 callersFunctionbuild_model
(args, only_teacher=False, img_size=224)
encoders/dinov2/models/__init__.py:14
↓ 1 callersFunctionbuild_rotation
(r)
utils/general_utils.py:81
↓ 1 callersMethodconvert_images
(self, images)
datasets/provider_re10k_map.py:171
↓ 1 callersFunctioncreate_window
(window_size, channel)
utils/loss_utils.py:28
↓ 1 callersFunctiondrop_path
(x, drop_prob: float = 0.0, training: bool = False)
model/transformer_utils.py:35
↓ 1 callersFunctiondrop_path
(x, drop_prob: float = 0.0, training: bool = False)
model/depth_anything/depth_anything_v2/dinov2_layers/drop_path.py:15
↓ 1 callersFunctiondrop_path
(x, drop_prob: float = 0.0, training: bool = False)
encoders/dinov2/layers/drop_path.py:14
↓ 1 callersMethodforward
(self, x, timestamp=None)
model/model_utils.py:69
↓ 1 callersMethodforward
(self, x)
model/depth_anything/depth_anything_v2/dpt.py:176
↓ 1 callersMethodforward
(self, x: Tensor)
model/depth_anything/depth_anything_v2/dinov2_layers/block.py:82
↓ 1 callersMethodforward
(self, x: Tensor)
model/depth_anything/depth_anything_v2/dinov2_layers/attention.py:49
↓ 1 callersMethodforward
(self, x: Tensor)
encoders/dinov2/layers/block.py:95
↓ 1 callersMethodforward
(self, x: Tensor)
encoders/dinov2/layers/attention.py:56
↓ 1 callersMethodforward_condition
(self, frames, depths)
model/model_utils.py:597
↓ 1 callersMethodforward_encoder
(self, frames, depths, cond_times=None)
model/model_utils.py:491
↓ 1 callersMethodforward_encoder
(self, frames, depths, cond_times=None)
model/model_utils.py:583
↓ 1 callersMethodforward_features
(self, x, masks=None)
model/depth_anything/depth_anything_v2/dinov2.py:253
↓ 1 callersMethodforward_features_list
(self, x_list, masks_list)
model/depth_anything/depth_anything_v2/dinov2.py:233
↓ 1 callersMethodforward_features_list
(self, x_list, masks_list)
encoders/dinov2/models/vision_transformer.py:260
↓ 1 callersMethodforward_gaussians
(self, frames, depths, cond_times=None)
model/encoder_model.py:62
↓ 1 callersMethodforward_gaussians
(self, frames, depths, cond_times=None)
model/splat_model_inference.py:31
↓ 1 callersMethodforward_gaussians
(self, frames, depths, cond_times=None)
model/splat_model.py:63
↓ 1 callersMethodforward_nested
x_list contains a list of tensors to nest together and run
model/depth_anything/depth_anything_v2/dinov2_layers/block.py:205
↓ 1 callersMethodforward_nested
x_list contains a list of tensors to nest together and run
encoders/dinov2/layers/block.py:248
↓ 1 callersFunctiongaussian
(window_size, sigma)
utils/loss_utils.py:24
↓ 1 callersFunctiongetOrthProjectionMatrix
()
gaussian_renderer_dynamic/__init__.py:36
↓ 1 callersFunctiongetWorld2View2
(R, t, translate=np.array([.0, .0, .0]), scale=1.0)
gaussian_renderer_dynamic/__init__.py:21
↓ 1 callersFunctionget_branges_scales
(x, sample_drop_ratio=0.0)
model/depth_anything/depth_anything_v2/dinov2_layers/block.py:134
↓ 1 callersFunctionget_branges_scales
(x, sample_drop_ratio=0.0)
encoders/dinov2/layers/block.py:172
↓ 1 callersFunctionget_depth
Load and resize a depth map.
splat_inference.py:42
↓ 1 callersFunctionget_image
Load and resize an image.
splat_inference.py:35
↓ 1 callersMethodget_intermediate_layers
( self, x: torch.Tensor, n: Union[int, Sequence] = 1, # Layers or n last layers to ta
model/depth_anything/depth_anything_v2/dinov2.py:297
↓ 1 callersMethodget_intermediate_layers
( self, x: torch.Tensor, n: Union[int, Sequence] = 1, # Layers or n last layers to ta
encoders/dinov2/models/vision_transformer.py:331
↓ 1 callersFunctionget_lpips
(device: torch.device)
utils/metrics.py:23
↓ 1 callersMethodget_lr
(self)
utils/general_utils.py:219
↓ 1 callersMethodget_mix_params
(self, logits, means, log_scales)
model/mixture_model_utils.py:47
↓ 1 callersMethodget_predicted_depth_path
(self, image_path)
preprocess_depth_davis.py:96
↓ 1 callersMethodget_size
(self, width, height)
model/depth_anything/depth_anything_v2/util/transform.py:62
↓ 1 callersMethodget_size
(self, width, height)
datasets/transform_utils.py:62
↓ 1 callersMethodicdf_fn
Inverse cumulative distribution function of the Gaussian distribution.
model/mixture_model_utils.py:71
↓ 1 callersMethodimage2tensor
(self, raw_image, input_size=518)
model/depth_anything/depth_anything_v2/dpt.py:196
↓ 1 callersMethodinit_weights
(self)
model/depth_anything/depth_anything_v2/dinov2.py:172
↓ 1 callersMethodinit_weights
(self)
encoders/dinov2/hub/depth/decode_heads.py:459
↓ 1 callersMethodinit_weights
(self)
encoders/dinov2/models/vision_transformer.py:199
↓ 1 callersMethodinterpolate_pos_encoding
(self, x, w, h)
model/depth_anything/depth_anything_v2/dinov2.py:179
↓ 1 callersMethodinterpolate_pos_encoding
(self, x, w, h)
encoders/dinov2/models/vision_transformer.py:206
↓ 1 callersMethodkey_activation
(self, v: torch.Tensor, key='')
model/model_utils.py:234
↓ 1 callersMethodkey_activation
Apply activation functions for dynamic keys.
model/model_utils.py:434
↓ 1 callersFunctionloadCam
(args, id, cam_info, resolution_scale)
utils/camera_utils.py:19
↓ 1 callersFunctionload_model_weights
Loads weights from the trained decoder checkpoint.
splat_inference.py:54
↓ 1 callersMethodload_state_dict
(self, state_dict, strict=True)
model/encoder_model.py:51
↓ 1 callersMethodlog_images
(self, img_path, depth_pred, depth_gt, img_meta)
encoders/dinov2/hub/depth/decode_heads.py:197
↓ 1 callersMethodlog_pdf_fn
Log probability density function of the Gaussian distribution.
model/mixture_model_utils.py:59
↓ 1 callersMethodlosses
Compute depth loss.
encoders/dinov2/hub/depth/decode_heads.py:180
↓ 1 callersFunctionmain
(opt: AllConfigs, args: argparse.Namespace)
splat_inference.py:72
↓ 1 callersFunctionmain
()
train.py:27
↓ 1 callersFunctionmain
()
train_decoder.py:25
↓ 1 callersFunctionnamed_apply
(fn: Callable, module: nn.Module, name="", depth_first=True, include_root=False)
model/depth_anything/depth_anything_v2/dinov2.py:26
↓ 1 callersFunctionnamed_apply
(fn: Callable, module: nn.Module, name="", depth_first=True, include_root=False)
encoders/dinov2/models/vision_transformer.py:35
↓ 1 callersFunctionpredict_depth
(image)
model/depth_anything/app.py:43
↓ 1 callersFunctionpreprocess_data
(frames, depths, timestamps, device)
splat_inference.py:18
↓ 1 callersFunctionrasterize_gaussians
( means3D, means2D, sh, colors_precomp, flow_2d, opacities, ts, scales, sc
gaussian_renderer_dynamic/diff_gaussian_rasterization.py:34
↓ 1 callersFunctionrasterize_gaussians
( means3D, means2D, sh, colors_precomp, opacities, scales, rotations, cov3Ds_p
submodules/diff-gaussian-rasterization-orth/diff_gaussian_rasterization_kiui_orth/__init__.py:21
↓ 1 callersFunctionread
()
gaussian_renderer_dynamic/network_gui.py:43
↓ 1 callersFunctionresize_to_multiple_of_14
Resize a PIL image so that its height and width are the closest multiples of 14. The resizing is done with bilinear interpolation.
preprocess_depth_davis.py:20
↓ 1 callersFunctionresize_to_shorter_side
Resize the input PIL image so that its shorter side equals target_size (default 518) while keeping the aspect ratio.
preprocess_depth_davis.py:31
↓ 1 callersFunctionrotate2d
(theta, **kwargs)
datasets/augmentv2.py:114
↓ 1 callersFunctionrotate3d
(v, theta, **kwargs)
datasets/augmentv2.py:121
↓ 1 callersFunctionrun_inference
(opt: Options)
preprocess_depth_davis.py:141
↓ 1 callersFunctionscale3d
(sx, sy, sz, **kwargs)
datasets/augmentv2.py:106
↓ 1 callersMethodset_transform
(self)
preprocess_depth_davis.py:91
↓ 1 callersMethodset_transform
(self)
datasets/provider_vos.py:71
↓ 1 callersMethodset_transform
(self)
datasets/provider_davis.py:71
↓ 1 callersMethodset_transform
(self)
datasets/provider_co3d.py:75
↓ 1 callersMethodset_transform
(self)
datasets/provider_combined.py:137
↓ 1 callersMethodset_transform
(self)
datasets/provider_re10k_map.py:83
↓ 1 callersMethodsimple_test
Simple test with single image.
encoders/dinov2/hub/depth/encoder_decoder.py:188
↓ 1 callersMethodslide_inference
Inference by sliding-window with overlap. If h_crop > h_img or w_crop > w_img, the small patch will be used to decode without padding
encoders/dinov2/hub/depth/encoder_decoder.py:119
↓ 1 callersFunctionstrip_lowerdiag
(L)
utils/general_utils.py:67
↓ 1 callersMethodtrain
(self, mode=True)
model/encoder_model.py:56
↓ 1 callersMethodtrain
(self, mode=True)
model/splat_model.py:43
↓ 1 callersFunctiontransformPoint4x3
submodules/diff-gaussian-rasterization-orth/cuda_rasterizer/auxiliary.h:58
↓ 1 callersFunctiontransformPoint4x4
submodules/diff-gaussian-rasterization-orth/cuda_rasterizer/auxiliary.h:68
↓ 1 callersFunctiontranslate3d
(tx, ty, tz, **kwargs)
datasets/augmentv2.py:91
↓ 1 callersMethodwhole_inference
Inference with full image.
encoders/dinov2/hub/depth/encoder_decoder.py:115
↓ 1 callersMethodwindow_attention
(self, x, attn_layer, index)
model/transformer_utils.py:321
FunctionPYBIND11_MODULE
submodules/diff-gaussian-rasterization-orth/ext.cpp:15
FunctionRGB2SH
(rgb)
utils/sh_utils.py:225
FunctionSH2RGB
(sh)
utils/sh_utils.py:228
Method__call__
(self, sample)
model/depth_anything/depth_anything_v2/util/transform.py:109
Method__call__
(self, sample)
model/depth_anything/depth_anything_v2/util/transform.py:133
Method__call__
(self, sample)
model/depth_anything/depth_anything_v2/util/transform.py:146
Method__call__
(self, sample)
datasets/transform_utils.py:109
← previousnext →101–200 of 474, ranked by callers