Functions — github.com/JubSteven/POEM-v2

CodeHub 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

(self, in_channel, out_chanel, kernel_size=3,

lib/external/cmr/net.py:127

lib/external/cmr/net.py:168

(self, center_idx, is_train=True)

lib/external/cmr/data_adaptor.py:65

(self, cfg: CN)

lib/external/cmr/data_adaptor.py:158

(self, backbone)

lib/external/cmr/model.py:27

(self, backbone, in_channel)

lib/external/cmr/model.py:50

Construct a layernorm module in the TF style (epsilon inside the square root).

lib/external/metro/base_model.py:22

lib/external/metro/base_model.py:39

(self, config, backbone, trans_encoder)

lib/external/metro/base_model.py:194

(self, filename="assets/mano_downsampling.npz", num_downsampling=1,

lib/external/metro/base_model.py:349

lib/external/metro/metro.py:35

(self, is_train)

lib/external/metro/data_adaptor.py:14

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

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

(self, num_branches, blocks, num_blocks, n

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

(self, cfg, **kwargs)

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

( self, in_filename: str, pix_fmt: str, cache_size: int = 32, num_fram

video_tool/ffmpeg_util.py:45

lib/utils/builder.py:104

lib/datasets/hdata.py:57

lib/datasets/dexycb.py:120

lib/datasets/dexycb.py:384

lib/datasets/dexycb.py:561

lib/datasets/interhand.py:83

lib/datasets/interhand.py:347

lib/datasets/freihand.py:291

lib/datasets/ho3d.py:175

lib/datasets/ho3d.py:699

lib/datasets/oakink2_dev.py:89

lib/datasets/oakink2_dev.py:364

lib/datasets/yt3d.py:159

lib/datasets/mix_dataset.py:41

lib/datasets/ho3d_official_test.py:175

lib/datasets/arctic.py:91

lib/datasets/arctic.py:289

lib/datasets/oakink.py:108

lib/datasets/oakink.py:479

lib/datasets/oakink.py:680

video_tool/ffmpeg_util.py:115

lib/utils/builder.py:110

str: a string that describes the module

lib/models/layers/petr_transformer.py:471

lib/external/cmr/net.py:53

Method__setattr__

(cls, name, value)

lib/utils/misc.py:59

lib/metrics/basic_metric.py:56

lib/metrics/mean_epe.py:44

lib/metrics/pa_eval.py:96

lib/metrics/pck.py:172

lib/metrics/pck.py:200

Method_build_evaluation

lib/models/model_abc.py:21

Method_build_head_module

lib/models/heads/ptEmb_head.py:207

Method_build_head_module

lib/models/heads/ptEmb_head.py:342

Method_build_head_module

lib/models/heads/ptEmb_head.py:697

Method_build_loss

lib/models/model_abc.py:18

Method_calculate_save_eval_from_train

(self, sample, multiview_info_list)

lib/datasets/ho3d.py:747

Method_debug_viz

(self, debug_metas, src_points_sub, idx=0)

lib/models/heads/ptEmb_head.py:816

Function_get_affine_transform

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

lib/utils/transform.py:649

Method_get_predictions

(self, preds: Dict, targs: Dict)

lib/metrics/pck.py:158

Method_get_predictions

(self, preds: Dict, targs: Dict)

lib/metrics/pck.py:184

Method_getitem_3d

lib/datasets/oakink.py:240

Method_getitem_3d_pack

lib/datasets/oakink.py:306

Method_ho3d_get_hand_info

lib/datasets/ho3d.py:490

Function_init_fn

scripts/train_ddp.py:29

Method_load_annots

(self, obj_meshes={}, seq_frames=[], subfolder="train")

lib/datasets/ho3d.py:483

Method_load_from_state_dict

(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys,

lib/models/backbones/resnet.py:50

lib/datasets/ho3d.py:468

Method_prune_heads

Prunes heads of the model. heads_to_prune: dict of {layer_num: list of heads to prune in this layer} See base class PreTrainedModel

lib/models/bricks/metro_transformer.py:55

Method_prune_heads

Prunes heads of the model. heads_to_prune: dict of {layer_num: list of heads to prune in this layer} See base class PreTrainedModel

lib/models/bricks/pt_metro_transformer.py:131

Method_prune_heads

Prunes heads of the model. heads_to_prune: dict of {layer_num: list of heads to prune in this layer} See base class PreTrainedModel

lib/external/metro/base_model.py:61

Method_register

lib/utils/builder.py:300

Method_save_tmp

lib/datasets/dexycb.py:564

Method_save_tmp

lib/datasets/oakink.py:683

Method_str_custom_args

lib/models/heads/ptEmb_head.py:212

Method_str_custom_args

lib/models/heads/ptEmb_head.py:391

Method_str_custom_args

lib/models/heads/ptEmb_head.py:741

(self, pose, joints, verts, shape, image, cam_intr)

lib/datasets/ho3d.py:703

Method_testing_step

(self, data, step_idx, **kwargs)

lib/external/cmr/model.py:457

Functionaa_to_quat

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

lib/utils/transform.py:411

Functionaa_to_rot6d

Convert axis angles to rotation 6d representations. Args: axis_angle (Union[torch.Tensor, numpy.ndarray]): input shape sh

lib/utils/transform.py:427

Methodadd_geometry_list

(self, geo_list)

lib/viztools/viz_o3d_utils.py:179

Methodadd_scalars

(self, main_tag, tag_scalar_dict, global_step=None, walltime=None)

lib/utils/summary_writer.py:17

Methodalign_by_trafo

lib/utils/testing.py:190

Functionalign_poly

(t, poly, vertex, K, size)

lib/external/cmr/regitstration.py:32

Functionalign_uv

(t, uv, vertex2xyz, K)

lib/external/cmr/regitstration.py:41

Functionapprox_avg_transf_np

(transf_list: list[np.ndarray])

transform/transform_np.py:100

Functionassemble_T

(tsl: torch.Tensor, rotmat: torch.Tensor)

transform/transform_jit.py:5

(ctx, grad_output)

lib/models/layers/mvp_decoder.py:80

Functionbatch_with_heatmap

( inputs, heatmaps, num_rows=2, parts_to_show=None, n_in_batch=1, )

lib/utils/heatmap.py:117

Functionbatch_xyz2uvd

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

lib/utils/transform.py:733

Functionbbox_xyxy_to_xywh

Convert bounding boxes from format (xmin, ymin, xmax, ymax) to (x, y, w, h). Parameters ---------- xyxy : list, tuple or numpy.ndarray

lib/utils/transform.py:1052

Functionbhwc_2_bchw

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

lib/utils/transform.py:1125

Functionblend_mask

(image, mask, random_color=False)

lib/utils/vis_cv2_util.py:570

(self, *args, **kwargs)

lib/utils/builder.py:203

Functionbuild_loss

(cfg: CN, **kwargs)

lib/utils/builder.py:331

Functionbuild_model

(cfg: CN, **kwargs)

lib/utils/builder.py:327

Functionbuild_resnet

lib/models/backbones/resnet.py:300

Functioncal_transform_mean

Calculate the mean of the transform matrix. The rotation mean is calculated thorugh quaternion mean. Args: batch_extr (Union[List[np.

lib/utils/transform.py:1164

Functioncamel_to_snake

lib/utils/misc.py:148

Functioncaption_combined_view

(combine_image, caption="This is a caption", font=DEFAULT_FONT)

lib/utils/vis_cv2_util.py:14

Methodchange_back

lib/utils/config.py:30

lib/utils/builder.py:175

Functioncnt_area

lib/external/cmr/regitstration.py:9

Methodcollate_first_two_dims

lib/models/heads/mvp_head.py:166

← previous next →701–800 of 1,165, ranked by callers