MCPcopy Create free account

hub / github.com/XiandaGuo/OpenStereo / functions

Functions2,487 in github.com/XiandaGuo/OpenStereo

Method__init__
(self, in_channels, base_channels=32)
stereo/modeling/models/casnet/cas_gwc.py:160
Method__init__
(self)
stereo/modeling/models/casnet/cas_gwc.py:260
Method__init__
(self, inplanes, planes, stride, downsample, pad, dilation)
stereo/modeling/models/casnet/submodule.py:66
Method__init__
(self, in_channels)
stereo/modeling/models/casnet/cas_psm.py:7
Method__init__
(self, arch_mode="fpn", num_stage=None)
stereo/modeling/models/casnet/cas_psm.py:47
Method__init__
(self, in_channels, base_channels=32)
stereo/modeling/models/casnet/cas_psm.py:183
Method__init__
(self)
stereo/modeling/models/casnet/cas_psm.py:283
Method__init__
(self, args, cfgs, local_rank, global_rank, logger, tb_writer)
stereo/modeling/models/casnet/trainer.py:13
Method__init__
(self, args, cfgs, local_rank, global_rank, logger, tb_writer)
stereo/modeling/models/foundationstereo/trainer.py:20
Method__init__
(self, in_planes, planes, norm_fn='group', stride=1)
stereo/modeling/models/foundationstereo/core/extractor.py:19
Method__init__
(self, output_dim=[128], norm_fn='batch', dropout=0.0, downsample=3)
stereo/modeling/models/foundationstereo/core/extractor.py:83
Method__init__
(self, args, output_dim=[128], norm_fn='batch', downsample=3)
stereo/modeling/models/foundationstereo/core/extractor.py:191
Method__init__
(self, encoder='vits')
stereo/modeling/models/foundationstereo/core/extractor.py:291
Method__init__
(self, cfg, in_channels, feat_dims=None)
stereo/modeling/models/foundationstereo/core/foundation_stereo.py:43
Method__init__
(self, args)
stereo/modeling/models/foundationstereo/core/foundation_stereo.py:125
Method__init__
(self, normalized_shape, eps=1e-6)
stereo/modeling/models/foundationstereo/core/submodule.py:32
Method__init__
(self, in_channels, out_channels, deconv=False, is_3d=False, bn=True, relu=True, norm='batch', **kwargs)
stereo/modeling/models/foundationstereo/core/submodule.py:51
Method__init__
(self, C_in, C_out, hidden=None, kernel_size=3, kernel_disp=None, stride=1, norm=nn.BatchNorm3d)
stereo/modeling/models/foundationstereo/core/submodule.py:88
Method__init__
(self, inplanes, planes, kernel_size=3, stride=1, padding=1, downsample=None, groups=1, base_width=64, dilatio
stereo/modeling/models/foundationstereo/core/submodule.py:118
Method__init__
(self, inplanes, planes, kernel_size=3, stride=1, padding=1, downsample=None, groups=1, base_width=64, dilatio
stereo/modeling/models/foundationstereo/core/submodule.py:158
Method__init__
(self, embed_dim, num_heads)
stereo/modeling/models/foundationstereo/core/submodule.py:228
Method__init__
(self, embed_dim, num_heads, dim_feedforward, dropout=0.1, act=nn.GELU, norm=nn.LayerNorm)
stereo/modeling/models/foundationstereo/core/submodule.py:268
Method__init__
(self, C_in, C_out, is_3d=False, kernel_size=3, bias=True, stride=1, padding=1)
stereo/modeling/models/foundationstereo/core/submodule.py:296
Method__init__
(self, in_channels, out_channels, deconv=False, is_3d=False, IN=True, relu=True, **kwargs)
stereo/modeling/models/foundationstereo/core/submodule.py:356
Method__init__
(self, in_channels, out_channels, deconv=False, is_3d=False, concat=True, keep_concat=True, IN=True, relu=True
stereo/modeling/models/foundationstereo/core/submodule.py:385
Method__init__
(self, cv_chan, feat_chan)
stereo/modeling/models/foundationstereo/core/submodule.py:473
Method__init__
(self, d_model, max_len=512)
stereo/modeling/models/foundationstereo/core/submodule.py:507
Method__init__
(self, d_model, nhead, dim_feedforward, dropout=0.1, act=nn.GELU, norm_first=False, num_transformer=6, max_len
stereo/modeling/models/foundationstereo/core/submodule.py:541
Method__init__
(self, in_planes, ratio=16)
stereo/modeling/models/foundationstereo/core/submodule.py:567
Method__init__
(self, kernel_size=7)
stereo/modeling/models/foundationstereo/core/submodule.py:584
Method__init__
(self, dim, layer_scale_init_value=1e-6, expan_ratio=4, kernel_size=7, norm='layer')
stereo/modeling/models/foundationstereo/core/submodule.py:600
Method__init__
(self, init_fmap1, init_fmap2, geo_volume, num_levels=2, dx=None)
stereo/modeling/models/foundationstereo/core/geometry.py:16
Method__init__
(self, input_dim=128, hidden_dim=256, output_dim=1)
stereo/modeling/models/foundationstereo/core/update.py:19
Method__init__
(self, args, ngroup=8)
stereo/modeling/models/foundationstereo/core/update.py:50
Method__init__
(self, hidden_dim=128, input_dim=256, kernel_size=3)
stereo/modeling/models/foundationstereo/core/update.py:82
Method__init__
(self, hidden_dim=128, input_dim=256, small_kernel_size=1, large_kernel_size=3, patch_size=None)
stereo/modeling/models/foundationstereo/core/update.py:97
Method__init__
(self, args, hidden_dim=128, volume_dim=8)
stereo/modeling/models/foundationstereo/core/update.py:121
Method__init__
(self, dims, mode='sintel', divis_by=8, force_square=False)
stereo/modeling/models/foundationstereo/core/utils/utils.py:19
Method__init__
(self, encoder='vitl', features=256, out_channels=[256, 512, 1024, 1024], use_bn=False, use_clstoken=False, pr
stereo/modeling/models/foundationstereo/depth_anything/dpt.py:149
Method__init__
(self, config)
stereo/modeling/models/foundationstereo/depth_anything/dpt.py:177
Method__init__
Init. Args: features (int): number of features
stereo/modeling/models/foundationstereo/depth_anything/blocks.py:99
Method__init__
Init. Args: width (int): desired output width height (int): desired output height resize_target (bool, op
stereo/modeling/models/foundationstereo/depth_anything/util/transform.py:58
Method__init__
(self, mean, std)
stereo/modeling/models/foundationstereo/depth_anything/util/transform.py:215
Method__init__
(self)
stereo/modeling/models/foundationstereo/depth_anything/util/transform.py:229
Method__init__
(self, base_value, final_value, total_iters, warmup_iters=0, start_warmup_value=0, freeze_iters=0)
stereo/modeling/models/foundationstereo/dinov2/utils/utils.py:68
Method__init__
(self, delimiter="\t", output_file=None)
stereo/modeling/models/foundationstereo/dinov2/logging/helpers.py:21
Method__init__
(self, window_size=20, fmt=None)
stereo/modeling/models/foundationstereo/dinov2/logging/helpers.py:138
Method__init__
(self)
stereo/modeling/models/foundationstereo/dinov2/distributed/__init__.py:152
Method__init__
( self, input_size, num_masking_patches=None, min_num_patches=4, max_n
stereo/modeling/models/foundationstereo/dinov2/data/masking.py:12
Method__init__
( self, *, size: int, sample_count: int, shuffle: bool = False,
stereo/modeling/models/foundationstereo/dinov2/data/samplers.py:18
Method__init__
( self, *, sample_count: int, shuffle: bool = False, seed: int = 0,
stereo/modeling/models/foundationstereo/dinov2/data/samplers.py:79
Method__init__
( self, *, sample_count: int, shuffle: bool = False, seed: int = 0,
stereo/modeling/models/foundationstereo/dinov2/data/samplers.py:166
Method__init__
(self, *, p: float = 0.5, radius_min: float = 0.1, radius_max: float = 2.0)
stereo/modeling/models/foundationstereo/dinov2/data/transforms.py:17
Method__init__
(self, dataset)
stereo/modeling/models/foundationstereo/dinov2/data/adapters.py:12
Method__init__
( self, global_crops_scale, local_crops_scale, local_crops_number, glo
stereo/modeling/models/foundationstereo/dinov2/data/augmentations.py:20
Method__init__
(self, *args, **kwargs)
stereo/modeling/models/foundationstereo/dinov2/data/datasets/extended.py:14
Method__init__
( self, *, root: str, extra: str, transforms: Optional[Callable] = Non
stereo/modeling/models/foundationstereo/dinov2/data/datasets/image_net_22k.py:101
Method__init__
(self, image_data: bytes)
stereo/modeling/models/foundationstereo/dinov2/data/datasets/decoders.py:18
Method__init__
(self, target: Any)
stereo/modeling/models/foundationstereo/dinov2/data/datasets/decoders.py:27
Method__init__
( self, *, split: "ImageNet.Split", root: str, extra: str, tra
stereo/modeling/models/foundationstereo/dinov2/data/datasets/image_net.py:59
Method__init__
(self, out_dim, use_n_blocks, use_avgpool, num_classes=1000)
stereo/modeling/models/foundationstereo/dinov2/eval/linear.py:189
Method__init__
(self, linear_classifier, class_mapping=None)
stereo/modeling/models/foundationstereo/dinov2/eval/linear.py:218
Method__init__
( self, C, max_iter=DEFAULT_MAX_ITER, dtype=torch.float64, device=_CPU
stereo/modeling/models/foundationstereo/dinov2/eval/log_regression.py:110
Method__init__
(self, feature_model, n_last_blocks, autocast_ctx)
stereo/modeling/models/foundationstereo/dinov2/eval/utils.py:31
Method__init__
( self, num_classes: int, top_k: int = 1, **kwargs: Any, )
stereo/modeling/models/foundationstereo/dinov2/eval/metrics.py:76
Method__init__
(self, keys)
stereo/modeling/models/foundationstereo/dinov2/eval/knn.py:188
Method__init__
Multi-Scale Deformable Attention Module. :param d_model hidden dimension :param n_levels number of feature levels :p
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/ops/modules/ms_deform_attn.py:64
Method__init__
(self, strides, offset=0.5)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/core/anchor/point_generator.py:25
Method__init__
(self, num, pos_fraction, neg_pos_ub=-1, add_gt_as_proposals=True, **kwargs)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/core/box/samplers/base_sampler.py:16
Method__init__
(self, pos_inds, neg_inds, masks, gt_masks, assign_result, gt_flags)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/core/box/samplers/mask_sampling_result.py:17
Method__init__
(self, pos_inds, neg_inds, bboxes, gt_bboxes, assign_result, gt_flags)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/core/box/samplers/sampling_result.py:28
Method__init__
(self, **kwargs)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/core/box/samplers/mask_pseudo_sampler.py:20
Method__init__
( self, backbone, decode_head, neck=None, auxiliary_head=None,
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/segmentors/encoder_decoder_mask2former.py:25
Method__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.0)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:50
Method__init__
( self, dim, num_heads=6, n_points=4, n_levels=1, deform_ratio
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:89
Method__init__
( self, dim, num_heads=6, n_points=4, n_levels=1, deform_ratio
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:137
Method__init__
( self, dim, num_heads=6, n_points=4, norm_layer=partial(nn.LayerNorm,
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:174
Method__init__
( self, dim, num_heads=6, n_points=4, norm_layer=partial(nn.LayerNorm,
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:269
Method__init__
(self, inplanes=64, embed_dim=384, with_cp=False)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:375
Method__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/vit.py:80
Method__init__
( self, img_size=224, patch_size=16, in_chans=3, embed_dim=768, norm_layer=None, flatten=True, bias=Tr
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/vit.py:108
Method__init__
(self, dim, num_heads=8, qkv_bias=False, attn_drop=0.0, proj_drop=0.0)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/vit.py:133
Method__init__
( self, dim: int, num_heads: int = 8, qkv_bias: bool = False, attn_dro
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/vit.py:160
Method__init__
( self, dim, num_heads=8, qkv_bias=False, attn_drop=0.0, proj_drop=0.0, window_size=14, pad_mode="cons
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/vit.py:225
Method__init__
( self, dim, num_heads, mlp_ratio=4.0, qkv_bias=False, drop=0.
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/vit.py:320
Method__init__
Args: img_size (int, tuple): input image size patch_size (int, tuple): patch size in_chans (int): number
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/vit.py:396
Method__init__
( self, pretrain_size=224, num_heads=12, conv_inplane=64, n_points=4,
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/vit_adapter.py:21
Method__init__
(self, drop_prob: float = 0.0)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/drop_path.py:27
Method__init__
(self, num_feats, row_num_embed=50, col_num_embed=50, init_cfg=dict(type="Uniform", layer="Embedding"))
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/utils/positional_encoding.py:113
Method__init__
(self, kernel_size=1, stride=1, dilation=1, padding="corner")
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/utils/transformer.py:66
Method__init__
( self, in_channels, out_channels, kernel_size=2, stride=None,
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/utils/transformer.py:133
Method__init__
( self, attn_cfgs, feedforward_channels, ffn_dropout=0.0, operation_or
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/utils/transformer.py:338
Method__init__
(self, *args, post_norm_cfg=dict(type="LN"), **kwargs)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/utils/transformer.py:372
Method__init__
(self, *args, post_norm_cfg=dict(type="LN"), return_intermediate=False, **kwargs)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/utils/transformer.py:402
Method__init__
(self, encoder=None, decoder=None, init_cfg=None)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/utils/transformer.py:463
Method__init__
(self, *args, return_intermediate=False, **kwargs)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/utils/transformer.py:526
Method__init__
(self, as_two_stage=False, num_feature_levels=4, two_stage_num_proposals=300, **kwargs)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/utils/transformer.py:604
Method__init__
( self, in_channels=256, feat_channels=64, out_channels=None, input_fe
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/utils/transformer.py:913
Method__init__
(self, num_gts, gt_inds, labels)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/utils/assigner.py:21
Method__init__
( self, cls_cost=dict(type="ClassificationCost", weight=1.0), dice_cost=dict(type="Dic
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/utils/assigner.py:67
← previousnext →1,201–1,300 of 2,487, ranked by callers