MCPcopy Create free account

hub / github.com/IDEA-Research/DINO / types & classes

Types & classes105 in github.com/IDEA-Research/DINO

↓ 7 callersClassMLP
Very simple multi-layer perceptron (also called FFN)
models/dino/utils.py:107
↓ 7 callersClassNestedTensor
util/misc.py:301
↓ 6 callersClassMSDeformAttn
models/dino/ops/modules/ms_deform_attn.py:30
↓ 4 callersClassConfigDict
util/slconfig.py:28
↓ 4 callersClasslabel2compat
datasets/coco.py:29
↓ 4 callersClasslabel_compat2onehot
datasets/coco.py:46
↓ 3 callersClassBboxPertuber
datasets/coco.py:189
↓ 3 callersClassBestMetricSingle
util/utils.py:398
↓ 3 callersClassLayerNorm
r""" LayerNorm that supports two data formats: channels_last (default) or channels_first. The ordering of the dimensions in the inputs. channels_
models/dino/convnext.py:152
↓ 3 callersClassModelEma
util/utils.py:373
↓ 3 callersClassRandomSelectBoxlabels
datasets/coco.py:90
↓ 3 callersClassSLConfig
config files. only support .py file as config now. ref: mmcv.utils.config Example: >>> cfg = Config(dict(a=1, b=dict(b1=[0,
util/slconfig.py:46
↓ 3 callersClassbox_label_catter
datasets/coco.py:75
↓ 2 callersClassCocoDetection
datasets/coco.py:323
↓ 2 callersClassPanopticEvaluator
datasets/panoptic_eval.py:13
↓ 2 callersClassPickleHandler
util/slio.py:51
↓ 2 callersClassSmoothedValue
Track a series of values and provide access to smoothed values over a window or the global series average.
util/misc.py:32
↓ 2 callersClassYamlHandler
util/slio.py:72
↓ 1 callersClassAverageMeter
Computes and stores the average and current value
util/time_counter.py:35
↓ 1 callersClassBackbone
ResNet backbone with frozen BatchNorm.
models/dino/backbone.py:109
↓ 1 callersClassBasicLayer
A basic Swin Transformer layer for one stage. Args: dim (int): Number of feature channels depth (int): Depths of this stage.
models/dino/swin_transformer.py:291
↓ 1 callersClassBestMetricHolder
util/utils.py:433
↓ 1 callersClassBlock
r""" ConvNeXt Block. There are two equivalent implementations: (1) DwConv -> LayerNorm (channels_first) -> 1x1 Conv -> GELU -> 1x1 Conv; all in (N
models/dino/convnext.py:18
↓ 1 callersClassCocoEvaluator
datasets/coco_eval.py:22
↓ 1 callersClassCocoPanoptic
datasets/coco_panoptic.py:15
↓ 1 callersClassConvNeXt
r""" ConvNeXt A PyTorch impl of : `A ConvNet for the 2020s` - https://arxiv.org/pdf/2201.03545.pdf Args: in_chans (int
models/dino/convnext.py:55
↓ 1 callersClassConvertCocoPolysToMask
datasets/coco.py:389
↓ 1 callersClassDETRsegm
models/dino/segmentation.py:36
↓ 1 callersClassDINO
This is the Cross-Attention Detector module that performs object detection
models/dino/dino.py:38
↓ 1 callersClassDeformableTransformer
models/dino/transformer_deformable.py:29
↓ 1 callersClassDeformableTransformer
models/dino/deformable_transformer.py:25
↓ 1 callersClassDeformableTransformerDecoder
models/dino/transformer_deformable.py:516
↓ 1 callersClassDeformableTransformerDecoderLayer
models/dino/transformer_deformable.py:331
↓ 1 callersClassDeformableTransformerDecoderLayer
models/dino/deformable_transformer.py:822
↓ 1 callersClassDeformableTransformerEncoder
models/dino/transformer_deformable.py:279
↓ 1 callersClassDeformableTransformerEncoderLayer
models/dino/transformer_deformable.py:219
↓ 1 callersClassDeformableTransformerEncoderLayer
models/dino/deformable_transformer.py:765
↓ 1 callersClassDummyArgs
tools/benchmark.py:570
↓ 1 callersClassEmbedder
util/utils.py:123
↓ 1 callersClassError
datasets/data_util.py:11
↓ 1 callersClassHungarianMatcher
This class computes an assignment between the targets and the predictions of the network For efficiency reasons, the targets don't include the no_
models/dino/matcher.py:25
↓ 1 callersClassJoiner
models/dino/backbone.py:131
↓ 1 callersClassJsonHandler
util/slio.py:40
↓ 1 callersClassMHAttentionMap
This is a 2D attention module, which only returns the attention softmax (no multiplication by value)
models/dino/segmentation.py:152
↓ 1 callersClassMaskCrop
datasets/coco.py:288
↓ 1 callersClassMaskHeadSmallConv
Simple convolutional head, using group norm. Upsampling is done using a FPN approach
models/dino/segmentation.py:81
↓ 1 callersClassMlp
Multilayer perceptron.
models/dino/swin_transformer.py:18
↓ 1 callersClassPatchEmbed
Image to Patch Embedding Args: patch_size (int): Patch token size. Default: 4. in_chans (int): Number of input image channels. De
models/dino/swin_transformer.py:393
↓ 1 callersClassPositionEmbeddingLearned
Absolute pos embedding, learned.
models/dino/position_encoding.py:110
↓ 1 callersClassPositionEmbeddingSineHW
This is a more standard version of the position embedding, very similar to the one used by the Attention is all you need paper, generalized t
models/dino/position_encoding.py:62
↓ 1 callersClassPostProcess
This module converts the model's output into the format expected by the coco api
models/dino/dino.py:639
↓ 1 callersClassPostProcessPanoptic
This class converts the output of the model to the final panoptic result, in the format expected by the coco panoptic API
models/dino/segmentation.py:253
↓ 1 callersClassPostProcessSegm
models/dino/segmentation.py:230
↓ 1 callersClassRandomBoxPerturber
models/dino/utils.py:64
↓ 1 callersClassRandomCutout
datasets/coco.py:220
↓ 1 callersClassRandomDrop
datasets/coco.py:176
↓ 1 callersClassRandomSelectBoxes
datasets/coco.py:241
↓ 1 callersClassRegistry
models/registry.py:12
↓ 1 callersClassSetCriterion
This class computes the loss for Conditional DETR. The process happens in two steps: 1) we compute hungarian assignment between ground tr
models/dino/dino.py:333
↓ 1 callersClassSimpleMinsumMatcher
This class computes an assignment between the targets and the predictions of the network For efficiency reasons, the targets don't include the no_
models/dino/matcher.py:98
↓ 1 callersClassSwinTransformer
Swin Transformer backbone. A PyTorch impl of : `Swin Transformer: Hierarchical Vision Transformer using Shifted Windows` - https:/
models/dino/swin_transformer.py:435
↓ 1 callersClassSwinTransformerBlock
Swin Transformer Block. Args: dim (int): Number of input channels. num_heads (int): Number of attention heads. window_siz
models/dino/swin_transformer.py:150
↓ 1 callersClassTrainer
run_with_submitit.py:51
↓ 1 callersClassTransformerDecoder
models/dino/deformable_transformer.py:579
↓ 1 callersClassTransformerEncoder
models/dino/deformable_transformer.py:434
↓ 1 callersClassWindowAttention
Window based multi-head self attention (W-MSA) module with relative position bias. It supports both of shifted and non-shifted window. Args:
models/dino/swin_transformer.py:69
↓ 1 callersClass_ColorfulFormatter
util/logger.py:9
↓ 1 callersClasscolor_sys
util/misc.py:561
ClassAPOPMeter
util/utils.py:174
ClassAdjustBrightness
datasets/sltransform.py:26
ClassAdjustContrast
datasets/sltransform.py:14
ClassAlbumentations
datasets/sltransform.py:216
ClassBackboneBase
models/dino/backbone.py:75
ClassBaseFileHandler
util/slio.py:18
ClassCOCOVisualizer
util/visualizer.py:58
ClassCenterCrop
datasets/transforms.py:178
ClassCocoClassMapper
util/utils.py:52
ClassColorMap
util/visualizer.py:43
ClassCompose
datasets/transforms.py:270
ClassDictAction
argparse action to split an argument into KEY=VALUE form on the first = and append to a dictionary. List options should be passed as comm
util/slconfig.py:408
ClassFrozenBatchNorm2d
BatchNorm2d where the batch statistics and the affine parameters are fixed. Copy-paste from torchvision.misc.ops with added eps before rqsrt
models/dino/backbone.py:36
ClassLightingNoise
datasets/sltransform.py:52
ClassMSDeformAttnFunction
models/dino/ops/functions/ms_deform_attn_func.py:21
ClassMetricLogger
util/misc.py:166
ClassMultiheadAttention
r"""Allows the model to jointly attend to information from different representation subspaces. See reference: Attention Is All You Need ..
models/dino/attention.py:59
ClassNiceRepr
Inherit from this class and define ``__nice__`` to "nicely" print your objects. Defines ``__str__`` and ``__repr__`` in terms of ``__nice__``
util/utils.py:239
ClassNormalize
datasets/transforms.py:251
ClassPatchMerging
Patch Merging Layer Args: dim (int): Number of input channels. norm_layer (nn.Module, optional): Normalization layer. Default: n
models/dino/swin_transformer.py:250
ClassPositionEmbeddingSine
This is a more standard version of the position embedding, very similar to the one used by the Attention is all you need paper, generalized t
models/dino/position_encoding.py:24
ClassRandomCrop
datasets/transforms.py:157
ClassRandomCrop
datasets/sltransform.py:171
ClassRandomCropDebug
datasets/sltransform.py:188
ClassRandomErasing
datasets/transforms.py:242
ClassRandomHorizontalFlip
datasets/transforms.py:190
ClassRandomPad
datasets/transforms.py:211
ClassRandomResize
datasets/transforms.py:200
ClassRandomSelect
Randomly selects between transforms1 and transforms2, with probability p for transforms1 and (1 - p) for transforms2
datasets/transforms.py:221
ClassRandomSelectMulti
Randomly selects between transforms1 and transforms2,
datasets/sltransform.py:202
ClassRandomSizeCrop
datasets/transforms.py:166
ClassResizeDebug
datasets/transforms.py:149
next →1–100 of 105, ranked by callers