Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Michel-liu/GroupPose
/ types & classes
Types & classes
66 in github.com/Michel-liu/GroupPose
⨍
Functions
340
◇
Types & classes
66
↓ 6 callers
Class
NestedTensor
util/misc.py:297
↓ 4 callers
Class
ConfigDict
util/slconfig.py:25
↓ 3 callers
Class
BestMetricSingle
util/utils.py:398
↓ 3 callers
Class
MLP
Very simple multi-layer perceptron (also called FFN)
models/grouppose/utils.py:87
↓ 3 callers
Class
ModelEma
util/utils.py:373
↓ 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:43
↓ 2 callers
Class
CocoEvaluator
datasets/crowdpose_eval.py:9
↓ 2 callers
Class
MSDeformAttn
models/grouppose/ops/modules/ms_deform_attn.py:30
↓ 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
↓ 1 callers
Class
Backbone
ResNet backbone with frozen BatchNorm.
models/grouppose/backbones/backbone.py:92
↓ 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/grouppose/backbones/swin_transformer.py:291
↓ 1 callers
Class
BestMetricHolder
util/utils.py:433
↓ 1 callers
Class
CocoDetection
datasets/crowdpose.py:13
↓ 1 callers
Class
CocoDetection
datasets/coco.py:21
↓ 1 callers
Class
ConvertCocoPolysToMask
datasets/crowdpose.py:73
↓ 1 callers
Class
ConvertCocoPolysToMask
datasets/coco.py:82
↓ 1 callers
Class
DeformableTransformerDecoderLayer
models/grouppose/transformer_deformable.py:51
↓ 1 callers
Class
DeformableTransformerEncoderLayer
models/grouppose/transformer_deformable.py:16
↓ 1 callers
Class
Embedder
util/utils.py:123
↓ 1 callers
Class
GroupPose
This is the Cross-Attention Detector module that performs object detection
models/grouppose/grouppose.py:35
↓ 1 callers
Class
HungarianMatcher
models/grouppose/matcher.py:21
↓ 1 callers
Class
Joiner
models/grouppose/backbones/backbone.py:114
↓ 1 callers
Class
Mlp
Multilayer perceptron.
models/grouppose/backbones/swin_transformer.py:18
↓ 1 callers
Class
OKSLoss
util/keypoint_loss.py:37
↓ 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/grouppose/backbones/swin_transformer.py:393
↓ 1 callers
Class
PositionEmbeddingLearned
Absolute pos embedding, learned.
models/grouppose/position_encoding.py:102
↓ 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/grouppose/position_encoding.py:58
↓ 1 callers
Class
PostProcess
This module converts the model's output into the format expected by the coco api
models/grouppose/postprocesses.py:6
↓ 1 callers
Class
Registry
models/registry.py:5
↓ 1 callers
Class
SetCriterion
models/grouppose/criterion.py:10
↓ 1 callers
Class
SwinTransformer
Swin Transformer backbone. A PyTorch impl of : `Swin Transformer: Hierarchical Vision Transformer using Shifted Windows` - https:/
models/grouppose/backbones/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/grouppose/backbones/swin_transformer.py:150
↓ 1 callers
Class
Transformer
models/grouppose/transformer.py:21
↓ 1 callers
Class
TransformerDecoder
models/grouppose/transformer.py:250
↓ 1 callers
Class
TransformerEncoder
models/grouppose/transformer.py:200
↓ 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/grouppose/backbones/swin_transformer.py:69
↓ 1 callers
Class
_ColorfulFormatter
util/logger.py:9
Class
APOPMeter
util/utils.py:174
Class
BackboneBase
models/grouppose/backbones/backbone.py:68
Class
CocoClassMapper
util/utils.py:52
Class
CocoEvaluator
datasets/coco_eval.py:22
Class
Compose
datasets/transforms_crowdpose.py:274
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:399
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/grouppose/backbones/backbone.py:29
Class
MSDeformAttn
models/grouppose/ops/modules/ms_deform_attn_key_aware.py:30
Class
MSDeformAttnFunction
models/grouppose/ops/functions/ms_deform_attn_func.py:21
Class
MetricLogger
util/misc.py:166
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_crowdpose.py:239
Class
Normalize
datasets/transforms.py:236
Class
PatchMerging
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
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/grouppose/position_encoding.py:20
Class
RandomCrop
datasets/transforms_crowdpose.py:176
Class
RandomCrop
datasets/transforms.py:173
Class
RandomHorizontalFlip
datasets/transforms_crowdpose.py:197
Class
RandomHorizontalFlip
datasets/transforms.py:194
Class
RandomResize
datasets/transforms_crowdpose.py:207
Class
RandomResize
datasets/transforms.py:204
Class
RandomSelect
Randomly selects between transforms1 and transforms2, with probability p for transforms1 and (1 - p) for transforms2
datasets/transforms_crowdpose.py:218
Class
RandomSelect
Randomly selects between transforms1 and transforms2, with probability p for transforms1 and (1 - p) for transforms2
datasets/transforms.py:215
Class
RandomSizeCrop
datasets/transforms_crowdpose.py:185
Class
RandomSizeCrop
datasets/transforms.py:182
Class
ToTensor
datasets/transforms_crowdpose.py:234
Class
ToTensor
datasets/transforms.py:231
Class
color_sys
util/misc.py:553