MCPcopy Create free account

hub / github.com/NVlabs/FoundationStereo / types & classes

Types & classes190 in github.com/NVlabs/FoundationStereo

↓ 17 callersClassBasicConv
core/submodule.py:47
↓ 15 callersClassConvModule
A conv block that bundles conv/norm/activation layers. This block simplifies the usage of convolution layers, which are commonly used with a
dinov2/dinov2/hub/depth/decode_heads.py:299
↓ 9 callersClassConv3dNormActReduced
core/submodule.py:85
↓ 8 callersClassResidualBlock
core/extractor.py:21
↓ 6 callersClassFeatureAtt
core/submodule.py:434
↓ 6 callersClassLayerNorm2d
r""" https://huggingface.co/spaces/Roll20/pet_score/blob/b258ef28152ab0d5b377d9142a23346f863c1526/lib/timm/models/convnext.py#L85 LayerNorm for c
core/submodule.py:25
↓ 5 callersClassBasicConv_IN
core/submodule.py:316
↓ 5 callersClassCosineScheduler
dinov2/dinov2/utils/utils.py:67
↓ 4 callersClassDinoVisionTransformer
dinov2/dinov2/models/vision_transformer.py:45
↓ 4 callersClassExtractor
dinov2/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:88
↓ 4 callersClassInputPadder
Pads images such that dimensions are divisible by 8
core/utils/utils.py:17
↓ 4 callersClassLayerScale
dinov2/dinov2/layers/layer_scale.py:15
↓ 4 callersClassMetricLogger
dinov2/dinov2/logging/helpers.py:20
↓ 3 callersClassConv2x_IN
core/submodule.py:345
↓ 3 callersClassGaussianBlur
Apply Gaussian Blur to the PIL image.
dinov2/dinov2/data/transforms.py:12
↓ 3 callersClassModuleDictWithForward
dinov2/dinov2/eval/knn.py:240
↓ 3 callersClassResnetBasicBlock3D
core/submodule.py:155
↓ 3 callersClassSelectiveConvGRU
core/update.py:97
↓ 3 callersClassautocast
core/foundation_stereo.py:34
↓ 2 callersClassAssignResult
Collection of assign results.
dinov2/dinov2/eval/segmentation_m2f/models/utils/assigner.py:18
↓ 2 callersClassCausalAttentionBlock
dinov2/dinov2/layers/block.py:117
↓ 2 callersClassCenterPadding
dinov2/dinov2/hub/utils.py:23
↓ 2 callersClassDepthEncoderDecoder
Encoder Decoder depther. EncoderDecoder typically consists of backbone and decode_head.
dinov2/dinov2/hub/depth/encoder_decoder.py:34
↓ 2 callersClassDropPath
Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).
dinov2/dinov2/eval/segmentation_m2f/models/backbones/drop_path.py:24
↓ 2 callersClassDropPath
Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).
dinov2/dinov2/layers/drop_path.py:26
↓ 2 callersClassEdgeNextConvEncoder
core/submodule.py:561
↓ 2 callersClassFSDPCheckpointer
dinov2/dinov2/fsdp/__init__.py:85
↓ 2 callersClassFoundationStereo
core/foundation_stereo.py:133
↓ 2 callersClassInjector
dinov2/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:136
↓ 2 callersClassMSDeformAttn
dinov2/dinov2/eval/segmentation_m2f/ops/modules/ms_deform_attn.py:63
↓ 2 callersClassMaybeToTensor
Convert a ``PIL Image`` or ``numpy.ndarray`` to tensor, or keep as is if already a tensor.
dinov2/dinov2/data/transforms.py:24
↓ 2 callersClassPreActResidualConvUnit
ResidualConvUnit, pre-activate residual unit. Args: in_channels (int): number of channels in the input feature map. act_cfg (dict)
dinov2/dinov2/eval/depth/models/decode_heads/dpt_head.py:124
↓ 2 callersClassPreActResidualConvUnit
ResidualConvUnit, pre-activate residual unit. Args: in_channels (int): number of channels in the input feature map. act_layer (nn.
dinov2/dinov2/hub/depth/decode_heads.py:600
↓ 2 callersClassRaftConvGRU
core/update.py:82
↓ 2 callersClassResidualConvUnit
Residual convolution module.
depth_anything/blocks.py:37
↓ 2 callersClassSmoothedValue
Track a series of values and provide access to smoothed values over a window or the global series average.
dinov2/dinov2/logging/helpers.py:133
↓ 1 callersClassAdaptivePadding
Applies padding to input (if needed) so that input can get fully covered by filter you specified. It support two modes "same" and "corner". The
dinov2/dinov2/eval/segmentation_m2f/models/utils/transformer.py:35
↓ 1 callersClassAllClassifiers
dinov2/dinov2/eval/linear.py:204
↓ 1 callersClassAttention
dinov2/dinov2/eval/segmentation_m2f/models/backbones/vit.py:132
↓ 1 callersClassAttention
dinov2/dinov2/layers/attention.py:36
↓ 1 callersClassBNHead
Just a batchnorm.
dinov2/dinov2/hub/depth/decode_heads.py:223
↓ 1 callersClassBasicMotionEncoder
core/update.py:50
↓ 1 callersClassBasicSelectiveMultiUpdateBlock
core/update.py:121
↓ 1 callersClassBlock
dinov2/dinov2/eval/segmentation_m2f/models/backbones/vit.py:319
↓ 1 callersClassBlockChunk
dinov2/dinov2/models/vision_transformer.py:38
↓ 1 callersClassChannelAttentionEnhancement
core/submodule.py:528
↓ 1 callersClassCombined_Geo_Encoding_Volume
core/geometry.py:16
↓ 1 callersClassContextNetDino
core/extractor.py:193
↓ 1 callersClassConv2x
core/submodule.py:277
↓ 1 callersClassConvFFN
dinov2/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:49
↓ 1 callersClassCostVolumeDisparityAttention
core/submodule.py:502
↓ 1 callersClassCuSFMDataInference
scripts/run_demo_batch.py:41
↓ 1 callersClassDINOLoss
dinov2/dinov2/loss/dino_clstoken_loss.py:12
↓ 1 callersClassDINOv2Wrapper
dinov2/dinov2/hub/text/dinov2_wrapper.py:6
↓ 1 callersClassDPTHead
depth_anything/dpt.py:22
↓ 1 callersClassDPTHead
Vision Transformers for Dense Prediction. This head is implemented of `DPT <https://arxiv.org/abs/2103.13413>`_. Args: embed_dims (int
dinov2/dinov2/hub/depth/decode_heads.py:690
↓ 1 callersClassDWConv
dinov2/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:70
↓ 1 callersClassDataAugmentationDINO
dinov2/dinov2/data/augmentations.py:19
↓ 1 callersClassDatasetWithEnumeratedTargets
dinov2/dinov2/data/adapters.py:11
↓ 1 callersClassDepthAnything
depth_anything/dpt.py:174
↓ 1 callersClassDepthAnythingFeature
core/extractor.py:287
↓ 1 callersClassDictKeysModule
dinov2/dinov2/eval/knn.py:187
↓ 1 callersClassDinoTxt
dinov2/dinov2/hub/text/dinotxt_model.py:37
↓ 1 callersClassDinoTxtConfig
dinov2/dinov2/hub/text/dinotxt_model.py:14
↓ 1 callersClassDispHead
core/update.py:19
↓ 1 callersClassEpochSampler
dinov2/dinov2/data/samplers.py:17
↓ 1 callersClassFeature
core/extractor.py:324
↓ 1 callersClassFeatureFusionBlock
Feature fusion block.
depth_anything/blocks.py:95
↓ 1 callersClassFeatureFusionBlock
FeatureFusionBlock, merge feature map from different stages. Args: in_channels (int): Input channels. act_cfg (dict): The activati
dinov2/dinov2/eval/depth/models/decode_heads/dpt_head.py:169
↓ 1 callersClassFeatureFusionBlock
FeatureFusionBlock, merge feature map from different stages. Args: in_channels (int): Input channels. act_layer (nn.Module): activ
dinov2/dinov2/hub/depth/decode_heads.py:644
↓ 1 callersClassFlashAttentionTransformerEncoderLayer
core/submodule.py:229
↓ 1 callersClassFlashMultiheadAttention
core/submodule.py:194
↓ 1 callersClassFoundationStereoOnnx
scripts/make_onnx.py:9
↓ 1 callersClassHeadDepth
dinov2/dinov2/eval/depth/models/decode_heads/dpt_head.py:31
↓ 1 callersClassHeadDepth
dinov2/dinov2/hub/depth/decode_heads.py:512
↓ 1 callersClassImageDataDecoder
dinov2/dinov2/data/datasets/decoders.py:17
↓ 1 callersClassImageNetReaLAccuracy
dinov2/dinov2/eval/metrics.py:71
↓ 1 callersClassInfiniteSampler
dinov2/dinov2/data/samplers.py:78
↓ 1 callersClassInterpolate
dinov2/dinov2/eval/depth/models/decode_heads/dpt_head.py:18
↓ 1 callersClassInterpolate
dinov2/dinov2/hub/depth/decode_heads.py:499
↓ 1 callersClassKoLeoLoss
Kozachenko-Leonenko entropic loss regularizer from Sablayrolles et al. - 2018 - Spreading vectors for similarity search
dinov2/dinov2/loss/koleo_loss.py:18
↓ 1 callersClassLinearClassifier
Linear layer to train on top of frozen features
dinov2/dinov2/eval/linear.py:186
↓ 1 callersClassLinearPostprocessor
dinov2/dinov2/eval/linear.py:217
↓ 1 callersClassLogRegModule
dinov2/dinov2/eval/log_regression.py:109
↓ 1 callersClassMaskSamplingResult
Mask sampling result.
dinov2/dinov2/eval/segmentation_m2f/core/box/samplers/mask_sampling_result.py:14
↓ 1 callersClassMaskingGenerator
dinov2/dinov2/data/masking.py:11
↓ 1 callersClassMemEffAttention
dinov2/dinov2/eval/segmentation_m2f/models/backbones/vit.py:159
↓ 1 callersClassMlp
dinov2/dinov2/layers/mlp.py:16
↓ 1 callersClassMlvlPointGenerator
Standard points generator for multi-level (Mlvl) feature maps in 2D points-based detectors. Args: strides (list[int] | list[tuple[int
dinov2/dinov2/eval/segmentation_m2f/core/anchor/point_generator.py:14
↓ 1 callersClassModelWithIntermediateLayers
dinov2/dinov2/eval/utils.py:30
↓ 1 callersClassModelWithNormalize
dinov2/dinov2/eval/utils.py:21
↓ 1 callersClassPositionalEmbedding
core/submodule.py:468
↓ 1 callersClassReassembleBlocks
ViTPostProcessBlock, process cls_token in ViT backbone output and rearrange the feature vector to feature map. Args: in_channels (int)
dinov2/dinov2/eval/depth/models/decode_heads/dpt_head.py:47
↓ 1 callersClassReassembleBlocks
ViTPostProcessBlock, process cls_token in ViT backbone output and rearrange the feature vector to feature map. Args: in_channels (int)
dinov2/dinov2/hub/depth/decode_heads.py:528
↓ 1 callersClassResnetBasicBlock
core/submodule.py:115
↓ 1 callersClassSSLMetaArch
dinov2/dinov2/train/ssl_meta_arch.py:31
↓ 1 callersClassSamplingResult
Bbox sampling result. Example: >>> # xdoctest: +IGNORE_WANT >>> from mmdet.core.bbox.samplers.sampling_result import * # NOQA
dinov2/dinov2/eval/segmentation_m2f/core/box/samplers/sampling_result.py:9
↓ 1 callersClassShardedInfiniteSampler
dinov2/dinov2/data/samplers.py:165
↓ 1 callersClassSpatialAttentionExtractor
core/submodule.py:545
↓ 1 callersClassSpatialPriorModule
dinov2/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:374
next →1–100 of 190, ranked by callers