Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/NVlabs/FoundationStereo
/ types & classes
Types & classes
190 in github.com/NVlabs/FoundationStereo
⨍
Functions
881
◇
Types & classes
190
↳
Endpoints
1
↓ 17 callers
Class
BasicConv
core/submodule.py:47
↓ 15 callers
Class
ConvModule
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 callers
Class
Conv3dNormActReduced
core/submodule.py:85
↓ 8 callers
Class
ResidualBlock
core/extractor.py:21
↓ 6 callers
Class
FeatureAtt
core/submodule.py:434
↓ 6 callers
Class
LayerNorm2d
r""" https://huggingface.co/spaces/Roll20/pet_score/blob/b258ef28152ab0d5b377d9142a23346f863c1526/lib/timm/models/convnext.py#L85 LayerNorm for c
core/submodule.py:25
↓ 5 callers
Class
BasicConv_IN
core/submodule.py:316
↓ 5 callers
Class
CosineScheduler
dinov2/dinov2/utils/utils.py:67
↓ 4 callers
Class
DinoVisionTransformer
dinov2/dinov2/models/vision_transformer.py:45
↓ 4 callers
Class
Extractor
dinov2/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:88
↓ 4 callers
Class
InputPadder
Pads images such that dimensions are divisible by 8
core/utils/utils.py:17
↓ 4 callers
Class
LayerScale
dinov2/dinov2/layers/layer_scale.py:15
↓ 4 callers
Class
MetricLogger
dinov2/dinov2/logging/helpers.py:20
↓ 3 callers
Class
Conv2x_IN
core/submodule.py:345
↓ 3 callers
Class
GaussianBlur
Apply Gaussian Blur to the PIL image.
dinov2/dinov2/data/transforms.py:12
↓ 3 callers
Class
ModuleDictWithForward
dinov2/dinov2/eval/knn.py:240
↓ 3 callers
Class
ResnetBasicBlock3D
core/submodule.py:155
↓ 3 callers
Class
SelectiveConvGRU
core/update.py:97
↓ 3 callers
Class
autocast
core/foundation_stereo.py:34
↓ 2 callers
Class
AssignResult
Collection of assign results.
dinov2/dinov2/eval/segmentation_m2f/models/utils/assigner.py:18
↓ 2 callers
Class
CausalAttentionBlock
dinov2/dinov2/layers/block.py:117
↓ 2 callers
Class
CenterPadding
dinov2/dinov2/hub/utils.py:23
↓ 2 callers
Class
DepthEncoderDecoder
Encoder Decoder depther. EncoderDecoder typically consists of backbone and decode_head.
dinov2/dinov2/hub/depth/encoder_decoder.py:34
↓ 2 callers
Class
DropPath
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 callers
Class
DropPath
Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).
dinov2/dinov2/layers/drop_path.py:26
↓ 2 callers
Class
EdgeNextConvEncoder
core/submodule.py:561
↓ 2 callers
Class
FSDPCheckpointer
dinov2/dinov2/fsdp/__init__.py:85
↓ 2 callers
Class
FoundationStereo
core/foundation_stereo.py:133
↓ 2 callers
Class
Injector
dinov2/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:136
↓ 2 callers
Class
MSDeformAttn
dinov2/dinov2/eval/segmentation_m2f/ops/modules/ms_deform_attn.py:63
↓ 2 callers
Class
MaybeToTensor
Convert a ``PIL Image`` or ``numpy.ndarray`` to tensor, or keep as is if already a tensor.
dinov2/dinov2/data/transforms.py:24
↓ 2 callers
Class
PreActResidualConvUnit
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 callers
Class
PreActResidualConvUnit
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 callers
Class
RaftConvGRU
core/update.py:82
↓ 2 callers
Class
ResidualConvUnit
Residual convolution module.
depth_anything/blocks.py:37
↓ 2 callers
Class
SmoothedValue
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 callers
Class
AdaptivePadding
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 callers
Class
AllClassifiers
dinov2/dinov2/eval/linear.py:204
↓ 1 callers
Class
Attention
dinov2/dinov2/eval/segmentation_m2f/models/backbones/vit.py:132
↓ 1 callers
Class
Attention
dinov2/dinov2/layers/attention.py:36
↓ 1 callers
Class
BNHead
Just a batchnorm.
dinov2/dinov2/hub/depth/decode_heads.py:223
↓ 1 callers
Class
BasicMotionEncoder
core/update.py:50
↓ 1 callers
Class
BasicSelectiveMultiUpdateBlock
core/update.py:121
↓ 1 callers
Class
Block
dinov2/dinov2/eval/segmentation_m2f/models/backbones/vit.py:319
↓ 1 callers
Class
BlockChunk
dinov2/dinov2/models/vision_transformer.py:38
↓ 1 callers
Class
ChannelAttentionEnhancement
core/submodule.py:528
↓ 1 callers
Class
Combined_Geo_Encoding_Volume
core/geometry.py:16
↓ 1 callers
Class
ContextNetDino
core/extractor.py:193
↓ 1 callers
Class
Conv2x
core/submodule.py:277
↓ 1 callers
Class
ConvFFN
dinov2/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:49
↓ 1 callers
Class
CostVolumeDisparityAttention
core/submodule.py:502
↓ 1 callers
Class
CuSFMDataInference
scripts/run_demo_batch.py:41
↓ 1 callers
Class
DINOLoss
dinov2/dinov2/loss/dino_clstoken_loss.py:12
↓ 1 callers
Class
DINOv2Wrapper
dinov2/dinov2/hub/text/dinov2_wrapper.py:6
↓ 1 callers
Class
DPTHead
depth_anything/dpt.py:22
↓ 1 callers
Class
DPTHead
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 callers
Class
DWConv
dinov2/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:70
↓ 1 callers
Class
DataAugmentationDINO
dinov2/dinov2/data/augmentations.py:19
↓ 1 callers
Class
DatasetWithEnumeratedTargets
dinov2/dinov2/data/adapters.py:11
↓ 1 callers
Class
DepthAnything
depth_anything/dpt.py:174
↓ 1 callers
Class
DepthAnythingFeature
core/extractor.py:287
↓ 1 callers
Class
DictKeysModule
dinov2/dinov2/eval/knn.py:187
↓ 1 callers
Class
DinoTxt
dinov2/dinov2/hub/text/dinotxt_model.py:37
↓ 1 callers
Class
DinoTxtConfig
dinov2/dinov2/hub/text/dinotxt_model.py:14
↓ 1 callers
Class
DispHead
core/update.py:19
↓ 1 callers
Class
EpochSampler
dinov2/dinov2/data/samplers.py:17
↓ 1 callers
Class
Feature
core/extractor.py:324
↓ 1 callers
Class
FeatureFusionBlock
Feature fusion block.
depth_anything/blocks.py:95
↓ 1 callers
Class
FeatureFusionBlock
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 callers
Class
FeatureFusionBlock
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 callers
Class
FlashAttentionTransformerEncoderLayer
core/submodule.py:229
↓ 1 callers
Class
FlashMultiheadAttention
core/submodule.py:194
↓ 1 callers
Class
FoundationStereoOnnx
scripts/make_onnx.py:9
↓ 1 callers
Class
HeadDepth
dinov2/dinov2/eval/depth/models/decode_heads/dpt_head.py:31
↓ 1 callers
Class
HeadDepth
dinov2/dinov2/hub/depth/decode_heads.py:512
↓ 1 callers
Class
ImageDataDecoder
dinov2/dinov2/data/datasets/decoders.py:17
↓ 1 callers
Class
ImageNetReaLAccuracy
dinov2/dinov2/eval/metrics.py:71
↓ 1 callers
Class
InfiniteSampler
dinov2/dinov2/data/samplers.py:78
↓ 1 callers
Class
Interpolate
dinov2/dinov2/eval/depth/models/decode_heads/dpt_head.py:18
↓ 1 callers
Class
Interpolate
dinov2/dinov2/hub/depth/decode_heads.py:499
↓ 1 callers
Class
KoLeoLoss
Kozachenko-Leonenko entropic loss regularizer from Sablayrolles et al. - 2018 - Spreading vectors for similarity search
dinov2/dinov2/loss/koleo_loss.py:18
↓ 1 callers
Class
LinearClassifier
Linear layer to train on top of frozen features
dinov2/dinov2/eval/linear.py:186
↓ 1 callers
Class
LinearPostprocessor
dinov2/dinov2/eval/linear.py:217
↓ 1 callers
Class
LogRegModule
dinov2/dinov2/eval/log_regression.py:109
↓ 1 callers
Class
MaskSamplingResult
Mask sampling result.
dinov2/dinov2/eval/segmentation_m2f/core/box/samplers/mask_sampling_result.py:14
↓ 1 callers
Class
MaskingGenerator
dinov2/dinov2/data/masking.py:11
↓ 1 callers
Class
MemEffAttention
dinov2/dinov2/eval/segmentation_m2f/models/backbones/vit.py:159
↓ 1 callers
Class
Mlp
dinov2/dinov2/layers/mlp.py:16
↓ 1 callers
Class
MlvlPointGenerator
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 callers
Class
ModelWithIntermediateLayers
dinov2/dinov2/eval/utils.py:30
↓ 1 callers
Class
ModelWithNormalize
dinov2/dinov2/eval/utils.py:21
↓ 1 callers
Class
PositionalEmbedding
core/submodule.py:468
↓ 1 callers
Class
ReassembleBlocks
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 callers
Class
ReassembleBlocks
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 callers
Class
ResnetBasicBlock
core/submodule.py:115
↓ 1 callers
Class
SSLMetaArch
dinov2/dinov2/train/ssl_meta_arch.py:31
↓ 1 callers
Class
SamplingResult
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 callers
Class
ShardedInfiniteSampler
dinov2/dinov2/data/samplers.py:165
↓ 1 callers
Class
SpatialAttentionExtractor
core/submodule.py:545
↓ 1 callers
Class
SpatialPriorModule
dinov2/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:374
next →
1–100 of 190, ranked by callers