MCPcopy Create free account

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

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

⨍ Functions 1,165◇ Types & classes 149
↓ 1 callersMethod_forward
Forward function for `TransformerCoder`. Returns: Tensor: forwarded results with shape [num_query, bs, embed_dims].
lib/models/layers/petr_transformer.py:207
↓ 1 callersMethod_forward
(self, xyz, features)
lib/models/bricks/point_transformers.py:70
↓ 1 callersMethod_forward
(self, xyz, features, query)
lib/models/bricks/point_transformers.py:125
↓ 1 callersMethod_generate_random_basis
Sample uniformly from d-dimensional unit ball Adapted from https://github.com/sergeyprokudin/bps/blob/master/bps/bps.py
lib/models/heads/ptEmb_head.py:773
↓ 1 callersFunction_get_activation_fn
Return an activation function given a string
lib/models/layers/mvp_decoder.py:49
↓ 1 callersFunction_get_clones
(module, N)
lib/models/layers/mvp_decoder.py:45
↓ 1 callersMethod_get_deconv_cfg
(self, deconv_kernel, index)
lib/models/darkpose.py:107
↓ 1 callersMethod_get_deconv_cfg
(self, deconv_kernel, index)
lib/models/integal_pose.py:318
↓ 1 callersMethod_get_epe
Returns end point error for one keypoint.
lib/metrics/pck.py:75
↓ 1 callersMethod_get_epe
Returns end point error for one keypoint.
lib/datasets/ho3d_official_test.py:359
↓ 1 callersMethod_get_pck
Returns pck for one keypoint for the given threshold.
lib/datasets/ho3d_official_test.py:350
↓ 1 callersMethod_get_predictions
(self, preds: Dict, targs: Dict)
lib/metrics/pck.py:27
↓ 1 callersMethod_get_ray_for_mvp
(self, img_metas, batch, all_feats)
lib/models/heads/mvp_head.py:205
↓ 1 callersFunction_get_sparse_transform
(faces, num_original_verts)
lib/external/cmr/utils.py:289
↓ 1 callersMethod_get_views
(self, all_feats)
lib/models/heads/mvp_head.py:181
↓ 1 callersMethod_hourglass_foward
(self, n, x)
lib/models/backbones/hourglass.py:87
↓ 1 callersFunction_is_power_of_2
(n)
lib/models/layers/mvp_decoder.py:60
↓ 1 callersMethod_load_seq_frames
trainval_idx (int): How many frames to include in training split when using trainval/val/test split
lib/datasets/ho3d.py:123
↓ 1 callersMethod_load_seq_frames_multiview
trainval_idx (int): How many frames to include in training split when using trainval/val/test split
lib/datasets/ho3d.py:674
↓ 1 callersMethod_load_video_frames
(self)
lib/datasets/dexycb.py:539
↓ 1 callersMethod_load_video_frames
(self)
lib/datasets/oakink.py:652
↓ 1 callersMethod_make_branches
(self, num_branches, block, num_blocks, num_channels)
lib/models/backbones/hrnet.py:169
↓ 1 callersMethod_make_branches
(self, num_branches, block, num_blocks, num_channels)
lib/external/metro/hrnet/hrnet.py:162
↓ 1 callersMethod_make_deconv_layer
(self, num_layers, num_filters, num_kernels)
lib/models/darkpose.py:120
↓ 1 callersMethod_make_deconv_layer
(self, num_layers, num_filters, num_kernels)
lib/models/integal_pose.py:333
↓ 1 callersMethod_make_fuse_layers
(self)
lib/models/backbones/hrnet.py:177
↓ 1 callersMethod_make_fuse_layers
(self)
lib/external/metro/hrnet/hrnet.py:170
↓ 1 callersMethod_make_head
(self, pre_stage_channels)
lib/models/backbones/hrnet.py:285
↓ 1 callersMethod_make_head
(self, pre_stage_channels)
lib/external/metro/hrnet/hrnet.py:278
↓ 1 callersMethod_make_hourglass
(self, block, nblocks, in_planes, depth)
lib/models/backbones/hourglass.py:58
↓ 1 callersMethod_make_one_branch
(self, branch_index, block, num_blocks, num_channels, stride=1)
lib/models/backbones/hrnet.py:147
↓ 1 callersMethod_make_one_branch
(self, branch_index, block, num_blocks, num_channels, stride=1)
lib/external/metro/hrnet/hrnet.py:140
↓ 1 callersMethod_mapping_idxs_infos
(self, source_set_name, subfolder)
lib/datasets/ho3d.py:650
↓ 1 callersMethod_mapping_idxs_infos
(self)
lib/datasets/ho3d_official_test.py:155
↓ 1 callersMethod_mapping_multiview
(self, seq_frames)
lib/datasets/ho3d_official_test.py:140
↓ 1 callersFunction_one_hot
(idx: np.ndarray, num_classes: int)
transform/rotation_np.py:125
↓ 1 callersMethod_preload
(self)
lib/datasets/dexycb.py:65
↓ 1 callersMethod_preload
(self)
lib/datasets/interhand.py:62
↓ 1 callersMethod_preload
(self)
lib/datasets/interhand.py:333
↓ 1 callersMethod_preload
(self)
lib/datasets/freihand.py:175
↓ 1 callersMethod_preload
(self)
lib/datasets/ho3d.py:68
↓ 1 callersMethod_preload
(self)
lib/datasets/yt3d.py:37
↓ 1 callersMethod_preload
(self)
lib/datasets/arctic.py:65
↓ 1 callersMethod_preload
(self)
lib/datasets/arctic.py:292
↓ 1 callersMethod_preload_extra
(self)
lib/datasets/freihand.py:458
↓ 1 callersMethod_record_init_info
(self: T)
lib/utils/recorder.py:50
↓ 1 callersFunction_rotate_smpl_pose
Rotate SMPL pose parameters. SMPL (https://smpl.is.tue.mpg.de/) is a 3D human model. Args: pose (np.ndarray([72])): SMPL pose paramet
lib/utils/transform.py:596
↓ 1 callersMethod_single_view_arctic
(self)
lib/datasets/arctic.py:305
↓ 1 callersMethod_single_view_dexycb
(self)
lib/datasets/dexycb.py:361
↓ 1 callersMethod_single_view_ho3d
(self)
lib/datasets/ho3d.py:617
↓ 1 callersMethod_single_view_ho3d
(self)
lib/datasets/ho3d_official_test.py:121
↓ 1 callersMethod_single_view_interhand
(self)
lib/datasets/interhand.py:310
↓ 1 callersFunction_sqrt_positive_part
Returns torch.sqrt(torch.max(0, x)) but with a zero subgradient where x is 0.
transform/rotation_jit.py:163
↓ 1 callersFunction_sqrt_positive_part
Returns torch.sqrt(torch.max(0, x)) but with a zero subgradient where x is 0.
transform/rotation_np.py:114
↓ 1 callersFunctionaccuracy_heatmap
Calculate accuracy according to PCK, but uses ground truth heatmap rather than x,y locations First to be returned is average accuracy across
lib/utils/heatmap.py:32
↓ 1 callersFunctionadjmat_sparse
Create row-normalized sparse graph adjacency matrix.
lib/external/metro/base_model.py:313
↓ 1 callersMethodalign_sc_tr
Align the 3D joint location with the ground truth by scaling and translation
lib/utils/testing.py:141
↓ 1 callersFunctionaxis_equal_3d
(ax, ratio=1.2)
lib/viztools/draw.py:339
↓ 1 callersFunctionbackproject_ortho
Calculate 3D coordinates from 2D coordinates and the camera parameters.
lib/datasets/freihand.py:36
↓ 1 callersFunctionbbox_get_center_scale
(bbox, expand=2.0, mindim=200)
tool/infer_hand.py:50
↓ 1 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
lib/utils/transform.py:1021
↓ 1 callersMethodboard_img
(self, phase, n_iter, img, **kwargs)
lib/external/cmr/model.py:339
↓ 1 callersFunctionbuild_attention
Builder for attention.
lib/models/bricks/transformer.py:34
↓ 1 callersFunctionbuild_feedforward_network
Builder for feed-forward network (FFN).
lib/models/bricks/transformer.py:39
↓ 1 callersFunctionbuild_transformer_layer
Builder for transformer layer.
lib/models/bricks/transformer.py:24
↓ 1 callersFunctioncalc_dists
(preds, target, normalize, mask)
lib/utils/heatmap.py:7
↓ 1 callersFunctioncenter_scale_to_box
Convert bbox center scale to bbox xyxy Args: center (np.array): center of the bbox (x, y) scale (np.float_): side length of the b
lib/utils/transform.py:1083
↓ 1 callersFunctioncolor_heatmap
(x)
lib/utils/heatmap.py:161
↓ 1 callersMethodcompute_interval
(fid, size, nframe)
video_tool/ffmpeg_util.py:92
↓ 1 callersMethodcompute_loss
(self, preds, gt)
lib/models/PETR.py:156
↓ 1 callersMethodcompute_loss
(self, preds, gt)
lib/models/darkpose.py:175
↓ 1 callersMethodcompute_loss
(self, preds, gt)
lib/models/integal_pose.py:67
↓ 1 callersMethodcompute_loss
(self, preds, gt)
lib/models/POEM.py:378
↓ 1 callersMethodcompute_loss
(self, **kwargs)
lib/external/cmr/model.py:355
↓ 1 callersMethodcompute_loss
(self, preds, gt)
lib/external/metro/metro.py:311
↓ 1 callersFunctioncreate_backbone
(cfg: CN)
lib/models/backbones/__init__.py:13
↓ 1 callersMethoddebug
(self, msg, **kwargs)
lib/utils/logger.py:105
↓ 1 callersFunctiondefault_config
Get a yacs CfgNode object with the default config values.
lib/utils/config.py:66
↓ 1 callersMethoddeinit
(self)
lib/viztools/viz_o3d_utils.py:60
↓ 1 callersMethoddeprecated_register_module
(self, cls=None, force=False)
lib/utils/builder.py:242
↓ 1 callersFunctiondist_acc
Return percentage below threshold while ignoring values with a -1
lib/utils/heatmap.py:23
↓ 1 callersFunctiondistance
(x, y)
lib/external/cmr/regitstration.py:14
↓ 1 callersFunctionedge_length_loss
(pred, gt, face, is_valid=None)
lib/external/cmr/loss.py:43
↓ 1 callersFunctioneuler_angle_to_rotmat
Convert rotations given as Euler angles in radians to rotation matrices. Args: euler_angles: Euler angles in radians as tensor of sha
transform/rotation_jit.py:335
↓ 1 callersFunctionexpand
(s)
lib/data_wds/multiview_wds.py:17
↓ 1 callersFunctionexpand_urls
(urls: Union[str, List[str]])
lib/data_wds/multiview_wds.py:21
↓ 1 callersFunctionextr_to_lookat
(extrinsic_matrix)
lib/datasets/oakink2_dev.py:518
↓ 1 callersFunctionextr_to_lookat
(extrinsic_matrix)
tool/flip_util.py:25
↓ 1 callersMethodextract_img_feat
(self, img)
lib/models/PETR.py:85
↓ 1 callersMethodextract_img_feat
(self, img)
lib/models/darkpose.py:147
↓ 1 callersMethodextract_img_feat
(self, img, backbone="HRNet")
lib/models/POEM.py:229
↓ 1 callersFunctionextract_pred
(pred, batch, req_flip, cam_extr_map)
tool/infer_hand.py:136
↓ 1 callersFunctionextract_spirals
(mesh, seq_length, dilation=1)
lib/external/cmr/utils.py:361
↓ 1 callersMethodfeat_decode
(self, mlvl_feats, backbone="HRNet")
lib/models/POEM.py:172
↓ 1 callersFunctionfit_ortho_param
Fit the orthographic projection parameters. Args: joints3d (np.ndarray): 3D joints. (N, 3) joints2d (np.ndarray): 2D joints. (N,
lib/utils/transform.py:708
↓ 1 callersFunctionflip_cam_extr
(cam_extr)
lib/datasets/oakink2_dev.py:539
↓ 1 callersFunctionformat_batch
(img_list, bbox_list, req_flip, camera_name_list, cam_intr_map, cam_extr_map, img_size, output_size, device)
tool/infer_hand.py:60
↓ 1 callersFunctionformat_cfg
(cfg, indent_lvl=0)
lib/utils/misc.py:115
↓ 1 callersMethodforward_ffn
(self, tgt)
lib/models/layers/mvp_decoder.py:319
↓ 1 callersMethodforward_mano
(self, tgt)
lib/models/layers/mvp_decoder.py:325
← previousnext →301–400 of 1,165, ranked by callers