MCPcopy Create free account

hub / github.com/Tsinghua-MARS-Lab/SLAM-Former / types & classes

Types & classes107 in github.com/Tsinghua-MARS-Lab/SLAM-Former

↓ 9 callersClassLayerScale
src/slamformer/models/dinov2/layers/layer_scale.py:15
↓ 6 callersClassDropPath
Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).
src/slamformer/models/dinov2/layers/drop_path.py:26
↓ 4 callersClassDinoVisionTransformer
src/slamformer/models/dinov2/models/vision_transformer.py:45
↓ 2 callersClassAttention
src/croco/models/blocks.py:96
↓ 2 callersClassBlockRope
src/slamformer/models/layers/block.py:260
↓ 2 callersClassCroCoDownstreamBinocular
src/croco/models/croco_downstream.py:76
↓ 2 callersClassDropPath
Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).
src/croco/models/blocks.py:53
↓ 2 callersClassInterpolate
Interpolation module.
src/croco/models/dpt_block.py:247
↓ 2 callersClassLinearPts3d
Linear head for dust3r Each token outputs: - 16x16 3D points (+ confidence)
src/slamformer/models/layers/transformer_head.py:58
↓ 2 callersClassMlp
MLP as used in Vision Transformer, MLP-Mixer and related networks
src/croco/models/blocks.py:68
↓ 2 callersClassMultiviewHabitatSimGenerator
src/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:144
↓ 2 callersClassPixelwiseTaskWithDPT
DPT module for CroCo. by default, hooks_idx will be equal to: * for encoder-only: 4 equally spread layers * for encoder+decoder: last enco
src/croco/models/head_downstream.py:19
↓ 2 callersClassResidualConvBlock
src/slamformer/models/layers/conv_head.py:25
↓ 2 callersClassResidualConvUnit_custom
Residual convolution module.
src/croco/models/dpt_block.py:84
↓ 2 callersClassSmoothedValue
Track a series of values and provide access to smoothed values.
src/croco/utils/misc.py:32
↓ 2 callersClassTransformerDecoder
src/slamformer/models/layers/transformer_head.py:9
↓ 1 callersClassBackprojectDepth
Layer that projects points from 2D camera to 3D space. The 3D points are represented in homogeneous coordinates.
slam/rerun_helper/geometry_utils.py:21
↓ 1 callersClassBlock
src/croco/models/blocks.py:151
↓ 1 callersClassBlockChunk
src/slamformer/models/dinov2/models/vision_transformer.py:38
↓ 1 callersClassCameraHead
src/slamformer/models/layers/camera_head.py:32
↓ 1 callersClassColorJitterPair
src/croco/datasets/transforms.py:41
↓ 1 callersClassComposePair
src/croco/datasets/transforms.py:12
↓ 1 callersClassConvHead
src/slamformer/models/layers/conv_head.py:62
↓ 1 callersClassCrossAttention
src/croco/models/blocks.py:193
↓ 1 callersClassDPTOutputAdapter
DPT output adapter. :param num_cahnnels: Number of output channels :param stride_level: tride level compared to the full-sized image.
src/croco/models/dpt_block.py:281
↓ 1 callersClassDecoderBlock
src/croco/models/blocks.py:268
↓ 1 callersClassFeatureFusionBlock_custom
Feature fusion block.
src/croco/models/dpt_block.py:150
↓ 1 callersClassFlowAugmentor
src/croco/stereoflow/augmentor.py:238
↓ 1 callersClassFlowDatasetMetrics
src/croco/stereoflow/criterion.py:236
↓ 1 callersClassMaskedMSE
src/croco/models/criterion.py:14
↓ 1 callersClassNoNaviguableSpaceError
src/croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:139
↓ 1 callersClassNormalGenerator
slam/rerun_helper/geometry_utils.py:91
↓ 1 callersClassNormalizeBoth
src/croco/datasets/transforms.py:19
↓ 1 callersClassPairsDataset
src/croco/datasets/pairs_dataset.py:126
↓ 1 callersClassPatchEmbed
just adding _init_weights + position getter compared to timm.models.layers.patch_embed.PatchEmbed
src/croco/models/blocks.py:338
↓ 1 callersClassPositionGetter
return positions of patches
src/slamformer/models/layers/pos_embed.py:164
↓ 1 callersClassPositionGetter
return positions of patches
src/croco/models/blocks.py:323
↓ 1 callersClassRandomCropPair
src/croco/datasets/transforms.py:33
↓ 1 callersClassRandomMask
random masking
src/croco/models/masking.py:13
↓ 1 callersClassResConvBlock
1x1 convolution residual block
src/slamformer/models/layers/camera_head.py:7
↓ 1 callersClassResultViewer
Interactive 3D viewer for pre-saved SLAM results.
slam/visualize_results.py:127
↓ 1 callersClassRoPE2D
src/slamformer/models/layers/pos_embed.py:114
↓ 1 callersClassRoPE2D
src/croco/models/pos_embed.py:126
↓ 1 callersClassSLAM
slam/demo.py:44
↓ 1 callersClassSLAMFormer
src/slamformer/models/slamformer.py:30
↓ 1 callersClassStereoAugmentor
src/croco/stereoflow/augmentor.py:27
↓ 1 callersClassStereoDatasetMetrics
src/croco/stereoflow/criterion.py:177
↓ 1 callersClassToTensorBoth
src/croco/datasets/transforms.py:26
ClassAttention
src/slamformer/models/layers/attention.py:38
ClassAttention
src/slamformer/models/dinov2/layers/attention.py:36
ClassAttentionRope
src/slamformer/models/layers/attention.py:244
ClassBlock
src/slamformer/models/layers/block.py:40
ClassBlock
src/slamformer/models/dinov2/layers/block.py:42
ClassBoosterDataset
src/croco/stereoflow/datasets_stereo.py:587
ClassCREStereoDataset
src/croco/stereoflow/datasets_stereo.py:157
ClassCenterPadding
src/slamformer/models/dinov2/hub/utils.py:23
ClassClusterType
src/slamformer/models/dinov2/utils/cluster.py:12
ClassCosineScheduler
src/slamformer/models/dinov2/utils/utils.py:67
ClassCroCoDownstreamMonocularEncoder
src/croco/models/croco_downstream.py:30
ClassCroCoNet
src/croco/models/croco.py:59
ClassCrocoConfig
src/croco/models/croco.py:24
ClassCrossAttentionRope
src/slamformer/models/layers/attention.py:121
ClassCrossBlockRope
src/slamformer/models/layers/block.py:380
ClassDINOHead
src/slamformer/models/dinov2/layers/dino_head.py:12
ClassETH3DLowResDataset
src/croco/stereoflow/datasets_stereo.py:516
ClassFlashAttention
src/slamformer/models/layers/attention.py:96
ClassFlashAttentionRope
src/slamformer/models/layers/attention.py:360
ClassFlowDataset
src/croco/stereoflow/datasets_flow.py:40
ClassFlowMetrics
src/croco/stereoflow/criterion.py:142
ClassFlyingChairsDataset
src/croco/stereoflow/datasets_flow.py:178
ClassFlyingThingsDataset
src/croco/stereoflow/datasets_flow.py:208
ClassKitti12Dataset
src/croco/stereoflow/datasets_flow.py:536
ClassKitti12Dataset
src/croco/stereoflow/datasets_stereo.py:719
ClassKitti15Dataset
src/croco/stereoflow/datasets_flow.py:582
ClassKitti15Dataset
src/croco/stereoflow/datasets_stereo.py:766
ClassL1Loss
src/croco/stereoflow/criterion.py:23
ClassLaplacianLoss
src/croco/stereoflow/criterion.py:46
ClassLaplacianLossBounded
src/croco/stereoflow/criterion.py:65
ClassLaplacianLossBounded2
src/croco/stereoflow/criterion.py:87
ClassMPISintelDataset
src/croco/stereoflow/datasets_flow.py:307
ClassMd05Dataset
src/croco/stereoflow/datasets_stereo.py:389
ClassMd06Dataset
src/croco/stereoflow/datasets_stereo.py:350
ClassMd14Dataset
src/croco/stereoflow/datasets_stereo.py:312
ClassMd21Dataset
src/croco/stereoflow/datasets_stereo.py:271
ClassMdEval3Dataset
src/croco/stereoflow/datasets_stereo.py:428
ClassMemEffAttention
src/slamformer/models/layers/attention.py:74
ClassMemEffAttention
src/slamformer/models/dinov2/layers/attention.py:72
ClassMemEffAttentionRope
src/slamformer/models/layers/attention.py:302
ClassMemEffCrossAttentionRope
src/slamformer/models/layers/attention.py:196
ClassMetricLogger
src/croco/utils/misc.py:91
ClassMlp
src/slamformer/models/dinov2/layers/mlp.py:16
ClassNativeScalerWithGradNormCount
src/croco/utils/misc.py:284
ClassNestedTensorBlock
src/slamformer/models/layers/block.py:209
ClassNestedTensorBlock
src/slamformer/models/dinov2/layers/block.py:210
ClassPatchEmbed
2D image to patch embedding: (B,C,H,W) -> (B,N,D) Args: img_size: Image size. patch_size: Patch token size. in_chans
src/slamformer/models/dinov2/layers/patch_embed.py:25
ClassProject3D
Layer that projects 3D points into the 2D camera
slam/rerun_helper/geometry_utils.py:61
ClassSceneFlowDataset
src/croco/stereoflow/datasets_stereo.py:187
ClassSpringDataset
src/croco/stereoflow/datasets_flow.py:399
ClassSpringDataset
src/croco/stereoflow/datasets_stereo.py:639
ClassStereoDataset
src/croco/stereoflow/datasets_stereo.py:55
next →1–100 of 107, ranked by callers