MCPcopy Create free account

hub / github.com/MotrixLab/ADHMR / functions

Functions2,101 in github.com/MotrixLab/ADHMR

↓ 361 callersFunctionprint
(*args, **kwargs)
HMR-Scorer/common/utils/distribute_utils.py:33
↓ 174 callersMethodto
(self, device)
ADHMR/lib/models/ema.py:113
↓ 94 callersMethodkeys
(self)
ADHMR/lib/dataset/humandata_utils/smplx/smplx/utils.py:43
↓ 92 callersMethodpermute
(self, x, size)
HMR-Scorer/main/transformer_utils/mmpose/models/backbones/modules/multihead_isa_attention.py:411
↓ 74 callersMethodload
(self, load_path)
HMR-Scorer/data/PW3D_DPO.py:165
↓ 70 callersMethodinfo
(self, msg)
HMR-Scorer/common/logger.py:39
↓ 61 callersMethodload
(self, load_path)
ADHMR/lib/dataset/humandata.py:53
↓ 56 callersMethodget
(self, key, default=None)
HMR-Scorer/common/utils/smplx/smplx/utils.py:37
↓ 54 callersMethoditems
(self)
HMR-Scorer/common/utils/smplx/smplx/utils.py:51
↓ 44 callersMethodto
(self, device)
HMR-Scorer/common/nets/ema.py:113
↓ 43 callersMethodkeys
(self)
ADHMR/lib/utils/smplx/smplx/utils.py:43
↓ 42 callersMethodkeys
(self)
HMR-Scorer/common/utils/smplx/smplx/utils.py:43
↓ 42 callersMethodload
(self, load_path)
ADHMR/lib/utils/cache.py:11
↓ 33 callersFunctionget_affine_transform
(center, scale, rot, output_size,
ADHMR/lib/utils/transforms.py:1336
↓ 29 callersFunction_center_scale_to_box
(center, scale)
ADHMR/lib/utils/bbox.py:284
↓ 29 callersFunctionto_tensor
( array: Union[Array, Tensor], dtype=torch.float32 )
HMR-Scorer/common/utils/smplx/smplx/utils.py:98
↓ 28 callersMethodget
(self, key, default=None)
ADHMR/lib/utils/smplx/smplx/utils.py:37
↓ 25 callersFunctionaffine_transform
(pt, t)
ADHMR/lib/utils/transforms.py:1408
↓ 22 callersFunction_box_to_center_scale
Convert box coordinates to center and scale. adapted from https://github.com/Microsoft/human-pose-estimation.pytorch
ADHMR/lib/utils/bbox.py:264
↓ 22 callersFunctionflip_joints_3d
Flip 3d joints. Parameters ---------- joints_3d : numpy.ndarray Joints in shape (num_joints, 3, 2) width : int Image
ADHMR/lib/utils/transforms.py:899
↓ 22 callersFunctionim_to_torch
Transform ndarray image to torch tensor. Parameters ---------- img: numpy.ndarray An ndarray with shape: `(H, W, 3)`. Return
ADHMR/lib/utils/transforms.py:75
↓ 22 callersMethodscatter
Scatter function. Args: inputs (Tensor): Input Tensor. kwargs (dict): Args for ``mmcv.parallel.scatte
HMR-Scorer/main/transformer_utils/mmpose/core/distributed_wrapper.py:97
↓ 20 callersFunctionlbs
Performs Linear Blend Skinning with the given shape and pose parameters Parameters ---------- betas : torch.tensor BxNB
HMR-Scorer/common/utils/smplx/smplx/lbs.py:153
↓ 20 callersMethodload_state_dict
(self, state_dict)
ADHMR/lib/models/ema.py:108
↓ 18 callersFunctionto_np
(array, dtype=np.float32)
HMR-Scorer/common/utils/smplx/smplx/utils.py:113
↓ 17 callersMethodcriterion
Criterion of wingloss. Note: - batch_size: N - num_keypoints: K - dimension of keypoints: D (D=2 or D=3)
HMR-Scorer/main/transformer_utils/mmpose/models/losses/regression_loss.py:161
↓ 17 callersMethodstate_dict
(self)
ADHMR/lib/models/ema.py:104
↓ 17 callersMethodupdate
Update a new data sample.
HMR-Scorer/main/transformer_utils/mmpose/utils/timer.py:21
↓ 16 callersFunctionprocess_bbox
保持检测框中心点位置不变调整检测框比例为cfg.input_img_size ration: 扩大检测框大小
HMR-Scorer/common/utils/preprocessing.py:61
↓ 16 callersFunctionrigid_align
(A, B)
HMR-Scorer/common/utils/transforms.py:53
↓ 15 callersMethodload_state_dict
(self, state_dict)
HMR-Scorer/common/nets/ema.py:108
↓ 15 callersFunctionto_tensor
( array: Union[Array, Tensor], dtype=torch.float32 )
ADHMR/lib/utils/smplx/smplx/utils.py:98
↓ 15 callersFunctionto_tensor
( array: Union[Array, Tensor], dtype=torch.float32 )
ADHMR/lib/dataset/humandata_utils/smplx/smplx/utils.py:98
↓ 15 callersMethodupdate
Update currently maintained parameters. Call this every time the parameters are updated, such as the result of the `optimizer.step()` ca
ADHMR/lib/models/ema.py:32
↓ 14 callersFunctionload_img
(path, order='RGB')
HMR-Scorer/common/utils/preprocessing.py:12
↓ 14 callersFunctionmake_linear_layers
(feat_dims, relu_final=True, use_bn=False)
HMR-Scorer/common/nets/layer.py:3
↓ 14 callersMethodnorm1
nn.Module: the normalization layer named "norm1"
HMR-Scorer/main/transformer_utils/mmpose/models/backbones/hrt.py:432
↓ 13 callersFunctionflip_xyz_joints_3d
Flip 3d xyz joints. Parameters ---------- joints_3d : numpy.ndarray Joints in shape (num_joints, 3) joint_pairs : list
ADHMR/lib/utils/transforms.py:957
↓ 13 callersFunctionget_root_logger
Use `get_logger` method in mmcv to get the root logger. The logger will be initialized if it has not been initialized. By default a StreamHan
HMR-Scorer/main/transformer_utils/mmpose/utils/logger.py:7
↓ 13 callersFunctionload_checkpoint
Load checkpoint from a file or URI. Args: model (Module): Module to load checkpoint. filename (str): Accept local filepath, URL, `
HMR-Scorer/main/transformer_utils/mmpose/models/backbones/utils/utils.py:312
↓ 13 callersMethodnorm2
nn.Module: the normalization layer named "norm2"
HMR-Scorer/main/transformer_utils/mmpose/models/backbones/hrt.py:437
↓ 13 callersFunctionnormalize_pose_cuda
(pose, mean_and_std=None,res_w_h=None, which='zero_center',scale = None)
ADHMR/lib/utils/pose_utils.py:23
↓ 13 callersFunctionreconstruction_error
Do Procrustes alignment and compute reconstruction error.
ADHMR/lib/utils/pose_utils.py:255
↓ 12 callersFunctioncam2pixel
(cam_coord, f, c)
ADHMR/lib/utils/pose_utils.py:277
↓ 12 callersFunctionget_3rd_point
Return vector c that perpendicular to (a - b).
ADHMR/lib/utils/transforms.py:328
↓ 12 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
HMR-Scorer/main/transformer_utils/mmpose/models/backbones/resnet.py:275
↓ 12 callersMethoditems
(self)
ADHMR/lib/dataset/humandata_utils/smplx/smplx/utils.py:51
↓ 12 callersMethodload_cache
(self, annot_path_cache)
HMR-Scorer/data/PW3D_DPO.py:64
↓ 11 callersFunctionaddDPG
Add dpg for data augmentation, including random crop and random sample.
ADHMR/lib/utils/transforms.py:38
↓ 11 callersFunctionbbox_clip_xyxy
Clip bounding box with format (xmin, ymin, xmax, ymax) to specified boundary. All bounding boxes will be clipped to the new region `(0, 0, width,
ADHMR/lib/utils/bbox.py:175
↓ 11 callersFunctionbbox_xywh_to_xyxy
Convert bounding boxes from format (x, y, w, h) to (xmin, ymin, xmax, ymax) Parameters ---------- xywh : list, tuple or numpy.ndarray
ADHMR/lib/utils/bbox.py:107
↓ 10 callersFunctionget_intrinsic_metrix
(f, c, inv=False)
ADHMR/lib/utils/pose_utils.py:342
↓ 10 callersMethoditems
(self)
ADHMR/lib/utils/smplx/smplx/utils.py:51
↓ 10 callersFunctionlbs
Performs Linear Blend Skinning with the given shape and pose parameters Parameters ---------- betas : torch.tensor BxNB
ADHMR/lib/utils/smplx/smplx/lbs.py:152
↓ 10 callersFunctionlbs
Performs Linear Blend Skinning with the given shape and pose parameters Parameters ---------- betas : torch.tensor BxNB
ADHMR/lib/dataset/humandata_utils/smplx/smplx/lbs.py:152
↓ 10 callersFunctionvertices2joints
Calculates the 3D joint locations from the vertices Parameters ---------- J_regressor : torch.tensor JxV The regressor array tha
ADHMR/lib/models/layers/smpl/lbs.py:505
↓ 9 callersMethod__init__
SMPL model constructor Parameters ---------- model_path: str The path to the folder or to the fi
ADHMR/lib/utils/smplx/smplx/body_models.py:48
↓ 9 callersMethod__init__
SMPL model constructor Parameters ---------- model_path: str The path to the folder or to the fi
ADHMR/lib/dataset/humandata_utils/smplx/smplx/body_models.py:49
↓ 9 callersMethod__init__
SMPL model constructor Parameters ---------- model_path: str The path to the folder or to the fi
HMR-Scorer/common/utils/smplx/smplx/body_models_origin.py:49
↓ 9 callersMethod__init__
SMPL model constructor Parameters ---------- model_path: str The path to the folder or to the fi
HMR-Scorer/common/utils/smplx/smplx/body_models.py:49
↓ 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
HMR-Scorer/main/transformer_utils/mmpose/core/fp16/utils.py:8
↓ 9 callersFunctionprocess_human_model_output
(human_model_param, cam_param, do_flip, img_shape, img2bb_trans, rot, human_model_type, joint_img=None)
HMR-Scorer/common/utils/preprocessing.py:323
↓ 9 callersFunctionto_np
(array, dtype=np.float32)
ADHMR/lib/utils/smplx/smplx/utils.py:113
↓ 9 callersFunctionto_np
(array, dtype=np.float32)
ADHMR/lib/dataset/humandata_utils/smplx/smplx/utils.py:113
↓ 8 callersMethod__init__
(self, use_target_weight=False, loss_weight=1.)
HMR-Scorer/main/transformer_utils/mmpose/models/losses/regression_loss.py:325
↓ 8 callersMethodcopy_to
Copy current parameters into given collection of parameters. Args: parameters: Iterable of `torch.nn.Parameter`; the parameters to be
ADHMR/lib/models/ema.py:53
↓ 8 callersFunctionfind_dynamic_lmk_idx_and_bcoords
Compute the faces, barycentric coordinates for the dynamic landmarks To do so, we first compute the rotation of the neck around the y-axis
HMR-Scorer/common/utils/smplx/smplx/lbs.py:31
↓ 8 callersFunctionflip_twist
(twist_phi, twist_weight, twist_pairs)
ADHMR/lib/utils/transforms.py:1044
↓ 8 callersFunctionget_model
(config, is_train = True, resume = False, resume_path = None)
ADHMR/lib/utils/function.py:92
↓ 8 callersMethodlog_prob
(self, x)
HMR-Scorer/main/transformer_utils/mmpose/models/heads/rle_regression_head.py:156
↓ 8 callersMethodlogger_info
(self, info)
HMR-Scorer/common/base.py:374
↓ 8 callersFunctionmake_conv_layers
(feat_dims, kernel=3, stride=1, padding=1, bnrelu_final=True)
HMR-Scorer/common/nets/layer.py:16
↓ 8 callersMethodsave
(cls, save_path, data_list)
ADHMR/lib/utils/cache.py:19
↓ 8 callersMethodsave
(cls, save_path, data_list, data_strategy)
HMR-Scorer/data/PW3D_DPO.py:174
↓ 8 callersMethodsave_cache
(self, annot_path_cache, datalist)
HMR-Scorer/data/PW3D_DPO.py:56
↓ 8 callersFunctionto_np
(array, dtype=np.float32)
ADHMR/lib/models/layers/smpl/SMPL.py:26
↓ 8 callersFunctionto_tensor
(array, dtype=torch.float32)
ADHMR/lib/models/layers/smpl/SMPL.py:15
↓ 8 callersFunctiontransform_joint_to_other_db
(src_joint, src_name, dst_name)
HMR-Scorer/common/utils/transforms.py:109
↓ 8 callersMethodvalues
(self)
HMR-Scorer/common/utils/smplx/smplx/utils.py:47
↓ 8 callersFunctionvertices2landmarks
Calculates landmarks by barycentric interpolation Parameters ---------- vertices: torch.tensor BxVx3, dtype = torch.float32
HMR-Scorer/common/utils/smplx/smplx/lbs.py:109
↓ 8 callersMethodwarning
(self, msg)
HMR-Scorer/common/logger.py:42
↓ 7 callersMethod__init__
(self, *args, **kwargs)
HMR-Scorer/main/transformer_utils/mmpose/deprecated.py:176
↓ 7 callersMethod__init__
(self, in_channels, stem_channels, out_channels,
HMR-Scorer/main/transformer_utils/mmpose/models/backbones/litehrnet.py:225
↓ 7 callersMethod__init__
( self, in_channels=3, embed_dims=768, conv_type='Conv2d',
HMR-Scorer/main/transformer_utils/mmpose/models/utils/transformer.py:187
↓ 7 callersFunctionaugmentation
(img, bbox, data_split)
HMR-Scorer/common/utils/preprocessing.py:105
↓ 7 callersFunctionbuild_loss
Build loss.
HMR-Scorer/main/transformer_utils/mmpose/models/builder.py:33
↓ 7 callersFunctioncam2pixel
(cam_coord, f, c)
HMR-Scorer/common/utils/transforms.py:10
↓ 7 callersMethodnorm3
nn.Module: the normalization layer named "norm3"
HMR-Scorer/main/transformer_utils/mmpose/models/backbones/resnet.py:237
↓ 7 callersFunctionprocess_bbox
(bbox, aspect_ratio=None, scale=1.0)
ADHMR/lib/utils/bbox.py:80
↓ 7 callersFunctionrigid_align
(A, B)
ADHMR/lib/utils/transforms_humandata.py:75
↓ 7 callersFunctionrotate_xyz_jts
(xyz_jts, rot)
ADHMR/lib/utils/transforms.py:1077
↓ 7 callersMethodupdate
Update currently maintained parameters. Call this every time the parameters are updated, such as the result of the `optimizer.step()` ca
HMR-Scorer/common/nets/ema.py:32
↓ 6 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
HMR-Scorer/main/transformer_utils/mmpose/models/backbones/vit.py:32
↓ 6 callersMethod__init__
(self, in_channels, out_channels, num_steps=4,
HMR-Scorer/main/transformer_utils/mmpose/models/backbones/rsn.py:35
↓ 6 callersMethod__init__
(self, unit_channels=256, num_stages=4, num_units=4,
HMR-Scorer/main/transformer_utils/mmpose/models/backbones/mspn.py:421
↓ 6 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None
HMR-Scorer/main/transformer_utils/mmpose/models/utils/tcformer_utils.py:479
↓ 6 callersMethod__init__
(self, feat_dim=768)
HMR-Scorer/common/nets/smpler_x.py:86
↓ 6 callersMethod_calc_metric_per_class
(self, error, seq_idx_dict)
ADHMR/lib/dataset/hp3d.py:292
↓ 6 callersFunctionbatch_rodrigues_numpy
Calculates the rotation matrices for a batch of rotation vectors Parameters ---------- rot_vecs: numpy.ndarray Nx3
ADHMR/lib/utils/transforms.py:1124
↓ 6 callersFunctionblend_shapes
Calculates the per vertex displacement due to the blend shapes Parameters ---------- betas : torch.tensor Bx(num_betas) Blend s
ADHMR/lib/models/layers/smpl/lbs.py:525
next →1–100 of 2,101, ranked by callers