MCPcopy Create free account

hub / github.com/TaeryungLee/MultiAct_RELEASE / functions

Functions217 in github.com/TaeryungLee/MultiAct_RELEASE

⨍ Functions 217◇ Types & classes 51
↓ 33 callersMethoddebug
(self, msg)
utils/logger.py:41
↓ 22 callersMethodcomp_register
(self, name, value, persistent=False)
utils/smplx/body_model/body_model.py:175
↓ 15 callersFunctionto_tensor
( array: Union[Array, Tensor], dtype=torch.float32 )
utils/smplx/smplx/utils.py:98
↓ 13 callersMethoditems
(self)
utils/smplx/smplx/utils.py:51
↓ 12 callersMethodevaluate_S2_BABEL
Inputs gen_joint (num_gen * maxlen * 52 * 3) gen_mask (num_gen * maxlen) gen_label (num_gen * 3)
main/tester.py:608
↓ 10 callersMethodkeys
(self)
utils/smplx/smplx/utils.py:43
↓ 10 callersFunctionlbs
Performs Linear Blend Skinning with the given shape and pose parameters Parameters ---------- betas : torch.tensor BxNB
utils/smplx/smplx/lbs.py:152
↓ 9 callersMethod__init__
SMPL model constructor Parameters ---------- model_path: str The path to the folder or to the fi
utils/smplx/smplx/body_models.py:49
↓ 9 callersFunctionto_np
(array, dtype=np.float32)
utils/smplx/smplx/utils.py:113
↓ 8 callersMethodsplit
(self, vec)
evaluate/motion_gan.py:341
↓ 7 callersFunctionget_dataloader
(cfg, split="train", sampled_file_path=None, sampled=None, filter_label=None, cap=None, drop_last=True)
data/dataset.py:48
↓ 7 callersFunctionmkdir
(path)
utils/dir.py:5
↓ 6 callersMethodinit_hidden
(self, num_samples, layers=1)
evaluate/motion_gan.py:189
↓ 5 callersMethod__init__
(self, use_noise, sigma=0.2)
evaluate/motion_gan.py:11
↓ 5 callersMethodinfo
(self, msg)
utils/logger.py:44
↓ 4 callersMethodestimate_distribution
(self, inputs)
main/tester.py:743
↓ 4 callersMethodevaluate_transition_BABEL
(self, gen_joint, gt_joint)
main/tester.py:679
↓ 4 callersMethodextra_repr
(self)
utils/smplx/smplx/body_models.py:280
↓ 4 callersFunctionfind_dynamic_lmk_idx_and_bcoords
Compute the faces, barycentric coordinates for the dynamic landmarks To do so, we first compute the rotation of the neck around the y-axis
utils/smplx/smplx/lbs.py:30
↓ 4 callersFunctionvertices2landmarks
Calculates landmarks by barycentric interpolation Parameters ---------- vertices: torch.tensor BxVx3, dtype = torch.float32
utils/smplx/smplx/lbs.py:108
↓ 4 callersFunctionvisualize_batch
(cfg, meshes, masks, paths, names, labels, out_size=1200, method="render")
utils/vis.py:258
↓ 3 callersMethod__init__
(self, cfg)
utils/loss.py:68
↓ 3 callersMethod__init__
(self, d_model, device=None, max_len=5000)
models/modules/transformer.py:18
↓ 3 callersFunctionbatch_rodrigues
Calculates the rotation matrices for a batch of rotation vectors Parameters ---------- rot_vecs: torch.tensor Nx3
utils/smplx/body_model/lbs.py:303
↓ 3 callersFunctionget_model
(cfg)
models/model.py:259
↓ 3 callersMethodinitHidden
(self, num_samples, layer)
evaluate/motion_gan.py:328
↓ 3 callersMethodreparameterize
(self, mu, logvar)
models/model.py:243
↓ 3 callersFunctionset_env
(env_file, arg_gpu_ids)
main/config.py:95
↓ 2 callersFunctionbatch_rodrigues
Calculates the rotation matrices for a batch of rotation vectors Parameters ---------- rot_vecs: torch.tensor Nx3
utils/smplx/smplx/lbs.py:295
↓ 2 callersMethodcalculate_fid
(self, statistics_1, statistics_2)
main/tester.py:750
↓ 2 callersFunctionfind_joint_kin_chain
(joint_id, kinematic_tree)
utils/smplx/smplx/utils.py:89
↓ 2 callersMethodform_into_result
(self, inputs, out, recur)
main/generator.py:398
↓ 2 callersMethodgenerate_all_gt
(self, split="train")
main/tester.py:572
↓ 2 callersMethodgenerate_motion
(self, num_samples)
evaluate/motion_gan.py:138
↓ 2 callersMethodgenerate_output
(self, split="unseen")
main/tester.py:532
↓ 2 callersMethodget_batch_smpl_mesh_cam
(self, body_params)
models/model.py:248
↓ 2 callersMethodget_correct_pred_conf
(self, pred, label)
main/tester.py:695
↓ 2 callersMethodget_correct_top5_pred_conf
(self, pred, label)
main/tester.py:716
↓ 2 callersMethodget_normal_noise
(self, num_samples)
evaluate/motion_gan.py:193
↓ 2 callersFunctionload_classifier_for_fid
(cfg, device)
evaluate/load_classifier.py:143
↓ 2 callersMethodsample_train_data
(self, method='basic')
data/datasets/BABEL.py:197
↓ 2 callersMethodsample_z_motion
(self, num_samples, motion_len=None)
evaluate/motion_gan.py:102
↓ 2 callersMethodsave_model
(self, name, dump_smpl=True)
main/trainer.py:152
↓ 2 callersFunctionsec2minhrs
(sec)
utils/timer.py:32
↓ 2 callersMethodtrain
(self)
main/trainer.py:53
↓ 2 callersFunctionvis_motion
visualize batched mesh Inputs: mesh ((batch) * duration * num_vertices * 3) face (num_faces * 3) Returns: out ((b
utils/vis.py:194
↓ 2 callersFunctionvis_motion_vertices
visualize batched mesh Inputs: mesh ((batch) * duration * num_vertices * 3) face (num_faces * 3) Returns: out ((b
utils/vis.py:227
↓ 1 callersMethod__init__
(self, cfg)
models/modules/postprocess.py:7
↓ 1 callersFunctionbatch_rigid_transform
Applies a batch of rigid transformations to the joints Parameters ---------- rot_mats : torch.tensor BxNx3x3 Tensor of rotat
utils/smplx/smplx/lbs.py:345
↓ 1 callersFunctionbatch_rigid_transform
Applies a batch of rigid transformations to the joints Parameters ---------- rot_mats : torch.tensor BxNx3x3 Tensor of rotat
utils/smplx/body_model/lbs.py:350
↓ 1 callersFunctionblend_shapes
Calculates the per vertex displacement due to the blend shapes Parameters ---------- betas : torch.tensor Bx(num_betas) Blend s
utils/smplx/smplx/lbs.py:271
↓ 1 callersFunctionblend_shapes
Calculates the per vertex displacement due to the blend shapes Parameters ---------- betas : torch.tensor Bx(num_betas) Blend s
utils/smplx/body_model/lbs.py:277
↓ 1 callersMethodcalculate_diversity_multimodality
(self, activations, labels, num_labels)
main/tester.py:754
↓ 1 callersFunctioncalculate_frechet_distance
Numpy implementation of the Frechet Distance. The Frechet distance between two multivariate Gaussians X_1 ~ N(mu_1, C_1) and X_2 ~ N(mu_2, C_2
evaluate/fid.py:5
↓ 1 callersFunctionclean_fn
(fn, output_folder='output')
utils/smplx/tools/clean_ch.py:32
↓ 1 callersMethodcreate_mean_pose
(self, data_struct, flat_hand_mean=False)
utils/smplx/smplx/body_models.py:648
↓ 1 callersMethodcreate_mean_pose
(self, data_struct, flat_hand_mean=False)
utils/smplx/smplx/body_models.py:1553
↓ 1 callersFunctionface_front_align
Regularize motion by given axis angle rotation and translation inputs: params ((batch): optional * duration * num_param): global orie
data/datautils/transforms.py:9
↓ 1 callersMethodfilter_train_data
(self, cut="twenty")
data/datasets/BABEL.py:155
↓ 1 callersMethodforward
:param root_orient: Nx3 :param pose_body: :param pose_hand: :param pose_jaw: :param pose_eye: :param
utils/smplx/body_model/body_model.py:185
↓ 1 callersMethodforward
(self, x)
evaluate/motion_gan.py:16
↓ 1 callersMethodgenerate_long_term
(self)
main/generator.py:125
↓ 1 callersMethodgenerate_short_term
(self)
main/generator.py:90
↓ 1 callersFunctionlbs
Performs Linear Blend Skinning with the given shape and pose parameters Parameters ---------- betas : torch.tensor BxNB
utils/smplx/body_model/lbs.py:164
↓ 1 callersFunctionload_classifier
(cfg, device)
evaluate/load_classifier.py:109
↓ 1 callersMethodload_data
(self, split)
data/datasets/BABEL.py:383
↓ 1 callersMethodload_model
(self, load_path)
main/tester.py:799
↓ 1 callersMethodload_model
(self, load_path)
main/generator.py:445
↓ 1 callersMethodload_model
(self, load_path)
main/trainer.py:190
↓ 1 callersFunctionmain
(model_folder, model_type='smplx', ext='npz', gender='neutral', plot_joint
utils/smplx/examples/demo_layers.py:26
↓ 1 callersFunctionmain
(model_folder, corr_fname, ext='npz', hand_color=(0.3, 0.3, 0.6), gender='neutral', hand='ri
utils/smplx/examples/vis_mano_vertices.py:28
↓ 1 callersFunctionmain
(model_folder, corr_fname, ext='npz', head_color=(0.3, 0.3, 0.6), gender='neutral')
utils/smplx/examples/vis_flame_vertices.py:28
↓ 1 callersFunctionmain
(model_folder, model_type='smplx', ext='npz', gender='neutral', plot_joint
utils/smplx/examples/demo.py:26
↓ 1 callersFunctionmerge_models
(smplh_fn, mano_left_fn, mano_right_fn, output_folder='output')
utils/smplx/tools/merge_smplh_mano.py:29
↓ 1 callersMethodname
(self)
utils/smplx/smplx/body_models.py:1019
↓ 1 callersMethodname
(self)
utils/smplx/smplx/body_models.py:1906
↓ 1 callersFunctionparse_args
()
train.py:8
↓ 1 callersFunctionparse_args
()
test.py:9
↓ 1 callersFunctionparse_args
()
generate.py:8
↓ 1 callersFunctionrender_mesh2
(mesh, face, out_size)
utils/vis.py:174
↓ 1 callersFunctionrot6d_to_axis_angle
(x)
data/datautils/transforms.py:123
↓ 1 callersFunctionrot_mat_to_euler
(rot_mats)
utils/smplx/smplx/utils.py:119
↓ 1 callersFunctionrot_mat_to_euler
(rot_mats)
utils/smplx/body_model/lbs.py:49
↓ 1 callersMethodrun_evaluate_transition
(self, seen_datas=None, unseen_datas=None)
main/tester.py:474
↓ 1 callersMethodsample_z_categ
(self, num_samples, motion_len=None)
evaluate/motion_gan.py:68
↓ 1 callersMethodsample_z_r
(self, num_samples, motion_len=None)
evaluate/motion_gan.py:45
↓ 1 callersMethodsubsequent_mask
(self, size)
models/modules/transformer.py:253
↓ 1 callersMethodtest
(self)
main/tester.py:59
↓ 1 callersMethodtic
(self)
utils/timer.py:18
↓ 1 callersFunctionto_front_view
Inputs: params (duration * 156) Outputs: params (duration * 156)
data/datautils/transforms.py:74
↓ 1 callersFunctionto_right_view
Inputs: params (duration * 156) Outputs: params (duration * 156)
data/datautils/transforms.py:100
↓ 1 callersMethodtoc
(self)
utils/timer.py:23
↓ 1 callersFunctiontransform_mat
Creates a batch of transformation matrices Args: - R: Bx3x3 array of a batch of rotation matrices - t: Bx3x1 array of
utils/smplx/smplx/lbs.py:332
↓ 1 callersFunctiontransform_mat
Creates a batch of transformation matrices Args: - R: Bx3x3 array of a batch of rotation matrices - t: Bx3x1 array of
utils/smplx/body_model/lbs.py:337
↓ 1 callersMethodvalues
(self)
utils/smplx/smplx/utils.py:47
↓ 1 callersFunctionvertices2joints
Calculates the 3D joint locations from the vertices Parameters ---------- J_regressor : torch.tensor JxV The regressor array tha
utils/smplx/smplx/lbs.py:251
↓ 1 callersFunctionvertices2joints
Calculates the 3D joint locations from the vertices Parameters ---------- J_regressor : torch.tensor JxV The regressor array tha
utils/smplx/body_model/lbs.py:257
↓ 1 callersFunctionvis_keypoints
(img, kps, alpha=1)
utils/vis.py:47
Method__getitem__
(self, key)
utils/smplx/smplx/utils.py:34
Method__getitem__
(self, index)
data/dataset.py:27
next →1–100 of 217, ranked by callers