Functions — github.com/JubSteven/POEM-v2

hub / github.com/JubSteven/POEM-v2 / functions

Functions1,165 in github.com/JubSteven/POEM-v2

⨍ Functions 1,165◇ Types & classes 149

↓ 153 callersMethodget

Get the registry record. Args: key (str): The class name in string format. Returns: class: The corresponding

lib/utils/builder.py:178

↓ 97 callersMethodinfo

(self, msg, **kwargs)

lib/utils/logger.py:93

↓ 58 callersMethodformat

(self, record)

lib/utils/logger.py:19

↓ 54 callersMethodwarning

(self, msg, **kwargs)

lib/utils/logger.py:97

↓ 50 callersMethodreset

(self)

lib/metrics/pa_eval.py:31

↓ 41 callersMethodfeed

(self, pred_joints_3d_abs, joints_3d_abs, pred_verts_3d_abs=None, verts_3d_abs=None, **kwargs)

lib/metrics/pa_eval.py:45

↓ 26 callersFunctionpersp_project

Apply perspective camera projection on a 3D point Args: points3d (np.ndarray): shape (N, 3) cam_intr (np.ndarray): shape (3, 3)

lib/utils/transform.py:949

↓ 22 callersMethodadd_image

(self, tag, img_tensor, global_step=None, walltime=None, dataformats='CHW')

lib/utils/summary_writer.py:21

↓ 22 callersMethodadd_scalar

(self, tag, value, global_step=None, walltime=None)

lib/utils/summary_writer.py:13

↓ 21 callersMethoderror

(self, msg, **kwargs)

lib/utils/logger.py:101

↓ 19 callersMethodflip_2d

(annot_2d, centerX)

lib/datasets/hdata.py:61

↓ 18 callersMethoddump

(self, *args, **kwargs)

lib/utils/config.py:28

↓ 17 callersFunctionparam_size

(net)

lib/utils/misc.py:143

↓ 17 callersMethodupdate

(self, val: float, n: int = 1)

lib/metrics/basic_metric.py:46

↓ 15 callersMethodrecord_metric

(self, metrics: List, epoch: int, comment="")

lib/utils/recorder.py:148

↓ 14 callersFunctionbatch_cam_extr_transf

apply batch camera extrinsic transformation on batch joints Args: batch_cam_extr (torch.Tensor): shape (BATCH, NPERSP, 4, 4) batc

lib/utils/transform.py:898

↓ 13 callersFunctionetqdm

(iterable, rank: Optional[int] = None, **kwargs)

lib/utils/etqdm.py:11

↓ 13 callersMethodget_loss

(self, loss_name: str)

lib/metrics/basic_metric.py:96

↓ 12 callersFunctionbatch_cam_intr_projection

apply camera projection on batch joints with batch intrinsics Args: batch_cam_intr (torch.Tensor): shape (BATCH, NPERSP, 3, 3) ba

lib/utils/transform.py:914

↓ 12 callersFunctionbatch_persp_project

Batch apply perspective procjection on points Args: verts (torch.Tensor): 3D points with shape (B, N, 3) camintr (torch.Tensor):

lib/utils/transform.py:933

↓ 12 callersFunctionget_annot_center

(annots, visibility=None)

lib/utils/transform.py:1011

↓ 12 callersFunctionget_annot_scale

Retreives the size of the square we want to crop by taking the maximum of vertical and horizontal span of the hand and multiplying it by

lib/utils/transform.py:994

↓ 12 callersMethodget_result

(self)

lib/metrics/pa_eval.py:93

↓ 11 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

lib/utils/builder.py:9

↓ 10 callersFunctiondraw_batch_joint_images

(joints2d, gt_jointd2d, tensor_image, step_idx, n_sample=16)

lib/viztools/draw.py:84

↓ 10 callersMethodflip_3d

(annot_3d)

lib/datasets/hdata.py:67

↓ 10 callersMethodload_pretrained

(self, state_dict)

lib/models/backbones/resnet.py:201

↓ 10 callersFunctionmano_to_openpose

Convert MANO vertices to OpenPose joints NOTE: the MANO Joints' order (with tips) 16-15-14-13-\

lib/utils/transform.py:836

↓ 9 callersMethod_load_pickle

(self, idx)

lib/datasets/oakink2_dev.py:93

↓ 9 callersMethodget_measures

(self, **kwargs)

lib/metrics/pa_eval.py:84

↓ 9 callersMethodload_sample

(self, idx)

lib/datasets/arctic.py:85

↓ 8 callersMethod__init__

(self, version, block, layers, num_classes=1000, freeze_batchnorm=True, early_return=4)

lib/models/backbones/resnet.py:149

↓ 8 callersMethoddownsample

Downsample mesh.

lib/external/metro/base_model.py:360

↓ 8 callersFunctionget_config

Read a config file and optionally merge it with the default config file. Args: config_file (str): Path to config file. merge (boo

lib/utils/config.py:75

↓ 8 callersMethodload_sample

(self, idx)

lib/datasets/interhand.py:89

↓ 8 callersFunctionrotmat_to_aa

Convert rotation matrixs to axis angles. Args: matrix (Union[torch.Tensor, numpy.ndarray]): input shape should be (..., 3

lib/utils/transform.py:393

↓ 7 callersFunctionaa_to_rotmat

Convert axis_angle to rotation matrixs. Args: axis_angle (Union[torch.Tensor, numpy.ndarray]): input shape should be

lib/utils/transform.py:377

↓ 7 callersFunctiondenormalize

(tensor, mean, std, inplace=False)

lib/utils/transform.py:1104

↓ 7 callersMethodget_joints_2d_vis

(self, joints_2d=None, raw_size=None, **kwargs)

lib/datasets/hdata.py:145

↓ 7 callersMethodrecord_loss

(self, loss_metric: LossMetric, epoch: int, comment="")

lib/utils/recorder.py:141

↓ 6 callersFunctiondraw_wireframe

( img, vert_list, edge_list, vert_color, edge_color, vert_size=3, edge_size=1,

lib/utils/vis_cv2_util.py:51

↓ 6 callersMethodget_cam_intr

(self, idx)

lib/datasets/freihand.py:316

↓ 6 callersFunctioninverse_sigmoid

Inverse function of sigmoid. Args: x (Tensor): The tensor to do the inverse. eps (float): EPS avoid numerical

lib/utils/transform.py:1145

↓ 6 callersFunctionkey_filter

(sample)

lib/utils/collation.py:42

↓ 6 callersFunctiontensor2array

(tensor, max_value=None, colormap='jet', channel_first=True, mean=0.5, std=0.5)

lib/external/cmr/utils.py:407

↓ 5 callersMethod__init__

Composes several transforms together. This transform does not support torchscript. Args: transforms (list): (list of tra

lib/utils/transform.py:347

↓ 5 callersMethod__init__

(self, cfg)

lib/models/backbones/hrnet.py:446

↓ 5 callersMethod_make_residual

(self, block, nblocks, in_planes)

lib/models/backbones/hourglass.py:81

↓ 5 callersFunctioncreate_dataset

Create a dataset instance.

lib/datasets/__init__.py:14

↓ 5 callersMethodget_cam_intr

(self, idx)

lib/datasets/oakink.py:136

↓ 5 callersFunctionindex_points

Input: points: input points data, [B, N, C] idx: sample index data, [B, S, [K]] Return: new_points:, indexed points d

lib/utils/points_utils.py:9

↓ 5 callersFunctioninv_transf_np

(transf: np.ndarray)

transform/transform_np.py:23

↓ 5 callersMethodon_val_finished

(self, recorder, epoch_idx)

lib/external/cmr/model.py:398

↓ 5 callersFunctionplot_hand

Plots a hand stick figure into a matplotlib figure.

lib/viztools/draw.py:134

↓ 5 callersMethodsetup

(self, summary_writer, **kwargs)

lib/external/cmr/model.py:275

↓ 5 callersFunctionsplit_mano_param

(mano_param)

lib/datasets/freihand.py:21

↓ 5 callersMethodstep

(self)

lib/viztools/viz_o3d_utils.py:190

↓ 4 callersFunctionSE3_transform

Apply SE3 transform on a 3D point Args: points3d (np.ndarray): shape (N, 3) transform (np.ndarray): shape (4, 4) Returns:

lib/utils/transform.py:964

↓ 4 callersMethod__call__

(self, rotation: Union[torch.Tensor, np.ndarray], convention: str = 'xyz', **kwargs)

lib/utils/transform.py:356

↓ 4 callersMethod__init__

(self, cfg)

lib/models/layers/petr_transformer.py:44

↓ 4 callersMethod__init__

(self, in_channel, out_chanel, kernel_size=3, padding=1, stride=1, dilation=1, relu=False, norm='bn')

lib/external/cmr/net.py:94

↓ 4 callersMethod_build_head_module

(self)

lib/models/heads/ptEmb_head.py:86

↓ 4 callersFunction_construct_rotation_matrix

Construct the in-plane rotation matrix. Args: rot (float): Rotation rad. size (int): The size of the rotation matrix.

lib/utils/transform.py:618

↓ 4 callersFunction_get_affine_trans_no_rot

(center, scale, res)

lib/utils/transform.py:697

↓ 4 callersMethod_make_layer

(self, block, planes, blocks, stride=1)

lib/models/backbones/resnet.py:185

↓ 4 callersMethod_str_custom_args

(self)

lib/models/heads/ptEmb_head.py:83

↓ 4 callersMethodbackward

(ctx, grad_output)

lib/external/metro/base_model.py:281

↓ 4 callersFunctionbatch_ref_bone_len

(joint: Union[np.ndarray, torch.Tensor], ref_bone_link=None)

lib/utils/transform.py:875

↓ 4 callersFunctionbchw_2_bhwc

:param x: torch tensor, B x C x H x W :return: torch tensor, B x H x W x C

lib/utils/transform.py:1135

↓ 4 callersFunctionbuild_backbone

(cfg, **kwargs)

lib/models/backbones/__init__.py:9

↓ 4 callersFunctioncaculate_align_mat

(vec)

lib/utils/transform.py:1192

↓ 4 callersFunctiondraw_batch_verts_images

(verts2d, gt_verts2d, tensor_image, step_idx, n_sample=16)

lib/viztools/draw.py:49

↓ 4 callersFunctiongauss

(x, a, b, c, d=0)

lib/utils/heatmap.py:171

↓ 4 callersMethodget_bbox_center_scale

(self, idx)

lib/datasets/hdata.py:125

↓ 4 callersMethodget_cam_intr

(self, idx)

lib/datasets/ho3d.py:245

↓ 4 callersMethodget_dist

(self, x, y)

lib/metrics/pa_eval.py:40

↓ 4 callersFunctionget_heatmap_pred

get predictions from heatmaps in torch Tensor return type: torch.LongTensor

lib/utils/heatmap.py:97

↓ 4 callersMethodget_image

(self, idx)

lib/datasets/hdata.py:77

↓ 4 callersMethodget_image_mask

(self, idx)

lib/datasets/hdata.py:81

↓ 4 callersMethodget_image_path

(self, idx)

lib/datasets/hdata.py:85

↓ 4 callersMethodget_joints_2d

(self, idx)

lib/datasets/hdata.py:105

↓ 4 callersMethodget_joints_3d

(self, idx)

lib/datasets/interhand.py:109

↓ 4 callersMethodget_joints_3d

(self, idx)

lib/datasets/arctic.py:107

↓ 4 callersMethodget_label

(self, label_file: str)

lib/datasets/dexycb.py:127

↓ 4 callersMethodget_sides

(self, idx)

lib/datasets/hdata.py:121

↓ 4 callersMethodinit_weights

(self)

lib/models/backbones/resnet.py:177

↓ 4 callersFunctionload_weights

(module: nn.Module, pretrained=None, strict=True)

lib/utils/net_utils.py:200

↓ 4 callersFunctionparse_exp_args

()

lib/opt.py:53

↓ 4 callersFunctionquat_to_rotmat_np

Convert rotations given as quaternions to rotation matrices. Args: quat: quaternions in form (w,x,y,z), ndarray of shap

transform/rotation_np.py:130

↓ 4 callersFunctionrotmat_to_quat_np

Convert rotations given as rotation matrices to quaternions. Args: rotmat: tensor of shape (..., 3, 3). Returns: quaterni

transform/rotation_np.py:159

↓ 4 callersFunctionspmm

(sparse, dense)

lib/external/metro/base_model.py:289

↓ 3 callersMethod__init__

(self, cfg)

lib/models/layers/ptEmb_transformer.py:126

↓ 3 callersMethod__init__

(self, cfg: CN)

lib/models/bricks/transformer.py:197

↓ 3 callersMethod__init__

(self, cfg: CN)

lib/models/heads/ptEmb_head.py:57

↓ 3 callersMethod__init__

(self, config)

lib/external/metro/base_model.py:143

↓ 3 callersMethod__init__

(self, inplanes, planes, stride=1, downsample=None)

lib/external/metro/hrnet/hrnet.py:34

↓ 3 callersFunction_affine_transform_post_rot

(center, scale, optical_center, out_res, rot=0)

lib/utils/transform.py:684

↓ 3 callersMethod_forward_impl

(self, inputs)

lib/models/integal_pose.py:168

↓ 3 callersMethod_forward_impl

(self, batch, **kwargs)

lib/models/POEM.py:255

↓ 3 callersMethod_forward_impl

(self, data)

lib/external/cmr/model.py:278

next →1–100 of 1,165, ranked by callers