MCPcopy Create free account

hub / github.com/Robbyant/lingbot-depth / functions

Functions198 in github.com/Robbyant/lingbot-depth

↓ 12 callersMethoddtype
(self)
mdm/model/v2.py:64
↓ 9 callersFunction_make_dinov2_model
( *, arch_name: str = "vit_large", img_size: int = 518, patch_size: int = 14, init_values:
mdm/model/dinov2_rgbd/hub/backbones.py:18
↓ 8 callersMethod__init__
(self, dims: Sequence[int])
mdm/model/modules_decoder.py:116
↓ 6 callersMethodtime
(self)
mdm/utils/tools.py:184
↓ 3 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, bias=True, drop=0.)
mdm/model/dinov2_rgbd/layers/patch_embed_mlp.py:63
↓ 3 callersFunction_run
(command)
mdm/model/dinov2_rgbd/utils/utils.py:49
↓ 3 callersFunctionget_cluster_type
(cluster_type: Optional[ClusterType] = None)
mdm/model/dinov2_rgbd/utils/cluster.py:31
↓ 3 callersFunctionwrap_module_with_gradient_checkpointing
(module: nn.Module)
mdm/model/utils.py:7
↓ 2 callersFunction_make_dinov2_model_name
(arch_name: str, patch_size: int, num_register_tokens: int = 0)
mdm/model/dinov2_rgbd/hub/utils.py:17
↓ 2 callersFunctiondepth_to_color_opencv
Convert depth map to color visualization using OpenCV colormap. Args: depth_map (np.ndarray): Depth map (H, W) vmin (float):
example.py:119
↓ 2 callersFunctiondepth_to_pointcloud
Convert depth map to point cloud (pure Tensor version, no point filtering) Args: depth: torch.Tensor, shape (H, W) or (B, H, W), dep
mdm/utils/geo.py:17
↓ 2 callersMethoddevice
(self)
mdm/model/v2.py:60
↓ 2 callersFunctiondrop_add_residual_stochastic_depth
( x: Tensor, residual_func: Callable[[Tensor], Tensor], sample_drop_ratio: float = 0.0, )
mdm/model/dinov2_rgbd/layers/block.py:116
↓ 2 callersFunctiondrop_add_residual_stochastic_depth_list
( x_list: List[Tensor], residual_func: Callable[[Tensor, Any], Tensor], sample_drop_ratio: float =
mdm/model/dinov2_rgbd/layers/block.py:187
↓ 2 callersFunctionget_attn_bias_and_cat
this will perform the index select, cat the tensors, and provide the attn_bias from cache
mdm/model/dinov2_rgbd/layers/block.py:163
↓ 2 callersMethodinterpolate_pos_encoding_without_cls
(self, x, h, w, input_pos_embed)
mdm/model/dinov2_rgbd/models/vision_transformer.py:244
↓ 2 callersFunctionmake_2tuple
(x)
mdm/model/dinov2_rgbd/layers/patch_embed_mlp.py:17
↓ 2 callersFunctionmake_2tuple
(x)
mdm/model/dinov2_rgbd/layers/patch_embed.py:16
↓ 2 callersMethodprepare_tokens_with_masks
(self, x_img, x_depth, x_img_mask=None, x_depth_mask=None, masks=None, **kwargs)
mdm/model/dinov2_rgbd/models/vision_transformer.py:276
↓ 1 callersMethod__enter__
(self)
mdm/utils/tools.py:179
↓ 1 callersMethod__exit__
(self, exc_type, exc_val, exc_tb)
mdm/utils/tools.py:198
↓ 1 callersMethod__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
mdm/model/dinov2_rgbd/layers/swiglu_ffn.py:15
↓ 1 callersFunction_build_mlp
(nlayers, in_dim, bottleneck_dim, hidden_dim=None, use_bn=False, bias=True)
mdm/model/dinov2_rgbd/layers/dino_head.py:44
↓ 1 callersFunction_compute_depth_invalid_mask
Compute depth validity for each patch Args: depth_values: [B, 1, H_img, W_img] raw depth map H_patch, W_patch: patch grid di
mdm/model/dinov2_rgbd/models/mask_utils.py:68
↓ 1 callersMethod_get_intermediate_layers_chunked
(self, x_img, x_depth, x_img_mask=None, x_depth_mask=None, n=1, return_mae_aux=False, **kwargs)
mdm/model/dinov2_rgbd/models/vision_transformer.py:354
↓ 1 callersMethod_get_intermediate_layers_not_chunked
(self, x_img, x_depth, x_img_mask=None, x_depth_mask=None, n=1, return_mae_aux=False, **kwargs)
mdm/model/dinov2_rgbd/models/vision_transformer.py:335
↓ 1 callersMethod_get_pad
(self, size)
mdm/model/dinov2_rgbd/hub/utils.py:28
↓ 1 callersFunction_guess_cluster_type
()
mdm/model/dinov2_rgbd/utils/cluster.py:18
↓ 1 callersFunction_ntuple
(n)
mdm/model/dinov2_rgbd/layers/patch_embed_mlp.py:53
↓ 1 callersFunctionadd_residual
(x, brange, residual, residual_scale_factor, scaling_vector=None)
mdm/model/dinov2_rgbd/layers/block.py:148
↓ 1 callersFunctionapply_scaling_rules_to_cfg
(cfg)
mdm/model/dinov2_rgbd/utils/config.py:21
↓ 1 callersFunctionbuild_model
(args, only_teacher=False, img_size=224)
mdm/model/dinov2_rgbd/models/__init__.py:25
↓ 1 callersFunctiondefault_setup
(args)
mdm/model/dinov2_rgbd/utils/config.py:49
↓ 1 callersFunctiondepth_masking
Perform patch masking based on depth validity Args: x: [B, N, D] input features (after patch embedding) patch_num_h: int, he
mdm/model/dinov2_rgbd/models/mask_utils.py:2
↓ 1 callersFunctiondepth_to_pointcloud
Convert depth map to point cloud (pure Tensor version, no point filtering) Args: depth: torch.Tensor, shape (H, W) or (B, H, W), dep
mdm/model/utils.py:68
↓ 1 callersFunctiondrop_path
(x, drop_prob: float = 0.0, training: bool = False)
mdm/model/dinov2_rgbd/layers/drop_path.py:14
↓ 1 callersMethodextract_features
(self, outputs, norm=True)
mdm/model/dinov2_rgbd/models/vision_transformer.py:369
↓ 1 callersFunctionflatten_nested_dict
Flattens a nested dictionary into a single-level dictionary, with keys as tuples.
mdm/utils/tools.py:85
↓ 1 callersMethodforward
(self, image: torch.Tensor, num_tokens: Union[int, torch.LongTensor],
mdm/model/v2.py:98
↓ 1 callersMethodforward
(self, x: Tensor)
mdm/model/dinov2_rgbd/layers/block.py:88
↓ 1 callersMethodforward
(self, x: Tensor, attn_bias=None)
mdm/model/dinov2_rgbd/layers/attention.py:70
↓ 1 callersMethodforward_feat
(self, image: torch.Tensor, num_tokens: Union[int, torch.LongTensor],
mdm/model/v2.py:237
↓ 1 callersMethodforward_nested
x_list contains a list of tensors to nest together and run
mdm/model/dinov2_rgbd/layers/block.py:211
↓ 1 callersMethodfrom_pretrained
( cls, pretrained_model_name_or_path: Union[str, Path, IO[bytes]], model_kwargs: Opt
mdm/model/v2.py:68
↓ 1 callersFunctionget_branges_scales
(x, sample_drop_ratio=0.0)
mdm/model/dinov2_rgbd/layers/block.py:140
↓ 1 callersFunctionget_cfg_from_args
(args)
mdm/model/dinov2_rgbd/utils/config.py:40
↓ 1 callersFunctionget_checkpoint_path
(cluster_type: Optional[ClusterType] = None)
mdm/model/dinov2_rgbd/utils/cluster.py:38
↓ 1 callersMethodget_intermediate_layers_mae
( self, x_img: torch.Tensor, x_depth: torch.Tensor, x_img_mask: torch.Tensor=N
mdm/model/dinov2_rgbd/models/vision_transformer.py:390
↓ 1 callersFunctionget_nested_dict
(d: Dict[str, Dict], keys: Tuple[str, ...], default: Any = None)
mdm/utils/tools.py:52
↓ 1 callersFunctionget_slurm_partition
(cluster_type: Optional[ClusterType] = None)
mdm/model/dinov2_rgbd/utils/cluster.py:61
↓ 1 callersFunctionget_vit_lr_decay_rate
Calculate lr decay rate for different ViT blocks. Args: name (string): parameter name. lr_decay_rate (float): base lr decay r
mdm/model/dinov2_rgbd/utils/param_groups.py:13
↓ 1 callersMethodinfer
( self, image: torch.Tensor, depth_in: torch.Tensor = None, num_tokens: int
mdm/model/v2.py:171
↓ 1 callersMethodinit_weights
(self)
mdm/model/dinov2_rgbd/models/vision_transformer.py:201
↓ 1 callersFunctionload_depth_map
Load depth map from PNG file (16-bit) and convert to meters. Args: depth_path (str): Path to depth image scale (float): Scal
example.py:57
↓ 1 callersFunctionload_intrinsics
Load camera intrinsics and normalize by image dimensions. Args: intrinsics_path (str): Path to intrinsics file (.txt or .json)
example.py:86
↓ 1 callersFunctionmain
Main function with argument parsing and execution.
example.py:157
↓ 1 callersFunctionnamed_apply
(fn: Callable, module: nn.Module, name="", depth_first=True, include_root=False)
mdm/model/dinov2_rgbd/models/vision_transformer.py:27
↓ 1 callersFunctionnormalized_view_plane_uv
UV with left-top corner as (-width / diagonal, -height / diagonal) and right-bottom corner as (width / diagonal, height / diagonal)
mdm/utils/geo.py:3
↓ 1 callersFunctionpreprocess_input_image
Load and preprocess RGB image. Args: image_path (str): Path to RGB image device (torch.device): Device to load tensor on
example.py:24
↓ 1 callersFunctionset_nested_dict
(d: Dict[str, Dict], keys: Tuple[str, ...], value: Any)
mdm/utils/tools.py:59
↓ 1 callersFunctiontraverse_nested_dict_keys
(d: Dict[str, Dict])
mdm/utils/tools.py:43
↓ 1 callersFunctionwrap_dinov2_attention_with_sdpa
(module: nn.Module)
mdm/model/utils.py:22
↓ 1 callersFunctionwrite_config
(cfg, output_dir, name="config.yaml")
mdm/model/dinov2_rgbd/utils/config.py:32
Method__call__
(self, func: Callable)
mdm/utils/tools.py:164
Method__call__
(self, fn)
mdm/utils/tools.py:268
Method__enter__
(self)
mdm/utils/tools.py:34
Method__enter__
(self)
mdm/utils/tools.py:276
Method__exit__
(self, exc_type, exc_val, exc_tb)
mdm/utils/tools.py:37
Method__exit__
(self, exc_type, exc_val, exc_tb)
mdm/utils/tools.py:281
Method__getitem__
(self, it)
mdm/model/dinov2_rgbd/utils/utils.py:83
Method__init__
(self, callback: Callable, exception: type)
mdm/utils/tools.py:30
Method__init__
(self, name: str = None, verbose: bool = True, average: bool = False)
mdm/utils/tools.py:155
Method__init__
(self, action: str = 'ignore', **kwargs)
mdm/utils/tools.py:264
Method__init__
(self, encoder: Dict[str, Any], neck: Dict[str, Any], depth_head: Dict[str, Any] = No
mdm/model/v2.py:29
Method__init__
( self, in_channels: int, out_channels: int = None, hidden_channels: int =
mdm/model/modules_decoder.py:17
Method__init__
(self, in_channels: int, out_channels: int, type_: Literal['pixel_shuffle', 'neares
mdm/model/modules_decoder.py:70
Method__init__
(self, dim_in: List[Optional[int]], dim_res_blocks: List[int], dim_out: List[Optional
mdm/model/modules_decoder.py:127
Method__init__
(self, backbone: str, intermediate_layers: Union[int, List[int]], dim_out: int, ignore_layers: Union[str, List
mdm/model/modules_rgbd_encoder.py:23
Method__init__
(self, base_value, final_value, total_iters, warmup_iters=0, start_warmup_value=0, freeze_iters=0)
mdm/model/dinov2_rgbd/utils/utils.py:68
Method__init__
(self, multiple)
mdm/model/dinov2_rgbd/hub/utils.py:24
Method__init__
(self, downscale_factor)
mdm/model/dinov2_rgbd/layers/patch_embed_mlp.py:26
Method__init__
(self, dims: tuple[int, ...])
mdm/model/dinov2_rgbd/layers/patch_embed_mlp.py:41
Method__init__
( self, img_size: Union[int, Tuple[int, int]] = 224, patch_size: Union[int, Tuple[int,
mdm/model/dinov2_rgbd/layers/patch_embed_mlp.py:96
Method__init__
(self, drop_prob=None)
mdm/model/dinov2_rgbd/layers/drop_path.py:29
Method__init__
( self, dim: int, num_heads: int, mlp_ratio: float = 4.0, qkv_bias: bo
mdm/model/dinov2_rgbd/layers/block.py:43
Method__init__
( self, in_dim, out_dim, use_bn=False, nlayers=3, hidden_dim=2
mdm/model/dinov2_rgbd/layers/dino_head.py:13
Method__init__
( self, dim: int, init_values: Union[float, Tensor] = 1e-5, inplace: bool = Fa
mdm/model/dinov2_rgbd/layers/layer_scale.py:16
Method__init__
( self, dim: int, num_heads: int = 8, qkv_bias: bool = False, proj_bia
mdm/model/dinov2_rgbd/layers/attention.py:38
Method__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
mdm/model/dinov2_rgbd/layers/mlp.py:17
Method__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
mdm/model/dinov2_rgbd/layers/swiglu_ffn.py:55
Method__init__
( self, img_size: Union[int, Tuple[int, int]] = 224, patch_size: Union[int, Tuple[int,
mdm/model/dinov2_rgbd/layers/patch_embed.py:37
Method__init__
Args: img_size (int, tuple): input image size patch_size (int, tuple): patch size in_chans (int): number
mdm/model/dinov2_rgbd/models/vision_transformer.py:46
Method__repr__
(self)
mdm/model/dinov2_rgbd/layers/patch_embed_mlp.py:45
Function_fn
(input)
mdm/utils/tools.py:242
Method_init_weights
(self, m)
mdm/model/dinov2_rgbd/layers/dino_head.py:30
Functionas_torch_dtype
(dtype: TypeSpec)
mdm/model/dinov2_rgbd/utils/dtype.py:31
Methodattn_residual_func
(x: Tensor)
mdm/model/dinov2_rgbd/layers/block.py:89
Methodattn_residual_func
(x: Tensor, attn_bias=None)
mdm/model/dinov2_rgbd/layers/block.py:219
Methodaverage_time
(self)
mdm/utils/tools.py:190
Functionbuild_model_from_cfg
(cfg, only_teacher=False)
mdm/model/dinov2_rgbd/models/__init__.py:53
next →1–100 of 198, ranked by callers