MCPcopy Create free account

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

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

↓ 1,274 callersMethodappend
(self, value)
ControlNet-v1-1-nightly/annotator/zoe/zoedepth/utils/misc.py:52
↓ 490 callersMethodsize
(self)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/video/io.py:27
↓ 299 callersMethodget
(self, key, default=None)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/video/io.py:37
↓ 201 callersMethodcat
Concatenates a list of Boxes into a single Boxes Arguments: boxes_list (list[Boxes]) Returns: Boxes
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/structures/boxes.py:279
↓ 164 callersMethodstack
(self)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/parallel/data_container.py:72
↓ 159 callersMethodnumpy
Convert all tensor to np.narray in data.
mmpose/mmpose/structures/multilevel_pixel_data.py:254
↓ 154 callersMethodcpu
Convert all tensors to CPU in data.
mmpose/mmpose/structures/multilevel_pixel_data.py:242
↓ 149 callersMethodget
Access an attribute and return its value if exists. Otherwise return default.
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/data/catalog.py:170
↓ 146 callersMethodclip
Clip (in place) the boxes by limiting x coordinates to the range [0, width] and y coordinates to the range [0, height]. Args
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/structures/boxes.py:183
↓ 146 callersMethodto
(self, device)
ControlNet-v1-1-nightly/annotator/zoe/zoedepth/models/depth_model.py:40
↓ 128 callersMethodpop
(self, k, d=None)
ControlNet-v1-1-nightly/annotator/zoe/zoedepth/utils/easydict/__init__.py:151
↓ 107 callersFunctionbuild_norm_layer
Build normalization layer. Args: cfg (dict): The norm layer config, which should contain: - type (str): Layer type.
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/cnn/bricks/norm.py:72
↓ 107 callersMethodupdate
Update a new data sample.
mmpose/mmpose/utils/timer.py:21
↓ 105 callersFunctionbuild_conv_layer
Build convolution layer. Args: cfg (None or dict): The conv layer config, which should contain: - type (str): Layer type.
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/cnn/bricks/conv.py:12
↓ 105 callersMethodflatten
(cls, obj)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/export/flatten.py:37
↓ 105 callersMethodopen
(self, fpath)
ControlNet-v1-1-nightly/annotator/zoe/zoedepth/data/data_mono.py:266
↓ 99 callersMethodbuild
(self, *args, **kwargs)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/registry.py:211
↓ 94 callersMethodcuda
Convert all tensors to GPU in data.
mmpose/mmpose/structures/multilevel_pixel_data.py:246
↓ 92 callersMethodwrite
(self)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/utils/events.py:105
↓ 91 callersMethoddetach
Detach all tensors in data.
mmpose/mmpose/structures/multilevel_pixel_data.py:250
↓ 91 callersMethoddim
(self)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/parallel/data_container.py:88
↓ 80 callersMethodcat
Cancatenates an instance list into one single instance. Similar to ``torch.cat``. Args: box_list (Sequence[T]): A sequenc
mmpose/projects/yolox_pose/datasets/bbox_keypoint_structure.py:193
↓ 78 callersFunctionexpand_dims
Expand the tensor `v` to the dim `dims`. Args: `v`: a PyTorch tensor with shape [N]. `dim`: a `int`. Returns: a P
ControlNet-v1-1-nightly/ldm/models/diffusion/dpm_solver/dpm_solver.py:1145
↓ 77 callersMethodload_state_dict
Loads the loss_scaler state dict. Args: state_dict (dict): scaler state.
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/fp16_utils.py:395
↓ 77 callersMethodlog
(self, runner)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/hooks/logger/base.py:35
↓ 71 callersMethodisfile
Check whether a file path is a file. Args: filepath (str or Path): Path to be checked whether it is a file. Returns:
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/fileio/file_client.py:1067
↓ 71 callersMethodto
(self, device: torch.device)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/structures/boxes.py:168
↓ 66 callersMethodupdate
(self, vars, count=1)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/log_buffer.py:24
↓ 61 callersMethodevaluate
Run evaluation for each partition.
mmpose/mmpose/evaluation/metrics/keypoint_partition_metric.py:194
↓ 58 callersMethodread
Read the next frame. If the next frame have been decoded before and in the cache, then return it directly, otherwise decode, cache an
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/video/io.py:134
↓ 57 callersMethoddevice
(self)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/structures/boxes.py:299
↓ 57 callersFunctionto_numpy
Convert torch tensor to numpy.ndarray. Args: x (Tensor | Sequence[Tensor]): A single tensor or a sequence of tensors
mmpose/mmpose/utils/tensor_utils.py:11
↓ 56 callersFunctionget_packed_inputs
Create a dummy batch of model inputs and data samples.
mmpose/mmpose/testing/_utils.py:81
↓ 55 callersMethodclone
Clone the Boxes. Returns: Boxes
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/structures/boxes.py:159
↓ 55 callersMethodhas
(self, name)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/export/c10.py:73
↓ 54 callersFunctionget_norm
Args: norm (str or callable): either one of BN, SyncBN, FrozenBN, GN; or a callable that takes a channel number and returns
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/layers/batch_norm.py:121
↓ 54 callersMethodregister_buffer
(self, name, attr)
ControlNet-v1-1-nightly/cldm/ddim_hacked.py:17
↓ 53 callersFunctiondigit_version
Convert a version string into a tuple of integers. This method is usually used for comparing two versions. For pre-release versions: alpha <
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/version_utils.py:9
↓ 53 callersMethodinit_weights
Initialize model weights.
mmpose/mmpose/models/necks/fpn.py:148
↓ 53 callersMethodload
Load model from file. Args: path (str): file path
ControlNet-v1-1-nightly/annotator/midas/midas/base_model.py:5
↓ 52 callersMethodget
(self)
ControlNet-v1-1-nightly/annotator/oneformer/oneformer/demo/predictor.py:161
↓ 49 callersMethodexists
Check whether a file path exists. Args: filepath (str or Path): Path to be checked whether exists. Returns:
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/fileio/file_client.py:1043
↓ 48 callersMethodtrain
(self, mode=True)
mmpose/mmpose/models/backbones/vgg.py:194
↓ 47 callersMethodlow_vram_shift
(self, is_diffusing)
ControlNet-v1-1-nightly/cldm/cldm.py:428
↓ 44 callersMethodnorm
(self)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/cnn/bricks/conv_module.py:169
↓ 43 callersMethodregister_buffer
(self, name, attr)
ControlNet-v1-1-nightly/ldm/models/diffusion/ddim.py:17
↓ 41 callersFunctionresize_image
(input_image, resolution)
ControlNet-v1-1-nightly/annotator/util.py:30
↓ 40 callersFunctionHWC3
(x)
ControlNet-v1-1-nightly/annotator/util.py:11
↓ 40 callersFunctionconstant_init
(module, val, bias=0)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/cnn/utils/weight_init.py:48
↓ 39 callersMethoddecode
(self, tokens)
ControlNet-v1-1-nightly/annotator/oneformer/oneformer/data/tokenizer.py:197
↓ 39 callersMethodpop
pop property in data and metainfo as the same as python.
mmpose/mmpose/structures/multilevel_pixel_data.py:202
↓ 38 callersFunction_gen_efficientnet
Creates an EfficientNet model. Ref impl: https://github.com/tensorflow/tpu/blob/master/models/official/efficientnet/efficientnet_model.py Pap
ControlNet-v1-1-nightly/annotator/normalbae/models/submodules/efficientnet_repo/geffnet/gen_efficientnet.py:502
↓ 37 callersMethodencode
Encode keypoints into root heatmaps and keypoint displacement fields. Note that the original keypoint coordinates should be in the inp
mmpose/mmpose/codecs/spr.py:138
↓ 37 callersMethodget_learned_conditioning
(self, c)
ControlNet-v1-1-nightly/ldm/models/diffusion/ddpm.py:664
↓ 35 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/coco_metric.py:143
↓ 34 callersMethoddecode_first_stage
(self, z, predict_cids=False, force_not_quantize=False)
ControlNet-v1-1-nightly/ldm/models/diffusion/ddpm.py:820
↓ 34 callersMethoddump
(self, file=None)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/config.py:534
↓ 34 callersFunctionexists
(x)
ControlNet-v1-1-nightly/ldm/util.py:47
↓ 34 callersMethodupdate
Args: predictions: D2 Instances for predictions of the current frame Return: D2 Instances for predictions of
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/tracking/base_tracker.py:32
↓ 33 callersMethodset_field
Special method for set union field, used as property.setter functions.
mmpose/mmpose/structures/multilevel_pixel_data.py:116
↓ 30 callersFunctionload_state_dict
(ckpt_path, location='cpu')
ControlNet-v1-1-nightly/cldm/model.py:12
↓ 30 callersMethodset
Set multiple metadata with kwargs.
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/data/catalog.py:162
↓ 29 callersMethodbuild_posetrack18_video_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_body_datasets/test_posetrack18_video_dataset.py:11
↓ 29 callersFunctioncat
Efficient version of torch.cat that avoids a copy if there is only a single element in a list
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/layers/wrappers.py:50
↓ 29 callersMethodsave
Save a config object to a yaml file. Note that when the config dictionary contains complex objects (e.g. lambda), it can't be
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/config/lazy.py:251
↓ 27 callersMethoddecode
Decode the keypoint coordinates from heatmaps and displacements. The decoded keypoint coordinates are in the input image space. Args:
mmpose/mmpose/codecs/spr.py:222
↓ 26 callersMethodregister_module
Register a module. A record will be added to `self._module_dict`, whose key is the class name or the specified name, and value is the
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/registry.py:260
↓ 26 callersFunctionresize
(input, size=None, scale_factor=None, mode='nearest', align_corner
ControlNet-v1-1-nightly/annotator/uniformer/mmseg/ops/wrappers.py:7
↓ 25 callersFunctionextract_into_tensor
(a, t, x_shape)
ControlNet-v1-1-nightly/ldm/modules/diffusionmodules/util.py:96
↓ 25 callersMethodput_scalar
Add a scalar `value` to the `HistoryBuffer` associated with `name`. Args: smoothing_hint (bool): a 'hint' on whether thi
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/utils/events.py:338
↓ 24 callersFunctionget_dist_info
()
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/dist_utils.py:78
↓ 23 callersFunctionconv_nd
Create a 1D, 2D, or 3D convolution module.
ControlNet-v1-1-nightly/ldm/modules/diffusionmodules/util.py:221
↓ 23 callersMethodconvert
Args: box: can be a k-tuple, k-list or an Nxk array/tensor, where k = 4 or 5 from_mode, to_mode (BoxMode) Re
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/structures/boxes.py:44
↓ 22 callersMethodencode
(self, text)
ControlNet-v1-1-nightly/annotator/oneformer/oneformer/data/tokenizer.py:189
↓ 22 callersMethodget_local_path
Download data from ``filepath`` and write the data to local path. ``get_local_path`` is decorated by :meth:`contxtlib.contextmanager`. It
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/fileio/file_client.py:1095
↓ 22 callersMethodnorm1
nn.Module: the normalization layer named "norm1"
ControlNet-v1-1-nightly/annotator/uniformer/mmseg/models/backbones/hrnet.py:362
↓ 22 callersMethodnorm2
nn.Module: the normalization layer named "norm2"
ControlNet-v1-1-nightly/annotator/uniformer/mmseg/models/backbones/hrnet.py:367
↓ 21 callersMethod_transform_inputs
Transform inputs for decoder. Args: inputs (list[Tensor]): List of multi-level img features. Returns: Tensor
ControlNet-v1-1-nightly/annotator/uniformer/mmseg/models/decode_heads/decode_head.py:137
↓ 21 callersMethodfromfile
(filename, use_predefined_variables=True, import_custom_modules=True)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/config.py:328
↓ 21 callersFunctionfunc
(x, weights, bias=None, stride=1, padding=0, dilation=1, groups=1)
ControlNet-v1-1-nightly/annotator/pidinet/model.py:276
↓ 21 callersFunctionget_coco_sample
Create a dummy data sample in COCO style.
mmpose/mmpose/testing/_utils.py:16
↓ 21 callersFunctionget_event_storage
Returns: The :class:`EventStorage` object that's currently being used. Throws an error if no :class:`EventStorage` is currently e
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/utils/events.py:26
↓ 21 callersFunctionload
Load data from json/yaml/pickle files. This method provides a unified api for loading data from serialized files. Note: In v1.3.16 a
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/fileio/io.py:18
↓ 21 callersFunctionparse_pose_metainfo
Load meta information of pose dataset and check its integrity. Args: metainfo (dict): Raw data of pose meta information, which should
mmpose/mmpose/datasets/datasets/utils.py:9
↓ 21 callersMethodto
(self, device)
ControlNet-v1-1-nightly/annotator/oneformer/oneformer/utils/misc.py:116
↓ 21 callersFunctionto_device
This function is a replacement of .to(another_device) such that it allows the casting to be traced properly by explicitly calling the underly
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/export/shared.py:24
↓ 20 callersMethodcls_seg
Classify each pixel with fc.
ControlNet-v1-1-nightly/annotator/uniformer/mmseg/models/decode_heads/point_head.py:108
↓ 20 callersMethodlog
mmpose/projects/rtmpose/examples/RTMPose-Deploy/Windows/TensorRT/src/RTMPoseTensorRT/main.cpp:20
↓ 20 callersMethodnorm
(c)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/modeling/meta_arch/retinanet.py:350
↓ 19 callersFunctionbuild_from_cfg
Build a module from config dict. Args: cfg (dict): Config dict. It should at least contain the key "type". registry (:obj:`Regist
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/registry.py:9
↓ 19 callersFunctionis_list_of
Check whether it is a list of some type. A partial method of :func:`is_seq_of`.
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/misc.py:152
↓ 19 callersFunctionkeypoint_pck_accuracy
Calculate the pose accuracy of PCK for each individual keypoint and the averaged accuracy across all keypoints for coordinates. Note:
mmpose/mmpose/evaluation/functional/keypoint_eval.py:67
↓ 19 callersMethodload
Load a config file. Args: filename: absolute path or relative path w.r.t. the current working directory keys
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/config/lazy.py:196
↓ 19 callersMethodmarginal_lambda
Compute lambda_t = log(alpha_t) - log(sigma_t) of a given continuous-time label t in [0, T].
ControlNet-v1-1-nightly/ldm/models/diffusion/dpm_solver/dpm_solver.py:132
↓ 19 callersMethodto_rgb
(self, x)
ControlNet-v1-1-nightly/ldm/models/autoencoder.py:192
↓ 18 callersMethodcall_hook
Call all hooks. Args: fn_name (str): The function name in each hook to be called, such as "before_train_epoch".
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/base_runner.py:299
↓ 18 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/resnet.py:281
↓ 18 callersMethodmarginal_std
Compute sigma_t of a given continuous-time label t in [0, T].
ControlNet-v1-1-nightly/ldm/models/diffusion/dpm_solver/dpm_solver.py:126
↓ 18 callersMethodmeshgrid
(self, h, w)
ControlNet-v1-1-nightly/ldm/models/diffusion/ddpm.py:677
↓ 17 callersMethodchildren
(self)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/registry.py:183
next →1–100 of 7,103, ranked by callers