Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/IDEA-Research/DINO
/ types & classes
Types & classes
105 in github.com/IDEA-Research/DINO
⨍
Functions
501
◇
Types & classes
105
↓ 7 callers
Class
MLP
Very simple multi-layer perceptron (also called FFN)
models/dino/utils.py:107
↓ 7 callers
Class
NestedTensor
util/misc.py:301
↓ 6 callers
Class
MSDeformAttn
models/dino/ops/modules/ms_deform_attn.py:30
↓ 4 callers
Class
ConfigDict
util/slconfig.py:28
↓ 4 callers
Class
label2compat
datasets/coco.py:29
↓ 4 callers
Class
label_compat2onehot
datasets/coco.py:46
↓ 3 callers
Class
BboxPertuber
datasets/coco.py:189
↓ 3 callers
Class
BestMetricSingle
util/utils.py:398
↓ 3 callers
Class
LayerNorm
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 callers
Class
ModelEma
util/utils.py:373
↓ 3 callers
Class
RandomSelectBoxlabels
datasets/coco.py:90
↓ 3 callers
Class
SLConfig
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 callers
Class
box_label_catter
datasets/coco.py:75
↓ 2 callers
Class
CocoDetection
datasets/coco.py:323
↓ 2 callers
Class
PanopticEvaluator
datasets/panoptic_eval.py:13
↓ 2 callers
Class
PickleHandler
util/slio.py:51
↓ 2 callers
Class
SmoothedValue
Track a series of values and provide access to smoothed values over a window or the global series average.
util/misc.py:32
↓ 2 callers
Class
YamlHandler
util/slio.py:72
↓ 1 callers
Class
AverageMeter
Computes and stores the average and current value
util/time_counter.py:35
↓ 1 callers
Class
Backbone
ResNet backbone with frozen BatchNorm.
models/dino/backbone.py:109
↓ 1 callers
Class
BasicLayer
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 callers
Class
BestMetricHolder
util/utils.py:433
↓ 1 callers
Class
Block
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 callers
Class
CocoEvaluator
datasets/coco_eval.py:22
↓ 1 callers
Class
CocoPanoptic
datasets/coco_panoptic.py:15
↓ 1 callers
Class
ConvNeXt
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 callers
Class
ConvertCocoPolysToMask
datasets/coco.py:389
↓ 1 callers
Class
DETRsegm
models/dino/segmentation.py:36
↓ 1 callers
Class
DINO
This is the Cross-Attention Detector module that performs object detection
models/dino/dino.py:38
↓ 1 callers
Class
DeformableTransformer
models/dino/transformer_deformable.py:29
↓ 1 callers
Class
DeformableTransformer
models/dino/deformable_transformer.py:25
↓ 1 callers
Class
DeformableTransformerDecoder
models/dino/transformer_deformable.py:516
↓ 1 callers
Class
DeformableTransformerDecoderLayer
models/dino/transformer_deformable.py:331
↓ 1 callers
Class
DeformableTransformerDecoderLayer
models/dino/deformable_transformer.py:822
↓ 1 callers
Class
DeformableTransformerEncoder
models/dino/transformer_deformable.py:279
↓ 1 callers
Class
DeformableTransformerEncoderLayer
models/dino/transformer_deformable.py:219
↓ 1 callers
Class
DeformableTransformerEncoderLayer
models/dino/deformable_transformer.py:765
↓ 1 callers
Class
DummyArgs
tools/benchmark.py:570
↓ 1 callers
Class
Embedder
util/utils.py:123
↓ 1 callers
Class
Error
datasets/data_util.py:11
↓ 1 callers
Class
HungarianMatcher
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 callers
Class
Joiner
models/dino/backbone.py:131
↓ 1 callers
Class
JsonHandler
util/slio.py:40
↓ 1 callers
Class
MHAttentionMap
This is a 2D attention module, which only returns the attention softmax (no multiplication by value)
models/dino/segmentation.py:152
↓ 1 callers
Class
MaskCrop
datasets/coco.py:288
↓ 1 callers
Class
MaskHeadSmallConv
Simple convolutional head, using group norm. Upsampling is done using a FPN approach
models/dino/segmentation.py:81
↓ 1 callers
Class
Mlp
Multilayer perceptron.
models/dino/swin_transformer.py:18
↓ 1 callers
Class
PatchEmbed
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 callers
Class
PositionEmbeddingLearned
Absolute pos embedding, learned.
models/dino/position_encoding.py:110
↓ 1 callers
Class
PositionEmbeddingSineHW
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 callers
Class
PostProcess
This module converts the model's output into the format expected by the coco api
models/dino/dino.py:639
↓ 1 callers
Class
PostProcessPanoptic
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 callers
Class
PostProcessSegm
models/dino/segmentation.py:230
↓ 1 callers
Class
RandomBoxPerturber
models/dino/utils.py:64
↓ 1 callers
Class
RandomCutout
datasets/coco.py:220
↓ 1 callers
Class
RandomDrop
datasets/coco.py:176
↓ 1 callers
Class
RandomSelectBoxes
datasets/coco.py:241
↓ 1 callers
Class
Registry
models/registry.py:12
↓ 1 callers
Class
SetCriterion
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 callers
Class
SimpleMinsumMatcher
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 callers
Class
SwinTransformer
Swin Transformer backbone. A PyTorch impl of : `Swin Transformer: Hierarchical Vision Transformer using Shifted Windows` - https:/
models/dino/swin_transformer.py:435
↓ 1 callers
Class
SwinTransformerBlock
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 callers
Class
Trainer
run_with_submitit.py:51
↓ 1 callers
Class
TransformerDecoder
models/dino/deformable_transformer.py:579
↓ 1 callers
Class
TransformerEncoder
models/dino/deformable_transformer.py:434
↓ 1 callers
Class
WindowAttention
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 callers
Class
_ColorfulFormatter
util/logger.py:9
↓ 1 callers
Class
color_sys
util/misc.py:561
Class
APOPMeter
util/utils.py:174
Class
AdjustBrightness
datasets/sltransform.py:26
Class
AdjustContrast
datasets/sltransform.py:14
Class
Albumentations
datasets/sltransform.py:216
Class
BackboneBase
models/dino/backbone.py:75
Class
BaseFileHandler
util/slio.py:18
Class
COCOVisualizer
util/visualizer.py:58
Class
CenterCrop
datasets/transforms.py:178
Class
CocoClassMapper
util/utils.py:52
Class
ColorMap
util/visualizer.py:43
Class
Compose
datasets/transforms.py:270
Class
DictAction
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
Class
FrozenBatchNorm2d
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
Class
LightingNoise
datasets/sltransform.py:52
Class
MSDeformAttnFunction
models/dino/ops/functions/ms_deform_attn_func.py:21
Class
MetricLogger
util/misc.py:166
Class
MultiheadAttention
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
Class
NiceRepr
Inherit from this class and define ``__nice__`` to "nicely" print your objects. Defines ``__str__`` and ``__repr__`` in terms of ``__nice__``
util/utils.py:239
Class
Normalize
datasets/transforms.py:251
Class
PatchMerging
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
Class
PositionEmbeddingSine
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
Class
RandomCrop
datasets/transforms.py:157
Class
RandomCrop
datasets/sltransform.py:171
Class
RandomCropDebug
datasets/sltransform.py:188
Class
RandomErasing
datasets/transforms.py:242
Class
RandomHorizontalFlip
datasets/transforms.py:190
Class
RandomPad
datasets/transforms.py:211
Class
RandomResize
datasets/transforms.py:200
Class
RandomSelect
Randomly selects between transforms1 and transforms2, with probability p for transforms1 and (1 - p) for transforms2
datasets/transforms.py:221
Class
RandomSelectMulti
Randomly selects between transforms1 and transforms2,
datasets/sltransform.py:202
Class
RandomSizeCrop
datasets/transforms.py:166
Class
ResizeDebug
datasets/transforms.py:149
next →
1–100 of 105, ranked by callers