MCPcopy Create free account

hub / github.com/XunshanMan/MVGFormer / functions

Functions561 in github.com/XunshanMan/MVGFormer

↓ 105 callersFunctionprint
(*args, **kwargs)
lib/utils/misc.py:363
↓ 70 callersMethodto
(self, device, non_blocking=False)
lib/models/util/misc.py:425
↓ 37 callersFunctionprint
(*args, **kwargs)
lib/models/util/misc.py:455
↓ 37 callersMethodupdate
(self, val, n=1)
lib/core/function.py:600
↓ 32 callersMethodcreate_node
Create a NODE and add it to the node_list. name, idx, parent respectively corespond to the "name", "index", "parent" keys.
lib/structural/structural_triangulation.py:32
↓ 31 callersMethodto
(self, device)
lib/utils/misc.py:338
↓ 17 callersMethodmax
(self)
lib/models/util/misc.py:133
↓ 11 callersFunctionis_main_process
()
lib/utils/misc.py:391
↓ 8 callersMethodupdate
(self, **kwargs)
lib/models/util/misc.py:277
↓ 7 callersFunctionstructural_triangulate_points
@ bone_length: (batch_size, N_joints, 1) or (1, N_joints, 1)
lib/structural/adapter.py:20
↓ 7 callersFunctiontime_synchronized
()
lib/core/function.py:39
↓ 7 callersMethodupdate
(self, val, n=1)
lib/models/dq_decoder.py:82
↓ 7 callersMethodwith_pos_embed
(tensor, pos)
lib/models/mvp_decoder.py:91
↓ 6 callersMethod__init__
(self, use_target_weight)
lib/core/loss.py:34
↓ 6 callersMethodevaluate
(self, preds)
lib/dataset/h36m.py:237
↓ 6 callersFunctionis_dist_avail_and_initialized
()
lib/models/util/misc.py:463
↓ 6 callersFunctionsave_ref_points_with_gt
(preds, meta, file_name, color=None, transparency=1, range_in=None, indices=None)
lib/utils/vis.py:202
↓ 5 callersMethod__init__
(self, input_channels, output_channels)
lib/mvn/models/v2v.py:142
↓ 5 callersFunctioncreate_human_tree
create human tree structure according to data_type return a DictTree object.
lib/structural/structural_triangulation.py:119
↓ 5 callersMethodget_loss
(self, loss, outputs, outputs_2d, targets, indices, num_samples, **kwargs)
lib/models/multi_view_pose_transformer.py:790
↓ 5 callersFunctionhomogeneous_to_euclidean
Converts homogeneous points to euclidean Args: points numpy array or torch tensor of shape (N, M + 1): N homogeneous points of dimension
lib/mvn/utils/multiview.py:72
↓ 5 callersMethodmax
(self)
lib/utils/misc.py:97
↓ 5 callersFunctionparse_string_to_keyvalue
(args)
lib/utils/string_parser.py:36
↓ 5 callersFunctionproject_pose
(x, camera)
lib/utils/cameras_cpu.py:63
↓ 5 callersFunctionretrieve_valid
(output, batch_ids, query_ids, num_joints, input_list=False)
lib/models/dq_decoder.py:1174
↓ 5 callersFunctionsave_debug_3d_images
(config, meta, preds, prefix, show_id=False)
lib/utils/vis.py:683
↓ 5 callersFunctionto_numpy
(tensor)
lib/mvn/utils/img.py:72
↓ 5 callersMethodupdate
(self, val, n=1)
lib/models/dq_transformer.py:78
↓ 4 callersMethod__init__
(self)
lib/mvn/models/loss.py:43
↓ 4 callersMethod__init__
(self, inplanes, planes, stride=1, downsample=None)
lib/mvn/models/pose_resnet.py:28
↓ 4 callersMethod__init__
(self, fc_layers, use_dropout, drop_prob, use_ac_func)
lib/smpl/smpl_common.py:36
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
lib/mvn/models/pose_resnet.py:236
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
lib/models/pose_resnet.py:141
↓ 4 callersMethodabsolute2norm
(self, absolute_coords)
lib/models/mvp_decoder.py:284
↓ 4 callersFunctionget_affine_transform
(center, scale, rot, output_size,
lib/utils/transforms.py:72
↓ 4 callersMethodnorm2absolute
(self, norm_coords)
lib/models/multi_view_pose_transformer.py:575
↓ 4 callersMethodrender
(self, proj_matrix, canvas)
lib/mvn/utils/volumetric.py:31
↓ 4 callersFunctionresize_image
(image, shape)
lib/mvn/utils/img.py:27
↓ 4 callersFunctionupdate_config
(config_file)
lib/core/config.py:360
↓ 4 callersFunctionwandb_log_io
(data)
lib/core/function.py:48
↓ 3 callersFunctionLagrangian_method
Implementation of Lagrangian Algorithm for constrained HPE problem.
lib/structural/structural_triangulation.py:487
↓ 3 callersFunction_get_clones
(module, N)
lib/models/multi_view_pose_transformer.py:105
↓ 3 callersFunction_run
(command)
lib/utils/misc.py:279
↓ 3 callersFunction_run
(command)
lib/models/util/misc.py:365
↓ 3 callersFunctionconstruct_output_from_origin
(init_reference, device, calculate_2d=False,num_joints=15,convert_joint_format_indices=None)
lib/models/dq_transformer.py:90
↓ 3 callersFunctioncreate_logger
(cfg, cfg_name, phase='train')
lib/utils/utils.py:36
↓ 3 callersFunctiondenormalize_image
Reverse to normalize_image() function
lib/mvn/utils/img.py:113
↓ 3 callersFunctiondraw_2d_pose
Visualizes a 2d skeleton Args keypoints numpy array of shape (19, 2): pose to draw in CMU format. ax: matplotlib axis to dra
lib/mvn/utils/vis.py:267
↓ 3 callersFunctionfig_to_array
(fig)
lib/mvn/utils/vis.py:77
↓ 3 callersFunctionget_affine_transform
:param center: (2,) :param scale: (2,) :param patch_size: :param inv: inv=0 image->crop_img :return:
lib/utils/cameras_cuda_col.py:130
↓ 3 callersFunctionget_proj_matricies_batch
(meta, nviews, device, inv_trans=False)
lib/models/dq_decoder.py:223
↓ 3 callersFunctionget_world_size
()
lib/models/util/misc.py:471
↓ 3 callersFunctionimage_batch_to_numpy
(image_batch)
lib/mvn/utils/img.py:90
↓ 3 callersFunctioninverse_sigmoid
(x, eps=1e-5)
lib/models/util/misc.py:608
↓ 3 callersFunctionis_dist_avail_and_initialized
()
lib/utils/misc.py:371
↓ 3 callersFunctionproject_3d_points_to_image_plane_without_distortion
Project 3D points to image plane not taking into account distortion Args: proj_matrix numpy array or torch tensor of shape (3, 4): project
lib/mvn/utils/multiview.py:89
↓ 3 callersFunctionupdate_config_dynamic_input
(args)
lib/core/config.py:377
↓ 2 callersFunctionST_SCA_torch
Structural Triangulation with step contrain algorithm. When n_step == 1, this is pure ST without SCA.
lib/structural/structural_triangulation.py:528
↓ 2 callersMethod__getitem__
(self, idx)
lib/dataset/panoptic.py:482
↓ 2 callersMethod__init__
(self, config, device='cuda:0')
lib/mvn/models/triangulation.py:18
↓ 2 callersMethod__init__
(self, dtype=DEFAULT_DTYPE, reduction='sum', **kwargs)
lib/smpl/prior.py:107
↓ 2 callersMethod__init__
(self, inplanes, planes, stride=1, downsample=None)
lib/models/pose_resnet.py:39
↓ 2 callersMethod__init__
(self, input_dim, hidden_dim, output_dim, num_layers)
lib/models/multi_view_pose_transformer.py:91
↓ 2 callersMethod__init__
(self, num_pos_feats=64, temperature=10000, normalize=False, scale=None)
lib/models/position_encoding.py:32
↓ 2 callersMethod__init__
(self, cfg, decoder_layer, num_layers, return_intermediate=False)
lib/models/dq_decoder.py:1102
↓ 2 callersFunction_get_activation_fn
Return an activation function given a string
lib/models/mvp_decoder.py:332
↓ 2 callersMethod_get_src_permutation_idx
(self, indices)
lib/models/multi_view_pose_transformer.py:776
↓ 2 callersFunction_output_size
(dim, input, size, scale_factor)
lib/models/util/misc.py:65
↓ 2 callersMethod_reset_parameters
(self)
lib/models/ops/modules/projattn.py:96
↓ 2 callersMethod_set_aux_loss
(self, outputs_class, outputs_coord)
lib/models/multi_view_pose_transformer.py:300
↓ 2 callersFunction_update_dict
(k, v)
lib/core/config.py:333
↓ 2 callersMethodabsolute2norm
(self, absolute_coords)
lib/models/multi_view_pose_transformer.py:283
↓ 2 callersFunctionaccuracy
Computes the precision@k for the specified values of k
lib/models/util/misc.py:548
↓ 2 callersFunctionadd_path
(path)
run/_init_paths.py:28
↓ 2 callersFunctionaffine_transform
(pt, t)
lib/utils/transforms.py:115
↓ 2 callersFunctionaffine_transform_pts
:param pts: (2, N) :param trans: (2, 3) :return:
lib/utils/cameras_cuda_col.py:161
↓ 2 callersMethodbackward
(ctx, grad_output)
lib/models/ops/functions/deform_func.py:50
↓ 2 callersMethodcollate_first_two_dims
(self, tensor)
lib/models/multi_view_pose_transformer.py:304
↓ 2 callersFunctioncollect_results
(result_part, size, tmpdir=None)
lib/models/util/misc.py:149
↓ 2 callersMethodcompute_human_scale
(self, pose, joints_vis)
lib/dataset/JointsDataset.py:224
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
lib/mvn/models/pose_resnet.py:19
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
lib/models/pose_resnet.py:30
↓ 2 callersFunctiondraw_plot
(data, save_dir, multiple=False)
process/extract_number.py:18
↓ 2 callersFunctioneuclidean_to_homogeneous
Converts euclidean points to homogeneous Args: points numpy array or torch tensor of shape (N, M): N euclidean points of dimension M
lib/mvn/utils/multiview.py:55
↓ 2 callersMethodevaluate_using_per_pose_error
(self, per_pose_error, split_by_subject)
lib/mvn/datasets/human36m.py:191
↓ 2 callersFunctionexplode
(data)
lib/mvn/utils/vis.py:440
↓ 2 callersFunctionextract_losses_from_file
(loss_name,file,add_prefix=True)
process/extract_number.py:35
↓ 2 callersMethodforward_ffn
(self, tgt)
lib/models/mvp_decoder.py:94
↓ 2 callersMethodgenerate_valid_masks
(self, outputs_class, method='threshold', value=0.5)
lib/models/dq_decoder.py:596
↓ 2 callersFunctionget_3rd_point
(a, b)
lib/utils/transforms.py:144
↓ 2 callersFunctionget_calib_matrix
(camera, device)
lib/models/dq_decoder.py:207
↓ 2 callersFunctionget_dir
(src_point, rot_rad)
lib/utils/transforms.py:149
↓ 2 callersFunctionget_inner_mat_torch
(u, v, device)
lib/structural/structural_triangulation.py:184
↓ 2 callersMethodget_valid_ratio
(self, mask)
lib/models/multi_view_pose_transformer.py:265
↓ 2 callersFunctionget_world_size
()
lib/utils/misc.py:379
↓ 2 callersFunctionmatch_name_keywords
(n, name_keywords)
process/extract_dataset.py:108
↓ 2 callersFunctionmatch_name_keywords
(n, name_keywords)
run/train_3d.py:107
↓ 2 callersFunctionnearby_joints_nms
Nearby joints NMS implementations. Args: kpts_db (list[dict]): keypoints and scores. dist_thr (float): threshold for judging whet
lib/core/nms.py:210
↓ 2 callersMethodnorm2absolute
(self, norm_coords)
lib/models/matcher.py:73
↓ 2 callersFunctionnormalize_image
(im, to_rgb=False)
lib/utils/vis.py:63
next →1–100 of 561, ranked by callers