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
↓ 17 callersMethodclear
(self)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/log_buffer.py:15
↓ 17 callersMethodclose
(self)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/utils/events.py:127
↓ 17 callersFunctionget_activation
(name)
ControlNet-v1-1-nightly/annotator/zoe/zoedepth/models/base_models/midas_repo/midas/backbones/utils.py:56
↓ 16 callersMethoddata
(self)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/parallel/data_container.py:57
↓ 16 callersFunctionkaiming_init
(module, a=0, mode='fan_out', nonlinearity='relu',
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/cnn/utils/weight_init.py:92
↓ 16 callersMethodmarginal_log_mean_coeff
Compute log(alpha_t) of a given continuous-time label t in [0, T].
ControlNet-v1-1-nightly/ldm/models/diffusion/dpm_solver/dpm_solver.py:106
↓ 16 callersFunctionprint_log
Print a log message. Args: msg (str): The message to be logged. logger (logging.Logger | str | None): The logger to be used.
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/logging.py:85
↓ 16 callersMethodremove
Remove a file. Args: filepath (str, Path): Path to be removed.
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/fileio/file_client.py:1035
↓ 16 callersMethodsample
(self, S, batch_size, shape, conditioning=None,
ControlNet-v1-1-nightly/cldm/ddim_hacked.py:55
↓ 15 callersFunctionalias
(x, name, is_backward=False)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/export/shared.py:597
↓ 15 callersFunctionbuild_activation_layer
Build activation layer. Args: cfg (dict): The activation layer config, which should contain: - type (str): Layer type.
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/cnn/bricks/activation.py:81
↓ 15 callersFunctioncreate_model
(config_path)
ControlNet-v1-1-nightly/cldm/model.py:24
↓ 15 callersMethodmodel_fn
Convert the model to the noise prediction model or the data prediction model.
ControlNet-v1-1-nightly/ldm/models/diffusion/dpm_solver/dpm_solver.py:367
↓ 15 callersMethodq_sample
(self, x_start, t, noise=None)
ControlNet-v1-1-nightly/ldm/models/diffusion/ddpm.py:356
↓ 15 callersMethodregister
Args: name (str): the name that identifies a dataset, e.g. "coco_2014_train". func (callable): a callable which takes
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/data/catalog.py:29
↓ 15 callersMethodrun
(self, data_loaders, workflow, **kwargs)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/base_runner.py:193
↓ 15 callersMethodstate_dict
Returns the state of the scaler as a :class:`dict`.
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/fp16_utils.py:385
↓ 15 callersMethodstep
User should either: (1) Call this function to increment storage.iter when needed. Or (2) Set `storage.iter` to the correct iteration
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/utils/events.py:475
↓ 15 callersMethodtransform
In-place transform all attributes of this class. By "in-place", it means after calling this method, accessing an attribute such
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/data/transforms/augmentation.py:331
↓ 14 callersMethod__init__
(self, *, ch, out_ch, ch_mult=(1,2,4,8), num_res_blocks, attn_resolutions, dropout=0.0, resam
ControlNet-v1-1-nightly/ldm/modules/diffusionmodules/model.py:301
↓ 14 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_rtmcc_head.py:19
↓ 14 callersMethod_init
(self, params=None)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/data/transforms/augmentation.py:109
↓ 14 callersFunction_make_scratch
(in_shape, out_shape, groups=1, expand=False)
ControlNet-v1-1-nightly/annotator/zoe/zoedepth/models/base_models/midas_repo/midas/blocks.py:133
↓ 14 callersFunctiondecode_arch_def
(arch_def, depth_multiplier=1.0, depth_trunc='ceil', experts_multiplier=1, fix_first_last=False)
ControlNet-v1-1-nightly/annotator/normalbae/models/submodules/efficientnet_repo/geffnet/efficientnet_builder.py:620
↓ 14 callersMethodisdir
Check whether a file path is a directory. Args: filepath (str or Path): Path to be checked whether it is a direct
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/fileio/file_client.py:1054
↓ 14 callersFunctionlayer_config_kwargs
Consume config kwargs and return contextmgr obj
ControlNet-v1-1-nightly/annotator/normalbae/models/submodules/efficientnet_repo/geffnet/config.py:118
↓ 14 callersFunctionload_checkpoint
Load checkpoint from a file or URI. Args: model (Module): Module to load checkpoint. filename (str): Accept local filepath, URL,
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/checkpoint.py:503
↓ 14 callersFunctionresolve_bn_args
(kwargs)
ControlNet-v1-1-nightly/annotator/normalbae/models/submodules/efficientnet_repo/geffnet/efficientnet_builder.py:34
↓ 14 callersMethodset
Set the field named `name` to `value`. The length of `value` must be the number of instances, and must agree with other exist
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/structures/instances.py:69
↓ 13 callersMethodbuild_h36m_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_body_datasets/test_h36m_dataset.py:11
↓ 13 callersFunctioncheck_set_pb_arg
(pb, arg_name, arg_attr, arg_value, allow_override=False)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/export/shared.py:209
↓ 13 callersFunctionconv2d
Helper for building a conv2d layer.
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/modeling/backbone/regnet.py:28
↓ 13 callersMethodcriterion
Criterion of wingloss. Note: - batch_size: N - num_keypoints: K - dimension of keypoints: D (D=2 or D=3)
mmpose/mmpose/models/losses/regression_loss.py:248
↓ 13 callersMethoddecode
Decode keypoints from outputs. Args: heatmaps (Tuple[Tensor]): The output heatmaps inferred from one image or mul
mmpose/mmpose/models/heads/hybrid_heads/dekr_head.py:476
↓ 13 callersMethodget_image
Get the drawn image. The format is RGB. Returns: np.ndarray: the drawn image which channel is RGB.
mmpose/mmpose/visualization/opencv_backend_visualizer.py:73
↓ 13 callersFunctioninstantiate_from_config
(config)
ControlNet-v1-1-nightly/ldm/util.py:72
↓ 13 callersMethodload
Load model from file. Args: path (str): file path
ControlNet-v1-1-nightly/annotator/zoe/zoedepth/models/base_models/midas_repo/midas/base_model.py:5
↓ 13 callersMethodregister_buffer
(self, name, attr)
ControlNet-v1-1-nightly/ldm/models/diffusion/plms.py:19
↓ 13 callersMethodset_dataset_meta
Assign dataset_meta to the visualizer. The default visualization settings will be overridden. Args: dataset_meta (dict):
mmpose/mmpose/visualization/local_visualizer.py:141
↓ 13 callersMethodshow
Show the drawn image. Args: drawn_img (np.ndarray, optional): The image to show. If drawn_img is None, it will sh
mmpose/mmpose/visualization/opencv_backend_visualizer.py:427
↓ 12 callersMethod__init__
Args: shape: (h, w) tuple or a int interp: PIL interpolation method
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/data/transforms/augmentation_impl.py:117
↓ 12 callersFunction_create_model
(model_kwargs, variant, pretrained=False)
ControlNet-v1-1-nightly/annotator/normalbae/models/submodules/efficientnet_repo/geffnet/gen_efficientnet.py:286
↓ 12 callersFunction_gen_mobilenet_v3
Creates a MobileNet-V3 large/small/minimal models. Ref impl: https://github.com/tensorflow/models/blob/master/research/slim/nets/mobilenet/mobile
ControlNet-v1-1-nightly/annotator/normalbae/models/submodules/efficientnet_repo/geffnet/mobilenetv3.py:171
↓ 12 callersMethodaccumulate
Accumulate per image evaluation results and store the result in self.eval. Does not support changing parameter settings from those u
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/evaluation/fast_eval_api.py:98
↓ 12 callersMethodadd_datasample
Draw datasample and save to all backends. - If GT and prediction are plotted at the same time, they are displayed in a stitched image
mmpose/mmpose/visualization/local_visualizer.py:454
↓ 12 callersMethodbuild_deepfashion_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_fashion_datasets/test_deepfashion_dataset.py:11
↓ 12 callersMethodinit_weights
(self, pretrained=None)
ControlNet-v1-1-nightly/annotator/uniformer/mmseg/models/backbones/mobilenet_v2.py:136
↓ 12 callersFunctionnormal_init
(module, mean=0, std=1, bias=0)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/cnn/utils/weight_init.py:66
↓ 12 callersFunctionresolve_act_layer
(kwargs, default='relu')
ControlNet-v1-1-nightly/annotator/normalbae/models/submodules/efficientnet_repo/geffnet/efficientnet_builder.py:67
↓ 12 callersFunctionselect_conv2d
(in_chs, out_chs, kernel_size, **kwargs)
ControlNet-v1-1-nightly/annotator/normalbae/models/submodules/efficientnet_repo/geffnet/conv2d_layers.py:290
↓ 12 callersMethodupdate
(self, val, n=1)
ControlNet-v1-1-nightly/annotator/normalbae/models/submodules/efficientnet_repo/utils.py:15
↓ 11 callersMethod__setattr__
(self, name, value)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/config.py:512
↓ 11 callersMethod_get_data_samples
(self, batch_size: int = 2, heatmap_size=(48, 64),
mmpose/tests/test_models/test_heads/test_heatmap_heads/test_mspn_head.py:33
↓ 11 callersMethod_get_feats
( self, num_stages: int = 1, num_units: int = 4, batch_size: int = 2,
mmpose/tests/test_models/test_heads/test_heatmap_heads/test_mspn_head.py:15
↓ 11 callersMethodbuild_coco_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_body_datasets/test_coco_dataset.py:11
↓ 11 callersMethodbuild_coco_wholebody_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_wholebody_datasets/test_coco_wholebody_dataset.py:11
↓ 11 callersMethodbuild_crowdpose_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_body_datasets/test_crowdpose_dataset.py:11
↓ 11 callersMethodbuild_halpe_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_wholebody_datasets/test_halpe_dataset.py:11
↓ 11 callersMethodbuild_humanart_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_body_datasets/test_humanart_dataset.py:11
↓ 11 callersMethodbuild_posetrack18_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_body_datasets/test_posetrack18_dataset.py:11
↓ 11 callersMethodclone
Reload ``clone`` for tensors.
mmpose/projects/yolox_pose/datasets/bbox_keypoint_structure.py:269
↓ 11 callersMethodcrop
Crop from ``img``
ControlNet-v1-1-nightly/annotator/uniformer/mmseg/datasets/pipelines/transforms.py:523
↓ 11 callersMethodgetConfidence
()
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:166
↓ 11 callersMethodget_input
(self, batch, k)
ControlNet-v1-1-nightly/ldm/models/diffusion/ddpm.py:419
↓ 11 callersMethodinit_weights
(self, pretrained=None)
mmpose/mmpose/models/backbones/shufflenet_v1.py:280
↓ 11 callersFunctionis_seq_of
Check whether it is a sequence of some type. Args: seq (Sequence): The sequence to be checked. expected_type (type): Expected typ
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/misc.py:128
↓ 11 callersFunctionis_str
Whether the input is an string instance. Note: This method is deprecated since python 2 is no longer supported.
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/misc.py:31
↓ 11 callersMethodoutput_shape
(self)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/modeling/backbone/fpn.py:169
↓ 11 callersMethodpredict
Predict results from features. Args: feats (Tuple[Tensor] | List[Tuple[Tensor]]): The multi-stage features (or mu
mmpose/mmpose/models/heads/coord_cls_heads/rtmcc_head.py:163
↓ 11 callersMethodregister_hook
Register a hook into the hook list. The hook will be inserted into a priority queue, with the specified priority (See :class:`Priorit
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/base_runner.py:255
↓ 11 callersFunctionretry_if_cuda_oom
Makes a function retry itself after encountering pytorch's CUDA OOM error. It will first retry after calling `torch.cuda.empty_cache()`.
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/utils/memory.py:26
↓ 11 callersFunctionto_tensor
Convert numpy.ndarray to torch tensor. Args: x (np.ndarray | Sequence[np.ndarray]): A single np.ndarray or a sequence of tens
mmpose/mmpose/utils/tensor_utils.py:53
↓ 11 callersMethodtrain
(self, mode=True)
mmpose/mmpose/models/backbones/shufflenet_v1.py:332
↓ 11 callersMethodtrain
(self, mode=True)
mmpose/mmpose/models/backbones/mobilenet_v2.py:273
↓ 11 callersMethodupdate
(self, num_tasks=1)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/utils/progressbar.py:35
↓ 10 callersMethod__init__
(self, by_epoch=True, warmup=None, warmup_iters=0,
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/hooks/lr_updater.py:25
↓ 10 callersMethod__init__
(self, w_in, w_se, activation_class)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/modeling/backbone/regnet.py:94
↓ 10 callersMethod_format_path
Convert a ``filepath`` to standard format of petrel oss. If the ``filepath`` is concatenated by ``os.path.join``, in a Windows enviro
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/fileio/file_client.py:134
↓ 10 callersFunction_gen_efficientnet_lite
Creates an EfficientNet-Lite model. Ref impl: https://github.com/tensorflow/tpu/tree/master/models/official/efficientnet/lite Paper: https://
ControlNet-v1-1-nightly/annotator/normalbae/models/submodules/efficientnet_repo/geffnet/gen_efficientnet.py:599
↓ 10 callersMethod_map_path
Map ``filepath`` to a string path whose prefix will be replaced by :attr:`self.path_mapping`. Args: filepath (str): Path
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/fileio/file_client.py:121
↓ 10 callersMethodarea
Computes the area of all the boxes. Returns: torch.Tensor: a vector with areas of each box.
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/structures/boxes.py:172
↓ 10 callersMethodbackward
(ctx, a=None)
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/ops/knn.py:73
↓ 10 callersFunctionbuild_pose_estimator
Build pose estimator.
mmpose/mmpose/models/builder.py:33
↓ 10 callersFunctionget_activation
(name)
ControlNet-v1-1-nightly/annotator/midas/midas/vit.py:159
↓ 10 callersFunctionget_activation
(name)
ControlNet-v1-1-nightly/ldm/modules/midas/midas/vit.py:159
↓ 10 callersMethodget_iter
Get the current training iteration step.
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/hooks/logger/base.py:84
↓ 10 callersFunctionget_pb_arg
(pb, arg_name)
ControlNet-v1-1-nightly/annotator/oneformer/detectron2/export/shared.py:172
↓ 10 callersFunctioninit_model
Initialize a pose estimator from a config file. Args: config (str, :obj:`Path`, or :obj:`mmengine.Config`): Config file path,
mmpose/mmpose/apis/inference.py:67
↓ 10 callersMethodinit_weights
(self)
mmpose/mmpose/models/backbones/shufflenet_v2.py:282
↓ 10 callersMethodinit_weights
Initialize the weights in backbone.
mmpose/mmpose/models/backbones/resnet.py:660
↓ 10 callersMethodload_checkpoint
(self, filename, map_location='cpu', s
ControlNet-v1-1-nightly/annotator/uniformer/mmcv/runner/base_runner.py:332
↓ 10 callersFunctionmerge_data_samples
Merge the given data samples into a single data sample. This function can be used to merge the top-down predictions with bboxes from the same
mmpose/mmpose/structures/utils.py:15
↓ 10 callersMethodremove
(self)
mmpose/mmpose/utils/hooks.py:42
↓ 9 callersFunctionNormalize
(in_channels, num_groups=32)
ControlNet-v1-1-nightly/ldm/modules/diffusionmodules/model.py:46
↓ 9 callersMethod__init__
(self, use_target_weight=False, loss_weight=1.)
mmpose/mmpose/models/losses/regression_loss.py:412
↓ 9 callersMethodapply_model
(self, x_noisy, t, cond, return_ids=False)
ControlNet-v1-1-nightly/ldm/models/diffusion/ddpm.py:850
↓ 9 callersMethodbuild_aflw_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_face_datasets/test_aflw_dataset.py:11
↓ 9 callersMethodbuild_aic_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_body_datasets/test_aic_dataset.py:11
↓ 9 callersMethodbuild_ak_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_animal_datasets/test_animalkingdom_dataset.py:11
↓ 9 callersMethodbuild_animalpose_dataset
(self, **kwargs)
mmpose/tests/test_datasets/test_datasets/test_animal_datasets/test_animalpose_dataset.py:11
← previousnext →101–200 of 7,103, ranked by callers