MCPcopy Create free account

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

Functions474 in github.com/DSL-Lab/StreamSplat

Methodforward
(self, x_or_x_list)
encoders/dinov2/layers/block.py:288
Methodforward
(self, x)
encoders/dinov2/layers/dino_head.py:36
Methodforward
(self, x: Tensor)
encoders/dinov2/layers/layer_scale.py:26
Methodforward
(self, x: Tensor, attn_bias=None)
encoders/dinov2/layers/attention.py:73
Methodforward
(self, x: Tensor)
encoders/dinov2/layers/mlp.py:34
Methodforward
(self, x: Tensor)
encoders/dinov2/layers/swiglu_ffn.py:30
Methodforward
(self, x: Tensor)
encoders/dinov2/layers/patch_embed.py:68
Methodforward
(self, x)
encoders/dinov2/models/vision_transformer.py:47
Methodforward
(self, *args, is_training=False, **kwargs)
encoders/dinov2/models/vision_transformer.py:357
Methodforward
( ctx, means3D, means2D, sh, colors_precomp, flow_2d,
gaussian_renderer_dynamic/diff_gaussian_rasterization.py:67
Methodforward
(self, means3D, means2D, opacities, shs = None, colors_precomp = None, flow_2d = None, ts=None,
gaussian_renderer_dynamic/diff_gaussian_rasterization.py:254
Methodforward
( ctx, means3D, means2D, sh, colors_precomp, opacities,
submodules/diff-gaussian-rasterization-orth/diff_gaussian_rasterization_kiui_orth/__init__.py:46
Methodforward
(self, means3D, means2D, opacities, shs = None, colors_precomp = None, scales = None, rotations = None, cov3D_
submodules/diff-gaussian-rasterization-orth/diff_gaussian_rasterization_kiui_orth/__init__.py:190
Methodforward_dummy
Dummy forward function.
encoders/dinov2/hub/depth/encoder_decoder.py:81
Methodforward_test
Forward function for testing. Args: inputs (list[Tensor]): List of multi-level img features. img_metas (list[dict]): L
encoders/dinov2/hub/depth/decode_heads.py:134
Methodforward_train
Forward function for training. Args: inputs (list[Tensor]): List of multi-level img features. img_metas (list[dict]):
encoders/dinov2/hub/depth/decode_heads.py:112
Functionfps
(x, k)
utils/general_utils.py:153
Functiongeom_transform_points
(points, transf_matrix)
utils/graphics_utils.py:23
FunctiongetOrthProjectionMatrix
(znear, zfar, fovX, fovY)
utils/graphics_utils.py:74
FunctiongetProjectionMatrix
(znear, zfar, fovX, fovY)
utils/graphics_utils.py:52
FunctiongetProjectionMatrixCenterShift
(znear, zfar, cx, cy, fl_x, fl_y, w, h)
utils/graphics_utils.py:100
FunctiongetRect
submodules/diff-gaussian-rasterization-orth/cuda_rasterizer/auxiliary.h:46
FunctiongetWorld2View
(R, t)
utils/graphics_utils.py:32
FunctiongetWorld2View2
(R, t, translate=np.array([.0, .0, .0]), scale=1.0)
utils/graphics_utils.py:39
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
utils/general_utils.py:32
Methodget_lr
(self)
utils/general_utils.py:173
Functionhelper
(step)
utils/general_utils.py:50
Functionin_frustum
submodules/diff-gaussian-rasterization-orth/cuda_rasterizer/auxiliary.h:139
Functioninit
(wish_host, wish_port)
gaussian_renderer_dynamic/network_gui.py:26
Functioninit_weights_vit_timm
ViT weight initialization, original timm impl (for reproducibility)
model/depth_anything/depth_anything_v2/dinov2.py:331
Functioninit_weights_vit_timm
ViT weight initialization, original timm impl (for reproducibility)
encoders/dinov2/models/vision_transformer.py:365
Functioninverse_sigmoid
(x)
utils/general_utils.py:21
Functionknn
(x, src, k, transpose=False)
utils/general_utils.py:137
Functionl1_loss
(network_output, gt)
utils/loss_utils.py:18
Functionl2_loss
(network_output, gt)
utils/loss_utils.py:21
Methodload_state_dict
(self, state_dict, strict=True)
model/splat_model_inference.py:24
MethodmarkVisible
(self, positions)
gaussian_renderer_dynamic/diff_gaussian_rasterization.py:243
MethodmarkVisible
(self, positions)
submodules/diff-gaussian-rasterization-orth/diff_gaussian_rasterization_kiui_orth/__init__.py:179
Functionmkdir_p
(folder_path)
utils/system_utils.py:16
Functionmse
(img1, img2)
utils/image_utils.py:14
Functionmsssim
(rgb, gts)
utils/loss_utils.py:68
Functionndc2Pix
submodules/diff-gaussian-rasterization-orth/cuda_rasterizer/auxiliary.h:41
Functionobtain
submodules/diff-gaussian-rasterization-orth/cuda_rasterizer/rasterizer_impl.h:23
Functionon_submit
(image)
model/depth_anything/app.py:60
Functionpsnr
(img1, img2)
utils/image_utils.py:17
Functionreceive
()
gaussian_renderer_dynamic/network_gui.py:57
Functionrender
(gaussians: dict, bg_color: torch.Tensor, timestamps: torch.Tensor = None, scaling_modifier=1.0, o
gaussian_renderer_dynamic/__init__.py:64
Methodreorder
(v)
model/model_utils.py:304
Functionrequired
submodules/diff-gaussian-rasterization-orth/cuda_rasterizer/rasterizer_impl.h:68
Functionsafe_state
(silent)
utils/general_utils.py:114
FunctionsearchForMaxIteration
(folder)
utils/system_utils.py:26
Functionsend
(message_bytes, verify)
gaussian_renderer_dynamic/network_gui.py:50
Methodset_grad_checkpointing
(self, set_checkpointing=True)
model/transformer_utils.py:412
Functionsigmoid
submodules/diff-gaussian-rasterization-orth/cuda_rasterizer/auxiliary.h:134
Functionssim
(img1, img2, window_size=11, size_average=True)
utils/loss_utils.py:34
Functionssimse_loss
Scale and translation invariant MSE loss Args: pred_depth: predicted depth [B, 1, H, W] target_depth: ground truth depth [B,
model/midas_loss.py:99
Functionssitrim_loss
pred_depth shape: [B, V, 1, H, W] target_depth shape: [B, V, 1, H, W] impl according to MiDaS paper Eq.6 and Splatter-a-Video paper Eq.4
model/midas_loss.py:4
Methodstate_dict
(self, **kwargs)
model/model_utils.py:554
Methodstep
(self, epoch=None)
utils/general_utils.py:233
Functionstrip_symmetric
(sym)
utils/general_utils.py:78
Methodtrain
(self, mode=True)
model/model_utils.py:562
Methodtrain_step
The iteration step during training. This method defines an iteration step during training, except for the back propagation and optimi
encoders/dinov2/hub/depth/encoder_decoder.py:264
FunctiontransformVec4x3
submodules/diff-gaussian-rasterization-orth/cuda_rasterizer/auxiliary.h:79
FunctiontransformVec4x3Transpose
submodules/diff-gaussian-rasterization-orth/cuda_rasterizer/auxiliary.h:89
Functiontry_connect
()
gaussian_renderer_dynamic/network_gui.py:34
Methodval_step
The iteration step during validation. This method shares the same signature as :func:`train_step`, but used during val epochs. Note t
encoders/dinov2/hub/depth/encoder_decoder.py:307
Functionvit_base
(patch_size=16, num_register_tokens=0, **kwargs)
model/depth_anything/depth_anything_v2/dinov2.py:353
Functionvit_base
(patch_size=16, num_register_tokens=0, **kwargs)
encoders/dinov2/models/vision_transformer.py:399
Functionvit_giant2
Close to ViT-giant, with embed-dim 1536 and 24 heads => embed-dim per head 64
model/depth_anything/depth_anything_v2/dinov2.py:381
Functionvit_giant2
Close to ViT-giant, with embed-dim 1536 and 24 heads => embed-dim per head 64
encoders/dinov2/models/vision_transformer.py:427
Functionvit_large
(patch_size=16, num_register_tokens=0, **kwargs)
model/depth_anything/depth_anything_v2/dinov2.py:367
Functionvit_large
(patch_size=16, num_register_tokens=0, **kwargs)
encoders/dinov2/models/vision_transformer.py:413
Functionvit_small
(patch_size=16, num_register_tokens=0, **kwargs)
model/depth_anything/depth_anything_v2/dinov2.py:339
Functionvit_small
(patch_size=16, num_register_tokens=0, **kwargs)
encoders/dinov2/models/vision_transformer.py:385
← previous401–474 of 474, ranked by callers