MCPcopy Create free account

hub / github.com/IDEA-Research/DWPose / functions

Functions7,103 in github.com/IDEA-Research/DWPose

⨍ Functions 7,103◇ Types & classes 1,442↳ Endpoints 6
↓ 9 callersMethodbuild_ap10k_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_animal_datasets/test_ap10k_dataset.py:11
↓ 9 callersMethodbuild_atrw_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_animal_datasets/test_atrw_dataset.py:11
↓ 9 callersMethodbuild_coco_wholebody_face_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_face_datasets/test_coco_wholebody_face_dataset.py:11
↓ 9 callersMethodbuild_coco_wholebody_hand_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_hand_datasets/test_coco_wholebody_hand_dataset.py:11
↓ 9 callersMethodbuild_cofw_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_face_datasets/test_cofw_dataset.py:11
↓ 9 callersMethodbuild_face_300w_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_face_datasets/test_face_300w_dataset.py:11
↓ 9 callersMethodbuild_fly_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_animal_datasets/test_fly_dataset.py:11
↓ 9 callersMethodbuild_freihand_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_hand_datasets/test_freihand_dataset.py:11
↓ 9 callersMethodbuild_horse10_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_animal_datasets/test_horse10_dataset.py:11
↓ 9 callersMethodbuild_jhmdb_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_body_datasets/test_jhmdb_dataset.py:11
↓ 9 callersMethodbuild_locust_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_animal_datasets/test_locust_dataset.py:11
↓ 9 callersMethodbuild_macaque_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_animal_datasets/test_macaque_dataset.py:11
↓ 9 callersMethodbuild_mhp_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_body_datasets/test_mhp_dataset.py:11
↓ 9 callersMethodbuild_mpii_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_body_datasets/test_mpii_dataset.py:11
↓ 9 callersMethodbuild_mpii_trb_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_body_datasets/test_mpii_trb_dataset.py:11
↓ 9 callersMethodbuild_ochuman_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_body_datasets/test_ochuman_dataset.py:11
↓ 9 callersMethodbuild_onehand10k_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_hand_datasets/test_onehand10k_dataset.py:11
↓ 9 callersMethodbuild_panoptic_hand2d_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_hand_datasets/test_panoptic_hand2d_dataset.py:11
↓ 9 callersMethodbuild_pose_lifting_label
(self, **kwargs)
mmpose/tests/test_codecs/test_video_pose_lifting.py:21
↓ 9 callersMethodbuild_pose_lifting_label
(self, **kwargs)
mmpose/tests/test_codecs/test_image_pose_lifting.py:32
↓ 9 callersMethodbuild_rhd2d_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_hand_datasets/test_rhd2d_dataset.py:11
↓ 9 callersMethodbuild_wflw_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_face_datasets/test_wflw_dataset.py:11
↓ 9 callersMethodbuild_zebra_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_animal_datasets/test_zebra_dataset.py:11
↓ 9 callersFunctioncast_tensor_type
Recursively convert Tensor in inputs from src_type to dst_type. Args: inputs: Inputs that to be casted. src_type (torch.dtype): S
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/fp16_utils.py:24
↓ 9 callersMethodclose
Closes the interpreter and model to release resources.
ControlNet-v1-1-nightly/annotator/zoe/zoedepth/models/base_models/midas_repo/mobile/android/lib_support/src/main/java/org/tensorflow/lite/examples/classification/tflite/Classifier.java:284
↓ 9 callersMethodextract_feat
Extract features. Args: inputs (Tensor): Image tensor with shape (N, C, H ,W). Returns: tuple[Tensor]: Multi
mmpose/mmpose/models/pose_estimators/base.py:178
↓ 9 callersMethodfrom_tensors
Args: tensors: a tuple or list of `torch.Tensor`, each of shape (Hi, Wi) or (C_1, ..., C_K, Hi, Wi) where K >= 1.
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/structures/image_list.py:59
↓ 9 callersMethodgetTitle
()
ControlNet-v1-1-nightly/annotator/zoe/zoedepth/models/base_models/midas_repo/mobile/android/lib_support/src/main/java/org/tensorflow/lite/examples/classification/tflite/Classifier.java:162
↓ 9 callersFunctionget_world_size
()
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/utils/comm.py:21
↓ 9 callersMethodinfer_client
Infer a suitable file client based on the URI and arguments. Args: file_client_args (dict, optional): Arguments to instantiate a
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/fileio/file_client.py:858
↓ 9 callersFunctionis_module_wrapper
Check if a module is a module wrapper. The following 3 modules in MMCV (and their subclasses) are regarded as module wrappers: DataParallel,
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/parallel/utils.py:5
↓ 9 callersMethodloss
(self, logit_s, logit_t, beta, weight)
mmpose/mmpose/models/losses/kd.py:40
↓ 9 callersFunctionmove_device_like
Tracing friendly way to cast tensor to another tensor's device. Device will be treated as constant during tracing, scripting the casting proc
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/layers/wrappers.py:157
↓ 9 callersFunctionnonlinearity
(x)
ControlNet-v1-1-nightly/ldm/modules/diffusionmodules/model.py:41
↓ 9 callersMethodoverlay_instances
Args: boxes (Boxes, RotatedBoxes or ndarray): either a :class:`Boxes`, or an Nx4 numpy array of XYXY_ABS format f
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/utils/visualizer.py:607
↓ 9 callersFunctionpairwise_iou
Given two lists of boxes of size N and M, compute the IoU (intersection over union) between **all** N x M pairs of boxes. The box order m
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/structures/boxes.py:336
↓ 9 callersFunctionregister_all_modules
Register all modules in mmpose into the registries. Args: init_default_scope (bool): Whether initialize the mmpose default scope.
mmpose/mmpose/utils/setup_env.py:52
↓ 9 callersMethodremove
Alias of ``pop``.
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/data/catalog.py:69
↓ 9 callersMethodsample_log
(self, cond, batch_size, ddim, ddim_steps, **kwargs)
ControlNet-v1-1-nightly/ldm/models/diffusion/ddpm.py:1110
↓ 9 callersMethodsummarize
Compute and display summary metrics for evaluation results given a custom value for max_dets_per_image
ControlNet-v1-1-nightly/annotator/oneformer/oneformer/evaluation/coco_evaluator.py:481
↓ 9 callersMethodto
Reload ``to`` for tensors.
mmpose/projects/yolox_pose/datasets/bbox_keypoint_structure.py:278
↓ 9 callersMethodtrain
(self, mode=True)
mmpose/mmpose/models/backbones/shufflenet_v2.py:305
↓ 9 callersMethodtrain
Convert the model into training mode.
mmpose/mmpose/models/backbones/resnet.py:693
↓ 8 callersMethod__init__
(self, channels, use_conv, dims=2, out_channels=None, padding=1)
ControlNet-v1-1-nightly/ldm/modules/diffusionmodules/openaimodel.py:99
↓ 8 callersFunction_rename
(cfg: CN, old: str, new: str)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/config/compat.py:116
↓ 8 callersFunctionadd_prefix
Add prefix for dict. Args: inputs (dict): The input dict with str keys. prefix (str): The prefix to add. Returns: d
ControlNet-v1-1-nightly/annotator/uniformer/mmseg/core/utils/misc.py:1
↓ 8 callersMethodcheck_data_info_keys
(self, data_info: dict, data_mode: str = 'topdown')
mmpose/tests/test_datasets/test_datasets/test_body_datasets/test_posetrack18_video_dataset.py:30
↓ 8 callersMethodcheck_data_info_keys
(self, data_info: dict, data_mode: str = 'topdown')
mmpose/tests/test_datasets/test_datasets/test_body_datasets/test_h36m_dataset.py:23
↓ 8 callersFunctiondump
Dump data to json/yaml/pickle strings or files. This method provides a unified api for dumping data as strings or to files, and also supports
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/fileio/io.py:69
↓ 8 callersMethoddump
Returns: str: a yaml string representation of the config
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/config/config.py:87
↓ 8 callersMethodevery_n_iters
(self, runner, n)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/hooks/hook.py:61
↓ 8 callersFunctionflip_heatmaps
Flip heatmaps for test-time augmentation. Args: heatmaps (Tensor): The heatmaps to flip. Should be a tensor in shape [B, C, H
mmpose/mmpose/models/utils/tta.py:9
↓ 8 callersFunctionget_heatmap_maximum
Get maximum response location and value from heatmaps. Note: batch_size: B num_keypoints: K heatmap height: H hea
mmpose/mmpose/codecs/utils/post_processing.py:97
↓ 8 callersFunctionget_root_logger
Get the root logger. The logger will be initialized if it has not been initialized. By default a StreamHandler will be added. If `log_file` i
ControlNet-v1-1-nightly/annotator/uniformer/mmseg/utils/logger.py:6
↓ 8 callersMethodinit_weights
(self, pretrained=None)
ControlNet-v1-1-nightly/annotator/uniformer/mmseg/models/backbones/mobilenet_v3.py:220
↓ 8 callersMethodinit_weights
(self, pretrained=None)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/cnn/vgg.py:125
↓ 8 callersMethodis_norm
Check if is one of the norms.
mmpose/tests/test_models/test_backbones/test_shufflenet_v1.py:22
↓ 8 callersMethodload
Load model from file. Args: path (str): file path
ControlNet-v1-1-nightly/ldm/modules/midas/midas/base_model.py:5
↓ 8 callersMethodnorm1
nn.Module: the normalization layer named "norm1"
mmpose/mmpose/models/backbones/hrnet.py:401
↓ 8 callersFunctionnorm_normalize
(norm_out)
ControlNet-v1-1-nightly/annotator/normalbae/models/submodules/submodules.py:64
↓ 8 callersFunctionpoint_sample
A wrapper around :func:`grid_sample` to support 3D point_coords tensors Unlike :func:`torch.nn.functional.grid_sample` it assumes point_coords to
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/ops/point_sample.py:242
↓ 8 callersMethodprocess
Split data samples by partitions, then call metric.process part by part.
mmpose/mmpose/evaluation/metrics/keypoint_partition_metric.py:149
↓ 8 callersMethodset_data
Set or change key-value pairs in ``data_field`` by parameter ``data``. Args: data (dict): A dict contains annotations of
mmpose/mmpose/structures/multilevel_pixel_data.py:103
↓ 8 callersFunctionsetup_logger
Initialize the detectron2 logger and set its verbosity level to "DEBUG". Args: output (str): a file name or a directory to save log.
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/utils/logger.py:39
↓ 8 callersMethodto
(self, *args: Any, **kwargs: Any)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/structures/masks.py:111
↓ 8 callersMethodtrain
(self, mode=True)
mmpose/mmpose/models/backbones/mobilenet_v3.py:179
↓ 8 callersFunctionupdate_init_info
Update the `_params_init_info` in the module if the value of parameters are changed. Args: module (obj:`nn.Module`): The module of Py
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/cnn/utils/weight_init.py:16
↓ 8 callersFunctionxavier_init
(module, gain=1, bias=0, distribution='normal')
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/cnn/utils/weight_init.py:55
↓ 7 callersMethod__init__
(self, in_channels, stem_channels, out_channels,
mmpose/mmpose/models/backbones/litehrnet.py:236
↓ 7 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
ControlNet-v1-1-nightly/annotator/uniformer/mmseg/models/backbones/uniformer.py:25
↓ 7 callersMethod__init__
(self, *, bias=0, bias_prob=None, layer=None)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/cnn/utils/weight_init.py:134
↓ 7 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
ControlNet-v1-1-nightly/annotator/oneformer/oneformer/modeling/backbone/dinat.py:77
↓ 7 callersMethod__init__
( self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.0 )
ControlNet-v1-1-nightly/annotator/oneformer/oneformer/modeling/backbone/swin.py:25
↓ 7 callersMethod__init__
(self, unet_config, timesteps=1000, beta_schedule="linear",
ControlNet-v1-1-nightly/ldm/models/diffusion/ddpm.py:48
↓ 7 callersMethod_generate_data
Generate data_batch and data_samples according to different settings.
mmpose/tests/test_evaluation/test_metrics/test_keypoint_2d_metrics.py:285
↓ 7 callersMethod_get_data_samples
(self, batch_size: int = 2)
mmpose/tests/test_models/test_heads/test_heatmap_heads/test_cpm_head.py:25
↓ 7 callersMethod_get_feats
(self, batch_size: int = 2, feat_shapes: List[Tuple[int, int, int]] = [(
mmpose/tests/test_models/test_heads/test_heatmap_heads/test_cpm_head.py:15
↓ 7 callersMethodapply_deltas
Apply transformation `deltas` (dx, dy, dw, dh) to `boxes`. Args: deltas (Tensor): transformation deltas of shape (N, k*4
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/modeling/box_regression.py:78
↓ 7 callersFunctionchannel_shuffle
Channel Shuffle operation. This function enables cross-group information flow for multiple groups convolution layers. Args: x (T
mmpose/mmpose/models/backbones/utils/channel_shuffle.py:5
↓ 7 callersMethodclear
Clears the log of function call outputs.
mmpose/projects/yolox_pose/models/utils.py:37
↓ 7 callersMethodconvert
Multiple with alpha and add beat with clip.
ControlNet-v1-1-nightly/annotator/uniformer/mmseg/datasets/pipelines/transforms.py:804
↓ 7 callersFunctioncreate_small_table
Create a small table using the keys of small_dict as headers. This is only suitable for small dictionaries. Args: small_dict (di
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/utils/logger.py:209
↓ 7 callersMethodcriterion
Criterion of wingloss. Note: batch_size: N num_keypoints: K Args: pred (torch.Tensor[NxKxHxW]):
mmpose/mmpose/models/losses/heatmap_loss.py:310
↓ 7 callersFunctionget_act_layer
Activation Layer Factory Fetching activation layers by name with this function allows export or torch script friendly functions to be returne
ControlNet-v1-1-nightly/annotator/normalbae/models/submodules/efficientnet_repo/geffnet/activations/__init__.py:118
↓ 7 callersFunctionget_expansion
Get the expansion of a residual block. The block expansion will be obtained by the following order: 1. If ``expansion`` is given, just retur
mmpose/mmpose/models/backbones/vipnas_resnet.py:179
↓ 7 callersMethodget_multi_level_pixel_data
(self)
mmpose/tests/test_structures/test_multilevel_pixel_data.py:13
↓ 7 callersFunctionget_pb_arg_vali
(pb, arg_name, default_val)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/export/shared.py:194
↓ 7 callersFunctionimread
Read an image. Args: img_or_path (ndarray or str or Path): Either a numpy array or str or pathlib.Path. If it is a numpy arra
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/image/io.py:140
↓ 7 callersFunctioninference_topdown
Inference image with a top-down pose estimator. Args: model (nn.Module): The top-down pose estimator img (np.ndarray | str): The
mmpose/mmpose/apis/inference.py:133
↓ 7 callersMethodinit
(m)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/cnn/utils/weight_init.py:224
↓ 7 callersFunctionis_tuple_of
Check whether it is a tuple of some type. A partial method of :func:`is_seq_of`.
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/misc.py:160
↓ 7 callersMethodjoin_path
Concatenate all file paths. Join one or more filepath components intelligently. The return value is the concatenation of filepath and
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/fileio/file_client.py:1079
↓ 7 callersFunctionload_state_dict
Load state_dict to a module. This method is modified from :meth:`torch.nn.Module.load_state_dict`. Default value for ``strict`` is set to ``F
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/checkpoint.py:41
↓ 7 callersFunctionlog_txt_as_img
(wh, xc, size=10)
ControlNet-v1-1-nightly/ldm/util.py:11
↓ 7 callersFunctionmake_attn
(in_channels, attn_type="vanilla", attn_kwargs=None)
ControlNet-v1-1-nightly/ldm/modules/diffusionmodules/model.py:280
↓ 7 callersMethodnorm2
nn.Module: the normalization layer named "norm2"
mmpose/mmpose/models/backbones/hrnet.py:406
↓ 7 callersMethodnorm3
nn.Module: normalization layer after the third convolution layer
ControlNet-v1-1-nightly/annotator/uniformer/mmseg/models/backbones/resnet.py:260
↓ 7 callersFunctionobsolete_torch_version
(torch_version, version_threshold)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/cnn/bricks/wrappers.py:24
↓ 7 callersMethodprocess
Process one batch of data samples and predictions. The processed results should be stored in ``self.results``, which will be used to c
mmpose/mmpose/evaluation/metrics/keypoint_2d_metrics.py:788
↓ 7 callersFunctionround_channels
Round number of filters based on depth multiplier.
ControlNet-v1-1-nightly/annotator/normalbae/models/submodules/efficientnet_repo/geffnet/efficientnet_builder.py:82
← previousnext →201–300 of 7,103, ranked by callers