MCPcopy Create free account

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

Functions340 in github.com/Michel-liu/GroupPose

↓ 1 callersFunctionevaluate
Run per image evaluation on given images and store results (a list of dict) in self.evalImgs :return: None
datasets/crowdpose_eval.py:218
↓ 1 callersMethodforward_FFN
(self, tgt)
models/grouppose/transformer_deformable.py:83
↓ 1 callersMethodforward_ffn
(self, src)
models/grouppose/transformer_deformable.py:37
↓ 1 callersMethodfromfile
(filename)
util/slconfig.py:184
↓ 1 callersFunctiongen_encoder_output_proposals
Input: - memory: bs, \sum{hw}, d_model - memory_padding_mask: bs, \sum{hw} - spatial_shapes: nlevel, 2 - learnedw
models/grouppose/utils.py:11
↓ 1 callersFunctiongen_sineembed_for_position
(pos_tensor)
models/grouppose/utils.py:116
↓ 1 callersMethodget
(self, key)
models/registry.py:27
↓ 1 callersFunctionget_args_parser
()
main.py:26
↓ 1 callersFunctionget_extensions
()
models/grouppose/ops/setup.py:23
↓ 1 callersFunctionget_param_dict
(args, model_without_ddp: nn.Module)
util/get_param_dicts.py:15
↓ 1 callersFunctionget_rank
()
util/misc.py:456
↓ 1 callersMethodget_reference_points
(spatial_shapes, valid_ratios, device)
models/grouppose/transformer.py:216
↓ 1 callersFunctionget_size
(image_size, size, max_size=None)
datasets/transforms_crowdpose.py:122
↓ 1 callersFunctionget_size
(image_size, size, max_size=None)
datasets/transforms.py:119
↓ 1 callersFunctionget_size_with_aspect_ratio
(image_size, size, max_size=None)
datasets/transforms_crowdpose.py:102
↓ 1 callersFunctionget_size_with_aspect_ratio
(image_size, size, max_size=None)
datasets/transforms.py:99
↓ 1 callersMethodget_valid_ratio
(self, mask)
models/grouppose/transformer.py:101
↓ 1 callersFunctionhflip
(image, target)
datasets/transforms_crowdpose.py:69
↓ 1 callersFunctionhflip
(image, target)
datasets/transforms.py:69
↓ 1 callersMethodisbetter
(self, new_res, old_res)
util/utils.py:407
↓ 1 callersFunctionkey_select_function
(keyname)
models/grouppose/backbones/backbone.py:175
↓ 1 callersFunctionkeypoint_xyzxyz_to_xyxyzz
_summary_ Args: keypoints (torch.Tensor): ..., 51
util/keypoint_ops.py:18
↓ 1 callersFunctionmain
(args)
main.py:82
↓ 1 callersFunctionmake_coco_transforms
(image_set, fix_size=False, args=None)
datasets/crowdpose.py:123
↓ 1 callersFunctionmake_coco_transforms
(image_set, args=None)
datasets/coco.py:135
↓ 1 callersFunctionmerge
(img_ids, eval_imgs)
datasets/coco_eval.py:201
↓ 1 callersFunctionmerge
(img_ids, eval_imgs)
datasets/crowdpose_eval.py:185
↓ 1 callersMethodmerge_from_dict
Merge list into cfg_dict Merge the dict parsed by MultipleKVAction into this cfg. Examples: >>> options = {'model.backbo
util/slconfig.py:356
↓ 1 callersFunctionoks_loss
(pred, target, valid=None, area=None, linear=False,
util/keypoint_loss.py:22
↓ 1 callersFunctionoks_overlaps
(kpt_preds, kpt_gts, kpt_valids, kpt_areas, sigmas)
util/keypoint_loss.py:5
↓ 1 callersMethodprepare
(self, predictions, iou_type)
datasets/crowdpose_eval.py:80
↓ 1 callersMethodprepare_for_coco_detection
(self, predictions)
datasets/coco_eval.py:106
↓ 1 callersMethodprepare_for_coco_detection
(self, predictions)
datasets/crowdpose_eval.py:90
↓ 1 callersMethodprepare_for_coco_keypoint
(self, predictions)
datasets/coco_eval.py:39
↓ 1 callersMethodprepare_for_coco_keypoint
(self, predictions)
datasets/crowdpose_eval.py:26
↓ 1 callersMethodprepare_for_coco_segmentation
(self, predictions)
datasets/coco_eval.py:136
↓ 1 callersMethodprepare_for_coco_segmentation
(self, predictions)
datasets/crowdpose_eval.py:120
↓ 1 callersMethodreset_parameters
(self)
models/grouppose/position_encoding.py:112
↓ 1 callersFunctionresize
(image, target, size, max_size=None)
datasets/transforms_crowdpose.py:99
↓ 1 callersFunctionresize
(image, target, size, max_size=None)
datasets/transforms.py:96
↓ 1 callersFunctionsetup_for_distributed
This function disables printing when not in master process
util/misc.py:427
↓ 1 callersFunctionsetup_logger
Initialize the detectron2 logger and set its verbosity level to "INFO". Args: output (str): a file name or a directory to save log.
util/logger.py:31
↓ 1 callersFunctionsigmoid_focal_loss
Loss used in RetinaNet for dense detection: https://arxiv.org/abs/1708.02002. Args: inputs: A float tensor of arbitrary shape.
models/grouppose/utils.py:59
↓ 1 callersMethodsummarize
(self)
datasets/coco_eval.py:91
↓ 1 callersMethodsynchronize_between_processes
(self)
datasets/coco_eval.py:82
↓ 1 callersFunctionto_device
(item, device)
util/utils.py:64
↓ 1 callersFunctiontrain_one_epoch
(model: torch.nn.Module, criterion: torch.nn.Module, data_loader: Iterable, optimizer: tor
engine.py:9
↓ 1 callersMethodupdate
(self, predictions)
datasets/coco_eval.py:63
↓ 1 callersFunctionwindow_reverse
Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image
models/grouppose/backbones/swin_transformer.py:53
↓ 1 callersMethodwith_pos_embed
(tensor, pos)
models/grouppose/transformer_deformable.py:34
FunctionPYBIND11_MODULE
models/grouppose/ops/src/vision.cpp:13
Method__call__
(self, img, target)
datasets/transforms_crowdpose.py:180
Method__call__
(self, img: PIL.Image.Image, target: dict)
datasets/transforms_crowdpose.py:190
Method__call__
(self, img, target)
datasets/transforms_crowdpose.py:201
Method__call__
(self, img, target=None)
datasets/transforms_crowdpose.py:213
Method__call__
(self, img, target)
datasets/transforms_crowdpose.py:228
Method__call__
(self, img, target)
datasets/transforms_crowdpose.py:235
Method__call__
(self, image, target=None)
datasets/transforms_crowdpose.py:244
Method__call__
(self, image, target)
datasets/transforms_crowdpose.py:278
Method__call__
(self, img, target)
datasets/transforms.py:177
Method__call__
(self, img: PIL.Image.Image, target: dict)
datasets/transforms.py:187
Method__call__
(self, img, target)
datasets/transforms.py:198
Method__call__
(self, img, target=None)
datasets/transforms.py:210
Method__call__
(self, img, target)
datasets/transforms.py:225
Method__call__
(self, img, target)
datasets/transforms.py:232
Method__call__
(self, image, target=None)
datasets/transforms.py:241
Method__call__
(self, image, target)
datasets/transforms.py:274
Method__call__
(self, image, target)
datasets/crowdpose.py:77
Method__call__
(self, image, target)
datasets/coco.py:86
Method__call__
(self, parser, namespace, values, option_string=None)
util/slconfig.py:422
Method__call__
(self, idx)
util/misc.py:564
Method__getattr__
(self, name)
util/slconfig.py:30
Method__getattr__
(self, attr)
util/misc.py:178
Method__getitem__
(self, idx)
datasets/crowdpose.py:43
Method__getitem__
(self, idx)
datasets/coco.py:52
Method__getitem__
(self, name)
util/slconfig.py:333
Method__init__
(self, size)
datasets/transforms_crowdpose.py:177
Method__init__
(self, min_size: int, max_size: int)
datasets/transforms_crowdpose.py:186
Method__init__
(self, p=0.5)
datasets/transforms_crowdpose.py:198
Method__init__
(self, sizes, max_size=None)
datasets/transforms_crowdpose.py:208
Method__init__
(self, transforms1, transforms2, p=0.5)
datasets/transforms_crowdpose.py:223
Method__init__
(self, mean, std)
datasets/transforms_crowdpose.py:240
Method__init__
(self, transforms)
datasets/transforms_crowdpose.py:275
Method__init__
(self, size)
datasets/transforms.py:174
Method__init__
(self, min_size: int, max_size: int)
datasets/transforms.py:183
Method__init__
(self, p=0.5)
datasets/transforms.py:195
Method__init__
(self, sizes, max_size=None)
datasets/transforms.py:205
Method__init__
(self, transforms1, transforms2, p=0.5)
datasets/transforms.py:220
Method__init__
(self, mean, std)
datasets/transforms.py:237
Method__init__
(self, transforms)
datasets/transforms.py:271
Method__init__
(self, coco_gt, iou_types, useCats=True, coco_path="")
datasets/coco_eval.py:23
Method__init__
(self, root_path, image_set, transforms)
datasets/crowdpose.py:14
Method__init__
(self, return_masks=False)
datasets/crowdpose.py:74
Method__init__
(self, root_path, image_set, transforms, return_masks=False)
datasets/coco.py:22
Method__init__
(self, return_masks=False)
datasets/coco.py:83
Method__init__
(self, coco_gt, iou_types, useCats=True, coco_path='')
datasets/crowdpose_eval.py:10
Method__init__
(self, name)
models/registry.py:7
Method__init__
(self, backbone, transformer, num_classes, num_queries, aux_loss=False,
models/grouppose/grouppose.py:37
Method__init__
(self, input_dim, hidden_dim, output_dim, num_layers)
models/grouppose/utils.py:90
Method__init__
(self, d_model=256, d_ffn=1024, dropout=0.1, activation="relu", n_levels=4,
models/grouppose/transformer_deformable.py:52
← previousnext →101–200 of 340, ranked by callers