Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/IDEA-Research/DWPose
/ functions
Functions
7,103 in github.com/IDEA-Research/DWPose
⨍
Functions
7,103
◇
Types & classes
1,442
↳
Endpoints
6
↓ 1,274 callers
Method
append
(self, value)
ControlNet-v1-1-nightly/annotator/zoe/zoedepth/utils/misc.py:52
↓ 490 callers
Method
size
(self)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/video/io.py:27
↓ 299 callers
Method
get
(self, key, default=None)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/video/io.py:37
↓ 201 callers
Method
cat
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 callers
Method
stack
(self)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/parallel/data_container.py:72
↓ 159 callers
Method
numpy
Convert all tensor to np.narray in data.
mmpose/mmpose/structures/multilevel_pixel_data.py:254
↓ 154 callers
Method
cpu
Convert all tensors to CPU in data.
mmpose/mmpose/structures/multilevel_pixel_data.py:242
↓ 149 callers
Method
get
Access an attribute and return its value if exists. Otherwise return default.
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/data/catalog.py:170
↓ 146 callers
Method
clip
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 callers
Method
to
(self, device)
ControlNet-v1-1-nightly/annotator/zoe/zoedepth/models/depth_model.py:40
↓ 128 callers
Method
pop
(self, k, d=None)
ControlNet-v1-1-nightly/annotator/zoe/zoedepth/utils/easydict/__init__.py:151
↓ 107 callers
Function
build_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 callers
Method
update
Update a new data sample.
mmpose/mmpose/utils/timer.py:21
↓ 105 callers
Function
build_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 callers
Method
flatten
(cls, obj)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/export/flatten.py:37
↓ 105 callers
Method
open
(self, fpath)
ControlNet-v1-1-nightly/annotator/zoe/zoedepth/data/data_mono.py:266
↓ 99 callers
Method
build
(self, *args, **kwargs)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/registry.py:211
↓ 94 callers
Method
cuda
Convert all tensors to GPU in data.
mmpose/mmpose/structures/multilevel_pixel_data.py:246
↓ 92 callers
Method
write
(self)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/utils/events.py:105
↓ 91 callers
Method
detach
Detach all tensors in data.
mmpose/mmpose/structures/multilevel_pixel_data.py:250
↓ 91 callers
Method
dim
(self)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/parallel/data_container.py:88
↓ 80 callers
Method
cat
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 callers
Function
expand_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 callers
Method
load_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 callers
Method
log
(self, runner)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/hooks/logger/base.py:35
↓ 71 callers
Method
isfile
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 callers
Method
to
(self, device: torch.device)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/structures/boxes.py:168
↓ 66 callers
Method
update
(self, vars, count=1)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/log_buffer.py:24
↓ 61 callers
Method
evaluate
Run evaluation for each partition.
mmpose/mmpose/evaluation/metrics/keypoint_partition_metric.py:194
↓ 58 callers
Method
read
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 callers
Method
device
(self)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/structures/boxes.py:299
↓ 57 callers
Function
to_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 callers
Function
get_packed_inputs
Create a dummy batch of model inputs and data samples.
mmpose/mmpose/testing/_utils.py:81
↓ 55 callers
Method
clone
Clone the Boxes. Returns: Boxes
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/structures/boxes.py:159
↓ 55 callers
Method
has
(self, name)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/export/c10.py:73
↓ 54 callers
Function
get_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 callers
Method
register_buffer
(self, name, attr)
ControlNet-v1-1-nightly/cldm/ddim_hacked.py:17
↓ 53 callers
Function
digit_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 callers
Method
init_weights
Initialize model weights.
mmpose/mmpose/models/necks/fpn.py:148
↓ 53 callers
Method
load
Load model from file. Args: path (str): file path
ControlNet-v1-1-nightly/annotator/midas/midas/base_model.py:5
↓ 52 callers
Method
get
(self)
ControlNet-v1-1-nightly/annotator/oneformer/oneformer/demo/predictor.py:161
↓ 49 callers
Method
exists
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 callers
Method
train
(self, mode=True)
mmpose/mmpose/models/backbones/vgg.py:194
↓ 47 callers
Method
low_vram_shift
(self, is_diffusing)
ControlNet-v1-1-nightly/cldm/cldm.py:428
↓ 44 callers
Method
norm
(self)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/cnn/bricks/conv_module.py:169
↓ 43 callers
Method
register_buffer
(self, name, attr)
ControlNet-v1-1-nightly/ldm/models/diffusion/ddim.py:17
↓ 41 callers
Function
resize_image
(input_image, resolution)
ControlNet-v1-1-nightly/annotator/util.py:30
↓ 40 callers
Function
HWC3
(x)
ControlNet-v1-1-nightly/annotator/util.py:11
↓ 40 callers
Function
constant_init
(module, val, bias=0)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/cnn/utils/weight_init.py:48
↓ 39 callers
Method
decode
(self, tokens)
ControlNet-v1-1-nightly/annotator/oneformer/oneformer/data/tokenizer.py:197
↓ 39 callers
Method
pop
pop property in data and metainfo as the same as python.
mmpose/mmpose/structures/multilevel_pixel_data.py:202
↓ 38 callers
Function
_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 callers
Method
encode
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 callers
Method
get_learned_conditioning
(self, c)
ControlNet-v1-1-nightly/ldm/models/diffusion/ddpm.py:664
↓ 35 callers
Method
process
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 callers
Method
decode_first_stage
(self, z, predict_cids=False, force_not_quantize=False)
ControlNet-v1-1-nightly/ldm/models/diffusion/ddpm.py:820
↓ 34 callers
Method
dump
(self, file=None)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/config.py:534
↓ 34 callers
Function
exists
(x)
ControlNet-v1-1-nightly/ldm/util.py:47
↓ 34 callers
Method
update
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 callers
Method
set_field
Special method for set union field, used as property.setter functions.
mmpose/mmpose/structures/multilevel_pixel_data.py:116
↓ 30 callers
Function
load_state_dict
(ckpt_path, location='cpu')
ControlNet-v1-1-nightly/cldm/model.py:12
↓ 30 callers
Method
set
Set multiple metadata with kwargs.
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/data/catalog.py:162
↓ 29 callers
Method
build_posetrack18_video_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_body_datasets/test_posetrack18_video_dataset.py:11
↓ 29 callers
Function
cat
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 callers
Method
save
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 callers
Method
decode
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 callers
Method
register_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 callers
Function
resize
(input, size=None, scale_factor=None, mode='nearest', align_corner
ControlNet-v1-1-nightly/annotator/uniformer/mmseg/ops/wrappers.py:7
↓ 25 callers
Function
extract_into_tensor
(a, t, x_shape)
ControlNet-v1-1-nightly/ldm/modules/diffusionmodules/util.py:96
↓ 25 callers
Method
put_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 callers
Function
get_dist_info
()
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/dist_utils.py:78
↓ 23 callers
Function
conv_nd
Create a 1D, 2D, or 3D convolution module.
ControlNet-v1-1-nightly/ldm/modules/diffusionmodules/util.py:221
↓ 23 callers
Method
convert
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 callers
Method
encode
(self, text)
ControlNet-v1-1-nightly/annotator/oneformer/oneformer/data/tokenizer.py:189
↓ 22 callers
Method
get_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 callers
Method
norm1
nn.Module: the normalization layer named "norm1"
ControlNet-v1-1-nightly/annotator/uniformer/mmseg/models/backbones/hrnet.py:362
↓ 22 callers
Method
norm2
nn.Module: the normalization layer named "norm2"
ControlNet-v1-1-nightly/annotator/uniformer/mmseg/models/backbones/hrnet.py:367
↓ 21 callers
Method
_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 callers
Method
fromfile
(filename, use_predefined_variables=True, import_custom_modules=True)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/config.py:328
↓ 21 callers
Function
func
(x, weights, bias=None, stride=1, padding=0, dilation=1, groups=1)
ControlNet-v1-1-nightly/annotator/pidinet/model.py:276
↓ 21 callers
Function
get_coco_sample
Create a dummy data sample in COCO style.
mmpose/mmpose/testing/_utils.py:16
↓ 21 callers
Function
get_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 callers
Function
load
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 callers
Function
parse_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 callers
Method
to
(self, device)
ControlNet-v1-1-nightly/annotator/oneformer/oneformer/utils/misc.py:116
↓ 21 callers
Function
to_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 callers
Method
cls_seg
Classify each pixel with fc.
ControlNet-v1-1-nightly/annotator/uniformer/mmseg/models/decode_heads/point_head.py:108
↓ 20 callers
Method
log
mmpose/projects/rtmpose/examples/RTMPose-Deploy/Windows/TensorRT/src/RTMPoseTensorRT/main.cpp:20
↓ 20 callers
Method
norm
(c)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/modeling/meta_arch/retinanet.py:350
↓ 19 callers
Function
build_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 callers
Function
is_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 callers
Function
keypoint_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 callers
Method
load
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 callers
Method
marginal_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 callers
Method
to_rgb
(self, x)
ControlNet-v1-1-nightly/ldm/models/autoencoder.py:192
↓ 18 callers
Method
call_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 callers
Function
get_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 callers
Method
marginal_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 callers
Method
meshgrid
(self, h, w)
ControlNet-v1-1-nightly/ldm/models/diffusion/ddpm.py:677
↓ 17 callers
Method
children
(self)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/registry.py:183
next →
1–100 of 7,103, ranked by callers