MCPcopy Create free account

hub / github.com/Michel-liu/GroupPose / types & classes

Types & classes66 in github.com/Michel-liu/GroupPose

↓ 6 callersClassNestedTensor
util/misc.py:297
↓ 4 callersClassConfigDict
util/slconfig.py:25
↓ 3 callersClassBestMetricSingle
util/utils.py:398
↓ 3 callersClassMLP
Very simple multi-layer perceptron (also called FFN)
models/grouppose/utils.py:87
↓ 3 callersClassModelEma
util/utils.py:373
↓ 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:43
↓ 2 callersClassCocoEvaluator
datasets/crowdpose_eval.py:9
↓ 2 callersClassMSDeformAttn
models/grouppose/ops/modules/ms_deform_attn.py:30
↓ 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
↓ 1 callersClassBackbone
ResNet backbone with frozen BatchNorm.
models/grouppose/backbones/backbone.py:92
↓ 1 callersClassBasicLayer
A basic Swin Transformer layer for one stage. Args: dim (int): Number of feature channels depth (int): Depths of this stage.
models/grouppose/backbones/swin_transformer.py:291
↓ 1 callersClassBestMetricHolder
util/utils.py:433
↓ 1 callersClassCocoDetection
datasets/crowdpose.py:13
↓ 1 callersClassCocoDetection
datasets/coco.py:21
↓ 1 callersClassConvertCocoPolysToMask
datasets/crowdpose.py:73
↓ 1 callersClassConvertCocoPolysToMask
datasets/coco.py:82
↓ 1 callersClassDeformableTransformerDecoderLayer
models/grouppose/transformer_deformable.py:51
↓ 1 callersClassDeformableTransformerEncoderLayer
models/grouppose/transformer_deformable.py:16
↓ 1 callersClassEmbedder
util/utils.py:123
↓ 1 callersClassGroupPose
This is the Cross-Attention Detector module that performs object detection
models/grouppose/grouppose.py:35
↓ 1 callersClassHungarianMatcher
models/grouppose/matcher.py:21
↓ 1 callersClassJoiner
models/grouppose/backbones/backbone.py:114
↓ 1 callersClassMlp
Multilayer perceptron.
models/grouppose/backbones/swin_transformer.py:18
↓ 1 callersClassOKSLoss
util/keypoint_loss.py:37
↓ 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/grouppose/backbones/swin_transformer.py:393
↓ 1 callersClassPositionEmbeddingLearned
Absolute pos embedding, learned.
models/grouppose/position_encoding.py:102
↓ 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/grouppose/position_encoding.py:58
↓ 1 callersClassPostProcess
This module converts the model's output into the format expected by the coco api
models/grouppose/postprocesses.py:6
↓ 1 callersClassRegistry
models/registry.py:5
↓ 1 callersClassSetCriterion
models/grouppose/criterion.py:10
↓ 1 callersClassSwinTransformer
Swin Transformer backbone. A PyTorch impl of : `Swin Transformer: Hierarchical Vision Transformer using Shifted Windows` - https:/
models/grouppose/backbones/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/grouppose/backbones/swin_transformer.py:150
↓ 1 callersClassTransformer
models/grouppose/transformer.py:21
↓ 1 callersClassTransformerDecoder
models/grouppose/transformer.py:250
↓ 1 callersClassTransformerEncoder
models/grouppose/transformer.py:200
↓ 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/grouppose/backbones/swin_transformer.py:69
↓ 1 callersClass_ColorfulFormatter
util/logger.py:9
ClassAPOPMeter
util/utils.py:174
ClassBackboneBase
models/grouppose/backbones/backbone.py:68
ClassCocoClassMapper
util/utils.py:52
ClassCocoEvaluator
datasets/coco_eval.py:22
ClassCompose
datasets/transforms_crowdpose.py:274
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:399
ClassFrozenBatchNorm2d
BatchNorm2d where the batch statistics and the affine parameters are fixed. Copy-paste from torchvision.misc.ops with added eps before rqsrt
models/grouppose/backbones/backbone.py:29
ClassMSDeformAttn
models/grouppose/ops/modules/ms_deform_attn_key_aware.py:30
ClassMSDeformAttnFunction
models/grouppose/ops/functions/ms_deform_attn_func.py:21
ClassMetricLogger
util/misc.py:166
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_crowdpose.py:239
ClassNormalize
datasets/transforms.py:236
ClassPatchMerging
Patch Merging Layer Args: dim (int): Number of input channels. norm_layer (nn.Module, optional): Normalization layer. Default: n
models/grouppose/backbones/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/grouppose/position_encoding.py:20
ClassRandomCrop
datasets/transforms_crowdpose.py:176
ClassRandomCrop
datasets/transforms.py:173
ClassRandomHorizontalFlip
datasets/transforms_crowdpose.py:197
ClassRandomHorizontalFlip
datasets/transforms.py:194
ClassRandomResize
datasets/transforms_crowdpose.py:207
ClassRandomResize
datasets/transforms.py:204
ClassRandomSelect
Randomly selects between transforms1 and transforms2, with probability p for transforms1 and (1 - p) for transforms2
datasets/transforms_crowdpose.py:218
ClassRandomSelect
Randomly selects between transforms1 and transforms2, with probability p for transforms1 and (1 - p) for transforms2
datasets/transforms.py:215
ClassRandomSizeCrop
datasets/transforms_crowdpose.py:185
ClassRandomSizeCrop
datasets/transforms.py:182
ClassToTensor
datasets/transforms_crowdpose.py:234
ClassToTensor
datasets/transforms.py:231
Classcolor_sys
util/misc.py:553