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

↓ 3 callersMethod_get_info_list

(data_dir, use_split_mv, split_key, data_split)

lib/datasets/oakink.py:28

↓ 3 callersMethod_ho3d_get_hand_info

Get the hand annotation in the raw ho3d datasets. !!! This Mehthods shoudln't be called outside. :param idx: :return:

lib/datasets/ho3d.py:384

↓ 3 callersMethod_make_stage

(self, layer_config, num_inchannels, multi_scale_output=True)

lib/models/backbones/hrnet.py:362

↓ 3 callersMethod_make_stage

(self, layer_config, num_inchannels, multi_scale_output=True)

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

↓ 3 callersMethod_make_transition_layer

(self, num_channels_pre_layer, num_channels_cur_layer)

lib/models/backbones/hrnet.py:319

↓ 3 callersMethod_make_transition_layer

(self, num_channels_pre_layer, num_channels_cur_layer)

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

↓ 3 callersFunction_out_of_frame

(pos, shape)

lib/utils/vis_cv2_util.py:43

↓ 3 callersMethod_register_module

(self, module_class, module_name=None, force=False)

lib/utils/builder.py:227

↓ 3 callersFunction_rotateY

Rotate the points by a specified angle.

lib/viztools/opendr_renderer.py:188

↓ 3 callersFunction_transform_coords

Args: pts(np.ndarray): (point_nb, 2)

lib/utils/transform.py:637

↓ 3 callersMethodadd_geometry

(self, geo)

lib/viztools/viz_o3d_utils.py:176

↓ 3 callersMethodadd_images

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

lib/utils/summary_writer.py:25

↓ 3 callersFunctionassemble_T_np

(tsl, rotmat)

transform/transform_np.py:10

↓ 3 callersMethodbatch_multi2single

(self, batch)

lib/models/darkpose.py:182

↓ 3 callersFunctionbce_loss

(pred, gt)

lib/external/cmr/loss.py:12

↓ 3 callersMethodboard_img

(self, phase, n_iter, img, **kwargs)

lib/models/darkpose.py:223

↓ 3 callersMethoddecoder

(self, x)

lib/external/cmr/model.py:233

↓ 3 callersFunctiondraw_2d_skeleton

:param image: H x W x 3 :param joints_uv: 21 x 2 wrist, thumb_mcp, thumb_pip, thumb_dip, thumb_tip index_mcp, index_pip, index_di

lib/viztools/draw.py:234

↓ 3 callersFunctiondraw_3d_skeleton

:param joints_xyz: 21 x 3 :param image_size: H, W :return:

lib/viztools/draw.py:349

↓ 3 callersFunctionformat_args_cfg

(args, cfg={})

lib/utils/misc.py:132

↓ 3 callersMethodget_cam_intr

(self, idx)

lib/datasets/dexycb.py:154

↓ 3 callersMethodget_cam_intr

(self, idx)

lib/datasets/interhand.py:153

↓ 3 callersMethodget_cam_intr

(self, idx)

lib/datasets/arctic.py:151

↓ 3 callersMethodget_image_path

(self, idx)

lib/datasets/oakink.py:114

↓ 3 callersMethodget_joints_2d

(self, idx)

lib/datasets/oakink.py:161

↓ 3 callersMethodget_joints_3d

(self, idx)

lib/datasets/freihand.py:329

↓ 3 callersMethodget_joints_3d

(self, idx)

lib/datasets/ho3d.py:214

↓ 3 callersMethodget_joints_3d

(self, idx)

lib/datasets/oakink.py:149

↓ 3 callersFunctionget_vert_connectivity

Returns a sparse matrix (of size #verts x #verts) where each nonzero element indicates a neighborhood relation. For example, if there is a non

lib/external/cmr/utils.py:63

↓ 3 callersMethodget_verts_3d

(self, idx)

lib/datasets/freihand.py:338

↓ 3 callersMethodget_verts_3d

(self, idx)

lib/datasets/ho3d.py:317

↓ 3 callersMethodget_verts_3d

(self, idx)

lib/datasets/oakink.py:155

↓ 3 callersMethodinference_step

(self, batch, batch_idx)

lib/models/model_abc.py:48

↓ 3 callersMethodinit_weights

(self, pretrained=None)

lib/external/metro/metro.py:381

↓ 3 callersFunctionnormalize

(v)

lib/datasets/oakink2_dev.py:499

↓ 3 callersFunctionnormalize

(v)

tool/flip_util.py:6

↓ 3 callersMethodposition_embeding

(self, img_feat, img_metas, masks=None)

lib/models/heads/ptEmb_head.py:114

↓ 3 callersFunctionquat_raw_multiply

Multiply two quaternions. Usual torch rules for broadcasting apply. Args: a: Quaternions as tensor of shape (..., 4), real part f

transform/rotation_jit.py:59

↓ 3 callersFunctionquat_raw_multiply_np

Multiply two quaternions. Usual torch rules for broadcasting apply. Args: a: Quaternions as tensor of shape (..., 4), real part f

transform/rotation_np.py:44

↓ 3 callersMethodrun

(self)

lib/viztools/viz_o3d_utils.py:194

↓ 3 callersFunctionsetup_seed

Setup all the random seeds Args: seed (int or float): seed value conv_repeatable (bool, optional): Whether the conv ops are repea

lib/utils/net_utils.py:135

↓ 3 callersFunctionspiral_tramsform

( transform_fp, template_fp, ds_factors=[2, 2, 2, 2], seq_length=[27, 27, 27, 27], dilatio

lib/external/cmr/utils.py:16

↓ 3 callersFunctionstandardize_quat

Convert a unit quaternion to a standard form: one in which the real part is non negative. Args: quat: quaternions in form (w,x,y,

transform/rotation_jit.py:26

↓ 3 callersFunctionstandardize_quat_np

Convert a unit quaternion to a standard form: one in which the real part is non negative. Args: quat: quaternions in form (w,x,y,

transform/rotation_np.py:21

↓ 3 callersFunctiontensor2array

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

lib/models/darkpose.py:19

↓ 3 callersFunctiontransf_point_array_np

(transf: np.ndarray, point: np.ndarray)

transform/transform_np.py:36

↓ 3 callersMethodupdate_by_mean

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

lib/metrics/basic_metric.py:51

↓ 3 callersMethodupdate_geometry

(self, geo)

lib/viztools/viz_o3d_utils.py:187

↓ 3 callersFunctionxavier_init

(module, gain=1, bias=0, distribution='normal')

lib/utils/net_utils.py:164

↓ 2 callersFunctionPool

:param x: input feature :param trans: upsample matrix :param dim: upsample dimension :return: upsampled feature

lib/external/cmr/net.py:6

↓ 2 callersMethod__init__

(self)

lib/metrics/basic_metric.py:9

↓ 2 callersMethod__init__

(self, **cfg)

lib/metrics/pck.py:179

↓ 2 callersMethod__init__

(self, cfg)

lib/datasets/dexycb.py:31

↓ 2 callersMethod__init__

(self, cfg)

lib/datasets/freihand.py:156

↓ 2 callersMethod__init__

(self, cfg)

lib/datasets/oakink.py:46

↓ 2 callersMethod__init__

Projective Attention Module :param d_model hidden dimension :param n_levels number of feature levels :param

lib/models/layers/mvp_decoder.py:99

↓ 2 callersMethod__init__

(self, in_channel, out_channel)

lib/models/bricks/point_transformers.py:37

↓ 2 callersMethod__init__

(self, config)

lib/models/bricks/metro_transformer.py:159

↓ 2 callersMethod__init__

(self, config)

lib/models/bricks/pt_metro_transformer.py:15

↓ 2 callersMethod__init__

(self, cfg)

lib/external/cmr/model.py:86

↓ 2 callersFunction_affine_transform

(center, scale, out_res, rot=0)

lib/utils/transform.py:674

↓ 2 callersFunction_angle_from_tan

Extract the first or third Euler angle from the two members of the matrix which are positive constant times its sine and cosine. Args:

transform/rotation_jit.py:374

↓ 2 callersFunction_angle_from_tan

Extract the first or third Euler angle from the two members of the matrix which are positive constant times its sine and cosine. Args:

transform/rotation_np.py:371

↓ 2 callersMethod_forward_impl

(self, batch, **kwargs)

lib/models/PETR.py:106

↓ 2 callersMethod_forward_impl

(self, batch, **kwargs)

lib/models/darkpose.py:167

↓ 2 callersMethod_forward_impl

(self, batch)

lib/external/metro/metro.py:265

↓ 2 callersMethod_get_pck

Returns pck for one keypoint for the given threshold.

lib/metrics/pck.py:84

↓ 2 callersFunction_index_from_letter

(letter: str)

transform/rotation_jit.py:404

↓ 2 callersFunction_index_from_letter

(letter: str)

transform/rotation_np.py:400

↓ 2 callersMethod_load_annots

(self, seq_frames=[], subfolder="train", **kwargs)

lib/datasets/ho3d.py:144

↓ 2 callersMethod_make_layer

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

lib/models/backbones/hrnet.py:346

↓ 2 callersMethod_make_layer

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

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

↓ 2 callersMethod_mapping_multiview

(self, seq_frames)

lib/datasets/ho3d.py:639

↓ 2 callersFunction_next_ring

(mesh, last_ring, other)

lib/external/cmr/utils.py:338

↓ 2 callersFunction_normalize

(mat: np.ndarray, axis: int, eps: Optional[float] = None)

transform/rotation_np.py:471

↓ 2 callersMethod_reset_parameters

(self)

lib/models/layers/mvp_decoder.py:146

↓ 2 callersMethodalign_w_scale

Align the predicted entity in some optimality sense with the ground truth.

lib/utils/testing.py:163

↓ 2 callersMethodalign_w_scale

Align the predicted entity in some optimality sense with the ground truth.

lib/metrics/pa_eval.py:104

↓ 2 callersFunctionanchor_points

(xyz, K, device)

lib/models/bricks/point_transformers.py:10

↓ 2 callersFunctionbatch_triangulate_dlt_torch

torch: Triangulate multiple 2D points from multiple sets of multiviews using the DLT algorithm. NOTE: Expend to Batch and nJoints dimension.

lib/utils/triangulation.py:5

↓ 2 callersFunctionbatch_uvd2xyz

(uvd: torch.Tensor, root_joint: torch.Tensor, intr: torch.Tensor,

lib/utils/transform.py:788

↓ 2 callersFunctionbuild_dataset

(cfg: CN, **kwargs)

lib/utils/builder.py:323

↓ 2 callersFunctionbuild_head

(cfg, **kwargs)

lib/models/heads/__init__.py:4

↓ 2 callersFunctionbuild_optimizer

(params: Iterable, cfg)

lib/utils/net_utils.py:57

↓ 2 callersFunctionbuild_scheduler

(optimizer: Optimizer, cfg, steps=None)

lib/utils/net_utils.py:82

↓ 2 callersFunctionbuild_transform

(cfg: CN, **kwargs)

lib/utils/builder.py:335

↓ 2 callersFunctionbuild_transformer

Builder for transformer encoder and decoder.

lib/models/bricks/transformer.py:19

↓ 2 callersFunctionbuild_transformer_layer_sequence

Builder for transformer encoder and transformer decoder.

lib/models/bricks/transformer.py:29

↓ 2 callersMethodcalculate_fscore

(gt, pr, th=0.01)

lib/utils/testing.py:122

↓ 2 callersMethodcam_P_transf

(self, P, x)

lib/models/heads/petr_FTL_head.py:45

↓ 2 callersFunctionclip_gradient

Clips gradients computed during backpropagation to avoid explosion of gradients. Args: optimizer (torch.optim.optimizer): optimizer with

lib/utils/net_utils.py:122

↓ 2 callersFunctioncol

(A)

lib/external/cmr/utils.py:59

↓ 2 callersFunctioncollapse_cost

(Qv, r, c, v)

lib/external/cmr/utils.py:202

↓ 2 callersFunctionconv3x3

3x3 convolution with padding

lib/models/backbones/hrnet.py:33

↓ 2 callersFunctionconv3x3

3x3 convolution with padding

lib/models/backbones/resnet.py:21

↓ 2 callersFunctionconv3x3

3x3 convolution with padding

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

↓ 2 callersMethodconvert

Convert the data to the format that the CMR can accept.

lib/external/cmr/data_adaptor.py:79

↓ 2 callersFunctiondraw_3d_mesh_mayavi

(image_size, hand_xyz=None, hand_face=None, obj_xyz=None, obj_face=None, ratio=400 / 224)

lib/viztools/draw.py:436

↓ 2 callersFunctiondraw_batch_mesh_images

(verts3d, gt_verts3d, face, intr, tensor_image, step_idx, n_sample=16)

lib/viztools/draw.py:17

↓ 2 callersFunctiondraw_mesh

:param image: H x W x 3 :param cam_param: 1 x 3 x 3 :param mesh_xyz: 778 x 3 :param face: 1538 x 3 x 2 :return:

lib/viztools/draw.py:206

← previous next →101–200 of 1,165, ranked by callers