MCPcopy Create free account

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

Functions340 in github.com/Michel-liu/GroupPose

↓ 35 callersFunctionprint
(*args, **kwargs)
util/misc.py:434
↓ 26 callersMethodupdate
(self, model)
util/utils.py:392
↓ 24 callersMethodto
(self, device)
util/misc.py:319
↓ 19 callersMethodmax
(self)
util/misc.py:80
↓ 12 callersMethodcopy
(self)
util/slconfig.py:392
↓ 12 callersFunctionmatch_name_keywords
(n: str, name_keywords: list)
util/get_param_dicts.py:6
↓ 11 callersMethoddeepcopy
(self)
util/slconfig.py:395
↓ 6 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
models/grouppose/backbones/swin_transformer.py:21
↓ 4 callersMethod__setattr__
(self, name, value)
util/slconfig.py:336
↓ 4 callersFunctionall_gather
Run all_gather on arbitrary picklable data (not necessarily tensors) Args: data: any picklable object Returns: list[data]
util/misc.py:96
↓ 4 callersFunctionis_dist_avail_and_initialized
()
util/misc.py:442
↓ 4 callersMethodsummary
(self)
util/utils.py:426
↓ 3 callersMethod__init__
(self, backbone, position_embedding)
models/grouppose/backbones/backbone.py:115
↓ 3 callersFunction_run
(command)
util/misc.py:265
↓ 3 callersMethodadd_meter
(self, name, meter)
util/misc.py:199
↓ 3 callersMethoddump
(self, file=None)
util/slconfig.py:349
↓ 3 callersFunctionevaluate
(model, criterion, postprocessors, data_loader, base_ds, device, output_dir, wo_class_error=False, args=None,
engine.py:85
↓ 3 callersMethodget_loss
(self, loss, outputs, targets, indices, num_boxes, **kwargs)
models/grouppose/criterion.py:103
↓ 3 callersFunctionget_world_size
()
util/misc.py:450
↓ 3 callersMethodprepare
(self, predictions, iou_type)
datasets/coco_eval.py:96
↓ 3 callersMethodupdate
(self, **kwargs)
util/misc.py:171
↓ 2 callersMethod__init__
(self, d_model=256, nhead=8, num_queries=300, num_encode
models/grouppose/transformer.py:23
↓ 2 callersMethod__init__
(self, num_pos_feats=64, temperature=10000, normalize=False, scale=None)
models/grouppose/position_encoding.py:25
↓ 2 callersMethod__nice__
str: a "nice" summary string describing this module
util/utils.py:273
↓ 2 callersMethod_freeze_stages
(self)
models/grouppose/backbones/swin_transformer.py:557
↓ 2 callersFunction_get_activation_fn
Return an activation function given a string
models/grouppose/utils.py:101
↓ 2 callersFunction_get_clones
(module, N, layer_share=False)
models/grouppose/transformer.py:317
↓ 2 callersMethod_get_src_permutation_idx
(self, indices)
models/grouppose/criterion.py:93
↓ 2 callersMethod_merge_a_into_b
merge dict `a` into dict `b` (non-inplace). values in `a` will overwrite `b`. copy first to avoid inplace modification
util/slconfig.py:143
↓ 2 callersMethod_update
(self, model, update_fn)
util/utils.py:385
↓ 2 callersMethodbackward
(ctx, grad_output)
models/grouppose/ops/functions/ms_deform_attn_func.py:32
↓ 2 callersFunctionbox_iou
(boxes1, boxes2)
util/box_ops.py:24
↓ 2 callersFunctionbox_xyxy_to_cxcywh
(x)
util/box_ops.py:16
↓ 2 callersFunctionbuild_dataset
(image_set, args)
datasets/__init__.py:16
↓ 2 callersFunctionbuild_swin_transformer
(modelname, pretrain_img_size, **kw)
models/grouppose/backbones/swin_transformer.py:683
↓ 2 callersFunctioncrop
(image, target, region)
datasets/transforms_crowdpose.py:15
↓ 2 callersFunctioncrop
(image, target, region)
datasets/transforms.py:15
↓ 2 callersMethoddevice
(self)
util/misc.py:354
↓ 2 callersMethodforward_raw
Forward function.
models/grouppose/backbones/swin_transformer.py:600
↓ 2 callersFunctionget_coco_api_from_dataset
(dataset)
datasets/__init__.py:6
↓ 2 callersFunctionget_gaussian_mean
Args: x (float): Input images(BxCxHxW) axis (int): The index for weighted mean other_axis (int): The other index Re
util/utils.py:77
↓ 2 callersFunctioninterpolate
Equivalent to nn.functional.interpolate, but with support for empty batch sizes. This will eventually be supported natively by PyTorch, and t
util/misc.py:532
↓ 2 callersFunctioninverse_sigmoid
(x, eps=1e-3)
util/misc.py:567
↓ 2 callersFunctionis_main_process
()
util/misc.py:462
↓ 2 callersMethodlog_every
(self, iterable, print_freq, header=None, logger=None)
util/misc.py:202
↓ 2 callersFunctionms_deform_attn_core_pytorch
(value, value_spatial_shapes, sampling_locations, attention_weights)
models/grouppose/ops/functions/ms_deform_attn_func.py:41
↓ 2 callersFunctionnested_tensor_from_tensor_list
(tensor_list: List[Tensor])
util/misc.py:371
↓ 2 callersMethodsummary
(self)
util/utils.py:456
↓ 2 callersMethodsynchronize_between_processes
(self)
util/misc.py:195
↓ 2 callersMethodto_img_list_single
(self, tensor, mask)
util/misc.py:330
↓ 2 callersMethodtrain
Convert the model into training mode while keep layers freezed.
models/grouppose/backbones/swin_transformer.py:676
↓ 2 callersMethodupdate
(self, value, n=1)
util/misc.py:45
↓ 2 callersMethodupdate
return if the results is the best.
util/utils.py:442
↓ 2 callersFunctionwindow_partition
Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)
models/grouppose/backbones/swin_transformer.py:39
↓ 2 callersMethodwith_pos_embed
(tensor, pos)
models/grouppose/transformer_deformable.py:77
↓ 1 callersMethod__getattr__
(self, name)
util/slconfig.py:320
↓ 1 callersMethod__init__
(self, d_model=256, d_ffn=1024, dropout=0.1, activation="relu",
models/grouppose/transformer_deformable.py:17
↓ 1 callersMethod__init__
(self, cfg_dict=None, cfg_text=None, filename=None)
util/slconfig.py:189
↓ 1 callersMethod__init__
(self, **kwargs)
util/utils.py:124
↓ 1 callersMethod__repr__
str: the string of the module
util/utils.py:284
↓ 1 callersMethod__repr__
(self)
util/utils.py:466
↓ 1 callersMethod__str__
(self)
util/utils.py:420
↓ 1 callersFunction_cached_log_stream
(filename)
util/logger.py:94
↓ 1 callersMethod_file2dict
(filename)
util/slconfig.py:78
↓ 1 callersFunction_is_power_of_2
(n)
models/grouppose/ops/modules/ms_deform_attn_key_aware.py:24
↓ 1 callersFunction_is_power_of_2
(n)
models/grouppose/ops/modules/ms_deform_attn.py:24
↓ 1 callersFunction_max_by_axis
(the_list)
util/misc.py:288
↓ 1 callersFunction_onnx_nested_tensor_from_tensor_list
(tensor_list: List[Tensor])
util/misc.py:399
↓ 1 callersMethod_parse_int_float_bool
(val)
util/slconfig.py:407
↓ 1 callersMethod_reset_parameters
(self)
models/grouppose/grouppose.py:129
↓ 1 callersMethod_reset_parameters
(self)
models/grouppose/transformer.py:89
↓ 1 callersMethod_reset_parameters
(self)
models/grouppose/ops/modules/ms_deform_attn_key_aware.py:64
↓ 1 callersMethod_reset_parameters
(self)
models/grouppose/ops/modules/ms_deform_attn.py:64
↓ 1 callersMethod_set_aux_loss
(self, outputs_class, outputs_keypoints)
models/grouppose/grouppose.py:195
↓ 1 callersMethod_validate_py_syntax
(filename)
util/slconfig.py:68
↓ 1 callersMethodaccumulate
(self)
datasets/coco_eval.py:87
↓ 1 callersFunctionaccuracy
Computes the precision@k for the specified values of k
util/misc.py:514
↓ 1 callersFunctionbox_iou_pairwise
(boxes1, boxes2)
util/box_ops.py:66
↓ 1 callersFunctionbuild_backbone
Useful args: - backbone: backbone name - lr_backbone: - dilation - return_interm_indices: available: [0,1,2,3],
models/grouppose/backbones/backbone.py:130
↓ 1 callersFunctionbuild_matcher
(args)
models/grouppose/matcher.py:113
↓ 1 callersFunctionbuild_model_main
(args)
main.py:75
↓ 1 callersFunctionbuild_position_encoding
(args)
models/grouppose/position_encoding.py:130
↓ 1 callersFunctionbuild_transformer
(args)
models/grouppose/transformer.py:324
↓ 1 callersFunctioncheck_file_exist
(filename, msg_tmpl='file "{}" does not exist')
util/slconfig.py:21
↓ 1 callersFunctioncheck_forward_equal_with_pytorch_double
()
models/grouppose/ops/test.py:32
↓ 1 callersFunctioncheck_forward_equal_with_pytorch_float
()
models/grouppose/ops/test.py:48
↓ 1 callersFunctioncheck_gradient_numerical
(channels=4, grad_value=True, grad_sampling_loc=True, grad_attn_weight=True)
models/grouppose/ops/test.py:63
↓ 1 callersFunctioncheck_keep
(keyname, ignorekeywordlist)
main.py:236
↓ 1 callersFunctionclean_state_dict
(state_dict)
util/misc.py:573
↓ 1 callersFunctionconvert_coco_poly_to_mask
(segmentations, height, width)
datasets/crowdpose.py:56
↓ 1 callersFunctionconvert_coco_poly_to_mask
(segmentations, height, width)
datasets/coco.py:65
↓ 1 callersFunctionconvert_to_xywh
(boxes)
datasets/coco_eval.py:196
↓ 1 callersFunctionconvert_to_xywh
(boxes)
datasets/crowdpose_eval.py:180
↓ 1 callersFunctioncreate_common_coco_eval
(coco_eval, img_ids, eval_imgs)
datasets/coco_eval.py:223
↓ 1 callersFunctioncreate_common_coco_eval
(coco_eval, img_ids, eval_imgs)
datasets/crowdpose_eval.py:207
↓ 1 callersMethodcreate_embedding_fn
(self)
util/utils.py:128
↓ 1 callersMethoddecompose
(self)
util/misc.py:357
↓ 1 callersMethodembed
(self, inputs)
util/utils.py:152
↓ 1 callersFunctionensure_rng
Coerces input into a random number generator. If the input is None, then a global random state is returned. If the input is a numeric value,
util/utils.py:306
↓ 1 callersFunctionevaluate
Run per image evaluation on given images and store results (a list of dict) in self.evalImgs :return: None
datasets/coco_eval.py:239
next →1–100 of 340, ranked by callers