Functions — github.com/NVlabs/FoundationStereo

hub / github.com/NVlabs/FoundationStereo / functions

Functions881 in github.com/NVlabs/FoundationStereo

⨍ Functions 881◇ Types & classes 190↳ Endpoints 1

↓ 1 callersFunctionwrite_config

(cfg, output_dir, name="config.yaml")

dinov2/dinov2/utils/config.py:32

Method__call__

(self, disp, coords, low_memory=False)

core/geometry.py:42

Method__call__

(self, sample)

depth_anything/util/transform.py:168

Method__call__

(self, sample)

depth_anything/util/transform.py:219

Method__call__

(self, sample)

depth_anything/util/transform.py:232

Method__call__

(self, num_masking_patches=0)

dinov2/dinov2/data/masking.py:73

Method__call__

Args: pic (PIL Image, numpy.ndarray or torch.tensor): Image to be converted to tensor. Returns: Tensor: Conve

dinov2/dinov2/data/transforms.py:29

Method__call__

(self, image)

dinov2/dinov2/data/augmentations.py:96

Method__call__

Args: cls_pred (Tensor): Predicted classification logits, shape [num_query, num_class]. gt_labels (Te

dinov2/dinov2/eval/segmentation_m2f/models/losses/match_costs.py:36

Method__call__

Args: mask_preds (Tensor): Mask prediction logits in shape (N1, H, W). gt_masks (Tensor): Ground truth in shape (N2,

dinov2/dinov2/eval/segmentation_m2f/models/losses/match_costs.py:89

Method__call__

Args: cls_pred (Tensor): Predicted classification logits. gt_labels (Tensor): Labels. Returns: Te

dinov2/dinov2/eval/segmentation_m2f/models/losses/match_costs.py:139

Method__call__

(self)

dinov2/dinov2/run/eval/linear.py:22

Method__call__

(self)

dinov2/dinov2/run/eval/log_regression.py:22

Method__call__

(self)

dinov2/dinov2/run/eval/knn.py:22

Method__call__

(self)

dinov2/dinov2/run/train/train.py:22

Method__enter__

(self)

core/foundation_stereo.py:37

Method__exit__

(self, *args)

core/foundation_stereo.py:39

Method__getattr__

(self, attr)

dinov2/dinov2/logging/helpers.py:33

Method__getitem__

(self, it)

dinov2/dinov2/utils/utils.py:83

Method__getitem__

(self, index: int)

dinov2/dinov2/data/adapters.py:22

Method__getitem__

(self, index: int)

dinov2/dinov2/data/datasets/extended.py:23

Method__getitem__

(self, index: int)

dinov2/dinov2/data/datasets/image_net_22k.py:234

Method__init__

(self, in_planes, planes, norm_fn='group', stride=1)

core/extractor.py:22

Method__init__

(self, output_dim=[128], norm_fn='batch', dropout=0.0, downsample=3)

core/extractor.py:86

Method__init__

(self, args, output_dim=[128], norm_fn='batch', downsample=3)

core/extractor.py:194

Method__init__

(self, encoder='vits')

core/extractor.py:294

Method__init__

(self, cfg, in_channels, feat_dims=None)

core/foundation_stereo.py:52

Method__init__

(self, args)

core/foundation_stereo.py:134

Method__init__

(self, normalized_shape, eps=1e-6)

core/submodule.py:30

Method__init__

(self, in_channels, out_channels, deconv=False, is_3d=False, bn=True, relu=True, norm='batch', **kwargs)

core/submodule.py:49

Method__init__

(self, C_in, C_out, hidden=None, kernel_size=3, kernel_disp=None, stride=1, norm=nn.BatchNorm3d)

core/submodule.py:86

Method__init__

(self, inplanes, planes, kernel_size=3, stride=1, padding=1, downsample=None, groups=1, base_width=64, dilatio

core/submodule.py:116

Method__init__

(self, inplanes, planes, kernel_size=3, stride=1, padding=1, downsample=None, groups=1, base_width=64, dilatio

core/submodule.py:156

Method__init__

(self, embed_dim, num_heads)

core/submodule.py:195

Method__init__

(self, embed_dim, num_heads, dim_feedforward, dropout=0.1, act=nn.GELU, norm=nn.LayerNorm)

core/submodule.py:230

Method__init__

(self, C_in, C_out, is_3d=False, kernel_size=3, bias=True, stride=1, padding=1)

core/submodule.py:258

Method__init__

(self, in_channels, out_channels, deconv=False, is_3d=False, IN=True, relu=True, **kwargs)

core/submodule.py:318

Method__init__

(self, in_channels, out_channels, deconv=False, is_3d=False, concat=True, keep_concat=True, IN=True, relu=True

core/submodule.py:347

Method__init__

(self, cv_chan, feat_chan)

core/submodule.py:435

Method__init__

(self, d_model, max_len=512)

core/submodule.py:469

Method__init__

(self, d_model, nhead, dim_feedforward, dropout=0.1, act=nn.GELU, norm_first=False, num_transformer=6, max_len

core/submodule.py:503

Method__init__

(self, in_planes, ratio=16)

core/submodule.py:529

Method__init__

(self, kernel_size=7)

core/submodule.py:546

Method__init__

(self, dim, layer_scale_init_value=1e-6, expan_ratio=4, kernel_size=7, norm='layer')

core/submodule.py:562

Method__init__

(self, init_fmap1, init_fmap2, geo_volume, num_levels=2, dx=None)

core/geometry.py:17

Method__init__

(self, input_dim=128, hidden_dim=256, output_dim=1)

core/update.py:20

Method__init__

(self, args, ngroup=8)

core/update.py:51

Method__init__

(self, hidden_dim=128, input_dim=256, kernel_size=3)

core/update.py:83

Method__init__

(self, hidden_dim=128, input_dim=256, small_kernel_size=1, large_kernel_size=3, patch_size=None)

core/update.py:98

Method__init__

(self, args, hidden_dim=128, volume_dim=8)

core/update.py:122

Method__init__

(self, dims, mode='sintel', divis_by=8, force_square=False)

core/utils/utils.py:19

Method__init__

(self, args)

scripts/run_demo_batch.py:42

Method__init__

(self, args)

scripts/make_onnx.py:10

Method__init__

(self, encoder='vitl', features=256, out_channels=[256, 512, 1024, 1024], use_bn=False, use_clstoken=False, pr

depth_anything/dpt.py:148

Method__init__

(self, config)

depth_anything/dpt.py:175

Method__init__

Init. Args: features (int): number of features

depth_anything/blocks.py:99

Method__init__

Init. Args: width (int): desired output width height (int): desired output height resize_target (bool, op

depth_anything/util/transform.py:58

Method__init__

(self, mean, std)

depth_anything/util/transform.py:215

Method__init__

(self)

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)

dinov2/dinov2/utils/utils.py:68

Method__init__

(self, delimiter="\t", output_file=None)

dinov2/dinov2/logging/helpers.py:21

Method__init__

(self, window_size=20, fmt=None)

dinov2/dinov2/logging/helpers.py:138

Method__init__

(self)

dinov2/dinov2/distributed/__init__.py:152

Method__init__

( self, input_size, num_masking_patches=None, min_num_patches=4, max_n

dinov2/dinov2/data/masking.py:12

Method__init__

( self, *, size: int, sample_count: int, shuffle: bool = False,

dinov2/dinov2/data/samplers.py:18

Method__init__

( self, *, sample_count: int, shuffle: bool = False, seed: int = 0,

dinov2/dinov2/data/samplers.py:79

Method__init__

( self, *, sample_count: int, shuffle: bool = False, seed: int = 0,

dinov2/dinov2/data/samplers.py:166

Method__init__

(self, *, p: float = 0.5, radius_min: float = 0.1, radius_max: float = 2.0)

dinov2/dinov2/data/transforms.py:17

Method__init__

(self, dataset)

dinov2/dinov2/data/adapters.py:12

Method__init__

( self, global_crops_scale, local_crops_scale, local_crops_number, glo

dinov2/dinov2/data/augmentations.py:20

Method__init__

(self, *args, **kwargs)

dinov2/dinov2/data/datasets/extended.py:14

Method__init__

( self, *, root: str, extra: str, transforms: Optional[Callable] = Non

dinov2/dinov2/data/datasets/image_net_22k.py:101

Method__init__

(self, image_data: bytes)

dinov2/dinov2/data/datasets/decoders.py:18

Method__init__

(self, target: Any)

dinov2/dinov2/data/datasets/decoders.py:27

Method__init__

( self, *, split: "ImageNet.Split", root: str, extra: str, tra

dinov2/dinov2/data/datasets/image_net.py:59

Method__init__

(self, out_dim, use_n_blocks, use_avgpool, num_classes=1000)

dinov2/dinov2/eval/linear.py:189

Method__init__

(self, linear_classifier, class_mapping=None)

dinov2/dinov2/eval/linear.py:218

Method__init__

( self, C, max_iter=DEFAULT_MAX_ITER, dtype=torch.float64, device=_CPU

dinov2/dinov2/eval/log_regression.py:110

Method__init__

(self, feature_model, n_last_blocks, autocast_ctx)

dinov2/dinov2/eval/utils.py:31

Method__init__

( self, num_classes: int, top_k: int = 1, **kwargs: Any, )

dinov2/dinov2/eval/metrics.py:76

Method__init__

(self, keys)

dinov2/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

dinov2/dinov2/eval/segmentation_m2f/ops/modules/ms_deform_attn.py:64

Method__init__

(self, strides, offset=0.5)

dinov2/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)

dinov2/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)

dinov2/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)

dinov2/dinov2/eval/segmentation_m2f/core/box/samplers/sampling_result.py:28

Method__init__

(self, **kwargs)

dinov2/dinov2/eval/segmentation_m2f/core/box/samplers/mask_pseudo_sampler.py:20

Method__init__

( self, backbone, decode_head, neck=None, auxiliary_head=None,

dinov2/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)

dinov2/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

dinov2/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

dinov2/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,

dinov2/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,

dinov2/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:269

Method__init__

(self, inplanes=64, embed_dim=384, with_cp=False)

dinov2/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:375

Method__init__

( self, in_features: int, hidden_features: Optional[int] = None, out_features:

dinov2/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

dinov2/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)

dinov2/dinov2/eval/segmentation_m2f/models/backbones/vit.py:133

Method__init__

( self, dim: int, num_heads: int = 8, qkv_bias: bool = False, attn_dro

dinov2/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

dinov2/dinov2/eval/segmentation_m2f/models/backbones/vit.py:225

Method__init__

( self, dim, num_heads, mlp_ratio=4.0, qkv_bias=False, drop=0.

dinov2/dinov2/eval/segmentation_m2f/models/backbones/vit.py:320

← previous next →401–500 of 881, ranked by callers