MCPcopy Create free account

hub / github.com/Michel-liu/GroupPose / functions

Functions340 in github.com/Michel-liu/GroupPose

Method__init__
(self, num_select=100, num_body_points=17)
models/grouppose/postprocesses.py:8
Method__init__
(self, cost_class: float = 1, focal_alpha=0.25, cost_keypoints=1.0, cost_oks=0.01, num_body_p
models/grouppose/matcher.py:22
Method__init__
(self, num_classes, matcher, weight_dict, focal_alpha, losses, num_body_points)
models/grouppose/criterion.py:11
Method__init__
(self, encoder_layer, num_layers, norm=None, d_model=256, num_queries=300)
models/grouppose/transformer.py:201
Method__init__
(self, decoder_layer, num_layers, return_intermediate=False, d_model
models/grouppose/transformer.py:251
Method__init__
(self, num_pos_feats=64, temperatureH=10000, temperatureW=10000, normalize=False, scale=None)
models/grouppose/position_encoding.py:63
Method__init__
(self, num_pos_feats=256)
models/grouppose/position_encoding.py:106
Method__init__
Multi-Scale Deformable Attention Module :param d_model hidden dimension :param n_levels number of feature levels
models/grouppose/ops/modules/ms_deform_attn_key_aware.py:31
Method__init__
Multi-Scale Deformable Attention Module :param d_model hidden dimension :param n_levels number of feature levels
models/grouppose/ops/modules/ms_deform_attn.py:31
Method__init__
(self, dim, window_size, num_heads, qkv_bias=True, qk_scale=None, attn_drop=0., proj_drop=0.)
models/grouppose/backbones/swin_transformer.py:82
Method__init__
(self, dim, num_heads, window_size=7, shift_size=0, mlp_ratio=4., qkv_bias=True, qk_scale=Non
models/grouppose/backbones/swin_transformer.py:167
Method__init__
(self, dim, norm_layer=nn.LayerNorm)
models/grouppose/backbones/swin_transformer.py:256
Method__init__
(self, dim, depth, num_heads, window_size=
models/grouppose/backbones/swin_transformer.py:309
Method__init__
(self, patch_size=4, in_chans=3, embed_dim=96, norm_layer=None)
models/grouppose/backbones/swin_transformer.py:402
Method__init__
(self, pretrain_img_size=224, patch_size=4, in_chans=3,
models/grouppose/backbones/swin_transformer.py:464
Method__init__
(self, n)
models/grouppose/backbones/backbone.py:38
Method__init__
(self, backbone: nn.Module, train_backbone: bool, num_channels: int, return_interm_indices: list)
models/grouppose/backbones/backbone.py:69
Method__init__
(self, name: str, train_backbone: bool, dilation: bool, ret
models/grouppose/backbones/backbone.py:94
Method__init__
(self, window_size=20, fmt=None)
util/misc.py:37
Method__init__
(self, delimiter="\t")
util/misc.py:167
Method__init__
(self, tensors, mask: Optional[Tensor])
util/misc.py:298
Method__init__
(self, num_colors)
util/misc.py:554
Method__init__
(self, linear=False, num_keypoints=17, eps=1e-6,
util/keypoint_loss.py:38
Method__init__
(self)
util/utils.py:53
Method__init__
(self)
util/utils.py:175
Method__init__
(self, model, decay=0.9997, device=None)
util/utils.py:374
Method__init__
(self, init_res=0.0, better='large')
util/utils.py:399
Method__init__
(self, init_res=0.0, better='large', use_ema=False)
util/utils.py:434
Method__init__
(self, *args, **kwargs)
util/logger.py:10
Method__iter__
(self)
util/slconfig.py:346
Method__len__
(self)
datasets/crowdpose.py:40
Method__len__
(self)
datasets/coco.py:49
Method__len__
(self)
models/registry.py:16
Method__len__
(self)
util/slconfig.py:317
Method__missing__
(self, name)
util/slconfig.py:27
Method__repr__
(self)
datasets/transforms_crowdpose.py:283
Method__repr__
(self)
datasets/transforms.py:279
Method__repr__
(self)
models/registry.py:11
Method__repr__
(self)
util/slconfig.py:314
Method__repr__
(self)
util/misc.py:360
Method__repr__
(self)
util/utils.py:423
Method__setitem__
(self, name, value)
util/slconfig.py:341
Method__setstate__
(self, state)
util/slconfig.py:388
Method__str__
(self)
util/misc.py:87
Method__str__
(self)
util/misc.py:186
Method__str__
str: the string of the module
util/utils.py:294
Method__str__
(self)
util/utils.py:469
Method_contain_invalid_identifier
(dict_str)
util/slconfig.py:266
Method_format_basic_types
(k, v, use_mapping=False)
util/slconfig.py:234
Method_format_dict
(input_dict, outest_level=False)
util/slconfig.py:273
Method_format_list
(k, v, use_mapping=False)
util/slconfig.py:249
Method_get_tgt_permutation_idx
(self, indices)
models/grouppose/criterion.py:98
Method_indent
(s_, num_spaces)
util/slconfig.py:224
Method_load_from_state_dict
(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys
models/grouppose/backbones/backbone.py:45
Methodaccumulate
(self)
datasets/crowdpose_eval.py:71
Methodavg
(self)
util/misc.py:71
Functionbox_cxcywh_to_xyxy
(x)
util/box_ops.py:9
Functionbuild
(image_set, args)
datasets/crowdpose.py:188
Functionbuild
(image_set, args)
datasets/coco.py:175
Functionbuild_grouppose
(args)
models/grouppose/grouppose.py:205
Functionclean_state_dict
(state_dict)
util/utils.py:22
Functioncollate_fn
(batch)
util/misc.py:282
Methodcompact2origin
(self, idx)
util/utils.py:61
Methodfilename
(self)
util/slconfig.py:212
MethodformatMessage
(self, record)
util/logger.py:17
Methodforward
(self, samples: NestedTensor)
models/grouppose/grouppose.py:135
Methodforward
(self, x)
models/grouppose/utils.py:96
Methodforward
(self, src, pos, reference_points, spatial_shapes, level_start_index, key_padding_mask=None)
models/grouppose/transformer_deformable.py:43
Methodforward
(self, # for tgt tgt_pose: Optional[Tensor], tgt_pose_query_p
models/grouppose/transformer_deformable.py:89
Methodforward
(self, outputs, target_sizes)
models/grouppose/postprocesses.py:14
Methodforward
(self, outputs, targets)
models/grouppose/matcher.py:47
Methodforward
(self, outputs, targets)
models/grouppose/criterion.py:113
Methodforward
Input: - srcs: List of multi features [bs, ci, hi, wi] - masks: List of multi masks [bs, hi, wi] - refpoi
models/grouppose/transformer.py:110
Methodforward
(self, src: Tensor, pos: Tensor, spatial_shapes: Tensor, l
models/grouppose/transformer.py:230
Methodforward
(self, tgt, memory, memory_key_padding_mask: Optional[Tensor] = None, refpoint
models/grouppose/transformer.py:268
Methodforward
(self, tensor_list: NestedTensor)
models/grouppose/position_encoding.py:36
Methodforward
(self, tensor_list: NestedTensor)
models/grouppose/position_encoding.py:75
Methodforward
(self, tensor_list: NestedTensor)
models/grouppose/position_encoding.py:116
Methodforward
:param query (N, Length_{query}, C) :param key (N, 1, C) :param reference_poin
models/grouppose/ops/modules/ms_deform_attn_key_aware.py:80
Methodforward
:param query (N, Length_{query}, C) :param reference_points (N, Length_{query}, n_levels, 2), range
models/grouppose/ops/modules/ms_deform_attn.py:82
Methodforward
(ctx, value, value_spatial_shapes, value_level_start_index, sampling_locations, attention_weights, im2col_step
models/grouppose/ops/functions/ms_deform_attn_func.py:23
Methodforward
(self, x)
models/grouppose/backbones/swin_transformer.py:30
Methodforward
Forward function. Args: x: input features with shape of (num_windows*B, N, C) mask: (0/-inf) mask with shape of (num_
models/grouppose/backbones/swin_transformer.py:116
Methodforward
Forward function. Args: x: Input feature, tensor size (B, H*W, C). H, W: Spatial resolution of the input feature.
models/grouppose/backbones/swin_transformer.py:191
Methodforward
Forward function. Args: x: Input feature, tensor size (B, H*W, C). H, W: Spatial resolution of the input feature.
models/grouppose/backbones/swin_transformer.py:262
Methodforward
Forward function. Args: x: Input feature, tensor size (B, H*W, C). H, W: Spatial resolution of the input feature.
models/grouppose/backbones/swin_transformer.py:351
Methodforward
Forward function.
models/grouppose/backbones/swin_transformer.py:416
Methodforward
(self, tensor_list: NestedTensor)
models/grouppose/backbones/swin_transformer.py:633
Methodforward
(self, x)
models/grouppose/backbones/backbone.py:55
Methodforward
(self, tensor_list: NestedTensor)
models/grouppose/backbones/backbone.py:81
Methodforward
(self, tensor_list: NestedTensor)
models/grouppose/backbones/backbone.py:118
Methodforward
(self, pred, target, valid, area,
util/keypoint_loss.py:66
Functiongeneralized_box_iou
Generalized IoU from https://giou.stanford.edu/ The boxes should be in [x0, y0, x1, y1] format Returns a [N, M] pairwise matrix, where
util/box_ops.py:40
Functiongeneralized_box_iou_pairwise
Generalized IoU from https://giou.stanford.edu/ Input: - boxes1, boxes2: N,4 Output: - giou: N, 4
util/box_ops.py:82
Functionget_embedder
(multires, i=0)
util/utils.py:156
Functionget_expected_points_from_map
get_gaussian_map_from_points B,C,H,W -> B,N,2 float(0, 1) float(0, 1) softargmax function Args: hm (float): Input images(
util/utils.py:102
Functionget_raw_dict
return the dicf contained in args. e.g: >>> with open(path, 'w') as f: json.dump(get_raw_dict(args), f, indent=2
util/utils.py:206
Functionget_sha
()
util/misc.py:262
Methodglobal_avg
(self)
util/misc.py:76
Methodimgsize
(self)
util/misc.py:310
← previousnext →201–300 of 340, ranked by callers