MCPcopy Index your code

hub / github.com/XiandaGuo/OpenStereo / functions

Functions2,487 in github.com/XiandaGuo/OpenStereo

↓ 190 callersMethodinfo
(self)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/utils/assigner.py:27
↓ 167 callersMethodto
Change the device of the data inplace. Example: >>> self = SamplingResult.random() >>> print(f'self = {self.to(None)}
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/core/box/samplers/sampling_result.py:58
↓ 55 callersMethodsplit
(self)
stereo/modeling/models/foundationstereo/dinov2/data/datasets/image_net.py:78
↓ 43 callersFunctionconvbn_3d
(in_channels, out_channels, kernel_size, stride, pad)
stereo/modeling/models/cfnet/submodule.py:115
↓ 39 callersMethodtime
(self)
deploy/deploy_utils.py:213
↓ 38 callersMethodlog
deploy/cpp/include/inference.h:23
↓ 37 callersMethodnorm
(self)
stereo/modeling/models/foundationstereo/dinov2/hub/depth/decode_heads.py:453
↓ 35 callersFunctionprint
(*args, **kwargs)
stereo/modeling/models/nmrf/utils/dist_utils.py:94
↓ 28 callersFunctionconvbn_3d
(in_channels, out_channels, kernel_size, stride, pad)
stereo/modeling/models/casnet/submodule.py:16
↓ 28 callersMethodload
(self, *args, **kwargs)
stereo/modeling/models/foundationstereo/dinov2/fsdp/__init__.py:114
↓ 27 callersMethodupdate
(self, **kwargs)
stereo/modeling/models/nmrf/utils/misc.py:156
↓ 26 callersFunctionconv
(in_planes, out_planes, kernel_size=3, stride=1, batchNorm=False)
stereo/modeling/models/fadnet/submodule.py:380
↓ 25 callersMethodupdate
(self, **kwargs)
stereo/modeling/models/foundationstereo/dinov2/logging/helpers.py:26
↓ 23 callersFunctionconvbn
(in_channels, out_channels, kernel_size, stride, pad, dilation)
stereo/modeling/models/cfnet/submodule.py:109
↓ 22 callersMethodmax
(self)
stereo/modeling/models/nmrf/utils/misc.py:135
↓ 20 callersMethodload_state_dict
Loads the warmup scheduler's state. Arguments: state_dict (dict): warmup scheduler state. Should be an object returned
stereo/utils/warmup.py:33
↓ 20 callersMethodstep
Performs a single optimization step. Arguments: closure (callable, optional): A closure that reevaluates the model
stereo/utils/lamb.py:48
↓ 18 callersMethodmax
(self)
stereo/modeling/models/foundationstereo/dinov2/logging/helpers.py:180
↓ 18 callersFunctionpool2x
(x)
stereo/modeling/models/monster/update.py:105
↓ 16 callersMethod__init__
(self, in_channels, out_channels, deconv=False, is_3d=False, concat=True, keep_concat=True, bn=True, relu=True
stereo/modeling/models/foundationstereo/core/submodule.py:317
↓ 16 callersMethod__init__
(self, max_in=98, max_out=128, stride = 1)
stereo/modeling/models/fadnet/submodule.py:305
↓ 16 callersFunctionbatched_index_select
(source, dim, index)
stereo/modeling/models/sttr/utilities/misc.py:31
↓ 15 callersMethod__init__
(self, in_channels, out_channels, deconv=False, is_3d=False, concat=True, keep_concat=True, bn=True, relu=True
stereo/modeling/models/fast_foundationstereo/core/submodule.py:263
↓ 15 callersMethod__init__
(self, hidden_dim, input_dim, kernel_size=3)
stereo/modeling/models/monster/update.py:26
↓ 15 callersMethoddevice
Returns the device that the model is on.
stereo/modeling/models/nmrf/NMRF.py:190
↓ 14 callersFunctioncolorstr
(*input)
deploy/deploy_utils.py:105
↓ 14 callersFunctionconvbn
(in_channels, out_channels, kernel_size, stride, pad, dilation)
stereo/modeling/models/casnet/submodule.py:10
↓ 14 callersMethodfeature_extraction
(self, img)
stereo/modeling/models/aanet/aanet.py:52
↓ 14 callersMethodpad
(self, *inputs)
stereo/modeling/models/nmrf/utils/frame_utils.py:303
↓ 14 callersFunctionpredict_flow
(in_planes, out_planes = 1)
stereo/modeling/models/fadnet/submodule.py:415
↓ 14 callersMethodupdate
(self, preds: Tensor, target: Tensor)
stereo/modeling/models/foundationstereo/dinov2/eval/metrics.py:87
↓ 13 callersMethod__init__
StereoNet uses leaky relu (alpha = 0.2)
stereo/modeling/models/aanet/submodule.py:36
↓ 13 callersMethoddecode
(self)
stereo/modeling/models/foundationstereo/dinov2/data/datasets/decoders.py:13
↓ 12 callersMethodbackward
(ctx, grad_output)
stereo/modeling/models/nmrf/ops/functions/downsample_func.py:11
↓ 12 callersFunctiondeconv
(in_planes, out_planes)
stereo/modeling/models/fadnet/submodule.py:474
↓ 12 callersFunctioninterp
(x, dest)
stereo/modeling/models/monster/update.py:72
↓ 11 callersFunction_make_dinov2_model
( *, arch_name: str = "vit_large", img_size: int = 518, patch_size: int = 14, init_values:
stereo/modeling/models/foundationstereo/dinov2/hub/backbones.py:18
↓ 11 callersMethodpad
(self, *inputs)
stereo/modeling/models/foundationstereo/core/utils/utils.py:33
↓ 11 callersFunctionresize
(input, size=None, scale_factor=None, mode="nearest", align_corners=None, warning=False)
stereo/modeling/models/foundationstereo/dinov2/eval/depth/ops/wrappers.py:11
↓ 11 callersFunctionto_2tuple
(x)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/vit.py:48
↓ 10 callersMethod__init__
(self, input_dim, hidden_dim, output_dim, num_layers)
stereo/modeling/models/nmrf/NMP.py:57
↓ 10 callersMethod_make_layer
(self, dim, stride=1)
stereo/modeling/models/foundationstereo/core/extractor.py:153
↓ 10 callersFunctionconv2d
(in_channels, out_channels, kernel_size=3, stride=1, dilation=1, groups=1)
stereo/modeling/models/monster/refinement.py:8
↓ 10 callersFunctionconvbn
(in_planes, out_planes, kernel_size, stride, pad, dilation)
stereo/modeling/models/fadnet/submodule.py:404
↓ 10 callersFunctiondisparity_regression
(x, maxdisp)
stereo/modeling/models/msnet/submodule.py:282
↓ 10 callersMethodsave
Args: filepath (str): a string that contains the absolute path, including the file name, where the visualized ima
stereo/modeling/models/nmrf/utils/visualization.py:130
↓ 10 callersMethodstate_dict
Returns the state of the warmup scheduler as a :class:`dict`. It contains an entry for every variable in self.__dict__ which is not t
stereo/utils/warmup.py:25
↓ 9 callersMethod__init__
( self, embed_dims=256, feedforward_channels=1024, num_fcs=2, act_cfg=
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/utils/transformer.py:265
↓ 9 callersMethod__init__
(self)
stereo/modeling/models/monster/refinement.py:483
↓ 9 callersMethodcheckpoint
(self)
stereo/modeling/models/foundationstereo/dinov2/run/eval/knn.py:28
↓ 9 callersFunctionconv1x1
1x1 convolution
stereo/modeling/models/aanet/submodule.py:20
↓ 9 callersFunctionconv3d_bn_relu
(batchNorm, in_planes, out_planes, kernel_size=3, stride=1, padding=1, dilation=1, bias=True)
stereo/modeling/models/psmnet/submodule.py:160
↓ 9 callersFunctionconv_bn_relu
(batchNorm, in_planes, out_planes, kernel_size=3, stride=1, padding=1, dilation=1, bias=True)
stereo/modeling/models/psmnet/submodule.py:103
↓ 9 callersFunctionprint
(*args, **kwargs)
stereo/modeling/models/foundationstereo/dinov2/distributed/__init__.py:82
↓ 9 callersFunctionwrite_tensorboard
(tb_writer, tb_info, step)
stereo/utils/common_utils.py:223
↓ 8 callersMethod__init__
(self, input_transform="resize_concat", in_index=(0, 1, 2, 3), upsample=1, **kwargs)
stereo/modeling/models/foundationstereo/dinov2/hub/depth/decode_heads.py:226
↓ 8 callersFunction_make_dinov2_linear_classifier
( *, arch_name: str = "vit_large", layers: int = 4, pretrained: bool = True, weights: Unio
stereo/modeling/models/foundationstereo/dinov2/hub/classifiers.py:86
↓ 8 callersFunction_make_fusion_block
(features, use_bn, size=None)
stereo/modeling/models/monster/depth_anything_v2/dpt.py:12
↓ 8 callersFunctioncheck_requirements
(requirements=ROOT / 'requirements.txt', exclude=(), install=True, cmds='', logger=None)
deploy/deploy_utils.py:69
↓ 8 callersFunctionconvbn
(in_channels, out_channels, kernel_size, stride, pad, dilation)
stereo/modeling/models/msnet/submodule.py:29
↓ 8 callersFunctionconvbn_3d
(in_channels, out_channels, kernel_size, stride, pad)
stereo/modeling/models/gwcnet/gwcnet_disp_processor.py:8
↓ 8 callersFunctiondisparity_regression
(x, disp_values)
stereo/modeling/models/casnet/submodule.py:22
↓ 8 callersFunctionmake_dataset
Creates a dataset with the specified parameters. Args: dataset_str: A dataset string description (e.g. ImageNet:split=TRAIN).
stereo/modeling/models/foundationstereo/dinov2/data/loaders.py:67
↓ 7 callersMethod__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/vit.py:53
↓ 7 callersFunctioncolor_map_tensorboard
(disp_gt, pred, disp_max=192)
stereo/utils/common_utils.py:195
↓ 7 callersMethodget_intermediate_layers
( self, x: torch.Tensor, n: Union[int, Sequence] = 1, # Layers or n last layers to ta
stereo/modeling/models/monster/depth_anything_v2/dinov2.py:297
↓ 7 callersMethodpad
(self, *inputs)
stereo/modeling/models/fast_foundationstereo/core/utils/utils.py:22
↓ 7 callersFunctionreadpfm
(file)
stereo/datasets/dataset_utils/readpfm.py:7
↓ 6 callersMethod__init__
(self, in_channels, out_channels, deconv=False, is_3d=False, concat=True, keep_concat=True, bn=True,
stereo/modeling/models/coex/submodule.py:75
↓ 6 callersMethod__init__
( self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.0 )
stereo/modeling/models/nmrf/swin.py:27
↓ 6 callersMethod__init__
(self, dim=768)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:71
↓ 6 callersMethod__init__
(self, in_channels, out_channels, deconv=False, is_3d=False, concat=True, keep_concat=True, bn=True, relu=True
stereo/modeling/models/monster/submodule.py:40
↓ 6 callersMethod__init__
(self, hidden_dim, input_dim, kernel_size=3)
stereo/modeling/models/igevpp/update.py:26
↓ 6 callersMethod_get_entries
(self)
stereo/modeling/models/foundationstereo/dinov2/data/datasets/image_net.py:105
↓ 6 callersFunction_make_dinov2_model_name
(arch_name: str, patch_size: int, num_register_tokens: int = 0)
stereo/modeling/models/foundationstereo/dinov2/hub/utils.py:17
↓ 6 callersFunctionconsistent_padding_with_dilation
(padding, dilation, dim=2)
stereo/modeling/models/psmnet/submodule.py:14
↓ 6 callersMethodconstrain_to_multiple_of
(self, x, min_val=0, max_val=None)
stereo/modeling/models/foundationstereo/depth_anything/util/transform.py:100
↓ 6 callersMethodconstrain_to_multiple_of
(self, x, min_val=0, max_val=None)
stereo/modeling/models/monster/depth_anything_v2/util/transform.py:51
↓ 6 callersFunctionconvbn
(in_channels, out_channels, kernel_size, stride, pad, dilation)
stereo/modeling/models/gwcnet/gwcnet_backbone.py:6
↓ 6 callersFunctionconvbn_3d
(in_channels, out_channels, kernel_size, stride, pad)
stereo/modeling/models/gwcnet/hourglass.py:5
↓ 6 callersFunctionget_format_idx
(df, format_name)
deploy/deploy_utils.py:146
↓ 6 callersMethodpad
(self, *inputs)
stereo/modeling/models/igev/utils.py:19
↓ 5 callersFunctionFMish
Applies the mish function element-wise: mish(x) = x * tanh(softplus(x)) = x * tanh(ln(1 + exp(x))) See additional documentation for mi
stereo/modeling/models/cfnet/submodule.py:249
↓ 5 callersMethod__init__
(self, hidden_dim, input_dim, kernel_size=3)
stereo/modeling/disp_refinement/gru_blocks.py:255
↓ 5 callersMethod__init__
(self)
stereo/modeling/models/igev_rt/extractor.py:329
↓ 5 callersMethod__init__
(self, hidden_dim, input_dim, kernel_size=3)
stereo/modeling/models/igev_rt/update.py:26
↓ 5 callersMethod__init__
(self, hidden_dim, input_dim, kernel_size=3)
stereo/modeling/models/stereobase/gru_blocks.py:255
↓ 5 callersMethod__init__
(self)
stereo/modeling/models/igev/extractor.py:322
↓ 5 callersMethod__init__
(self, hidden_dim, input_dim, kernel_size=3)
stereo/modeling/models/igev/update.py:29
↓ 5 callersMethod__init__
(self, hidden_dim, input_dim, kernel_size=3)
stereo/modeling/models/foundationstereo/core/update.py:33
↓ 5 callersMethod__init__
( self, embed_dims=768, post_process_channels=[96, 192, 384, 768], readout_typ
stereo/modeling/models/foundationstereo/dinov2/eval/depth/models/decode_heads/dpt_head.py:227
↓ 5 callersMethod__init__
(self, channel_list, disable_final_activation=True)
stereo/modeling/models/iinet/networks.py:217
↓ 5 callersMethod__init__
(self)
stereo/modeling/models/monster/extractor.py:328
↓ 5 callersMethod__init__
(self)
stereo/modeling/models/igevpp/extractor.py:326
↓ 5 callersMethod__init__
(self)
stereo/modeling/models/cfnet/submodule.py:100
↓ 5 callersMethod_make_layer
(self, dim, stride=1)
stereo/modeling/disp_refinement/gru_blocks.py:121
↓ 5 callersMethod_make_layer
(self, dim, stride=1)
stereo/modeling/models/igev_rt/extractor.py:272
↓ 5 callersMethod_make_layer
(self, dim, stride=1)
stereo/modeling/models/stereobase/gru_blocks.py:121
↓ 5 callersMethod_make_layer
(self, dim, stride=1)
stereo/modeling/models/igev/extractor.py:265
↓ 5 callersMethod_make_layer
(self, dim, stride=1)
stereo/modeling/models/monster/extractor.py:271
next →1–100 of 2,487, ranked by callers