MCPcopy Create free account

hub / github.com/Xuehao-Gao/GUESS / functions

Functions912 in github.com/Xuehao-Gao/GUESS

⨍ Functions 912◇ Types & classes 142
Functionall_collate
(batch)
mld/data/utils.py:25
Functionassign_adain_params
(adain_params, model)
mld/models/operator/adain.py:48
Functionaxis_angle_to
(newtype, rotations)
mld/utils/easyconvert.py:17
Functionaxis_angle_to
(newtype, rotations)
mld/utils/temos_utils.py:44
Functionaxis_angle_to_matrix
Convert rotations given as axis/angle to rotation matrices. Args: axis_angle: Rotations given as a vector in axis angle form,
mld/tools/geometry.py:432
Functionaxis_angle_to_matrix
Convert rotations given as axis/angle to rotation matrices. Args: axis_angle: Rotations given as a vector in axis angle form,
mld/utils/rotation_conversions.py:417
Functionaxis_angle_to_matrix
Convert rotations given as axis/angle to rotation matrices. Args: axis_angle: Rotations given as a vector in axis angle form,
mld/data/a2m/utils/rotation_conversions.py:417
Functionaxisangle2matrots
:param axisangle: N*num_joints*3 :return: N*num_joints*9
mld/utils/geometry.py:60
Methodbackward
(ctx, grad_grad_weight)
mld/models/operator/conv2d_gradfix.py:205
Functionbatch_rodrigues
(axisang)
mld/utils/geometry.py:82
Functionbody_fitting_loss
Loss function for body fitting
mld/transforms/joints2rots/customloss.py:55
Functionbuild_transformer
(args)
mld/models/operator/cross_attention.py:391
Functioncalculate_R_precision
(embedding1, embedding2, top_k, sum_all=False)
mld/data/humanml/utils/metrics.py:37
Functioncalculate_activation_statistics
Params: -- activation: num_samples x dim_feat Returns: -- mu: dim_feat -- sigma: dim_feat x dim_feat
mld/data/humanml/utils/metrics.py:60
Functioncalculate_diversity
(activation, diversity_times)
mld/data/humanml/utils/metrics.py:73
Functioncalculate_diversity
(activation, diversity_times)
mld/models/metrics/utils.py:214
Functioncalculate_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
mld/data/humanml/utils/metrics.py:95
Functioncalculate_matching_score
(embedding1, embedding2, sum_all=False)
mld/data/humanml/utils/metrics.py:47
Functioncalculate_multimodality
(activation, multimodality_times)
mld/data/humanml/utils/metrics.py:84
Functioncamera_fitting_loss
Loss function for camera optimization.
mld/transforms/joints2rots/customloss.py:91
Functioncfg_mean_nsamples_resolution
(cfg)
mld/utils/sample_utils.py:5
Functioncleanexit
()
mld/utils/misc.py:22
Functioncleanexit
()
mld/data/a2m/utils/misc.py:22
Methodclosure
()
mld/transforms/joints2rots/smplify.py:174
Functioncode_path
(path="")
mld/launch/prepare.py:11
Functioncollate
(batch)
mld/utils/tensors.py:24
Methodcollate
(self, lst_datastruct)
mld/transforms/base.py:6
Functioncollate_data3d
(batch)
mld/utils/tensors.py:59
Functioncollate_data3d_slow
(batch)
mld/utils/tensors.py:45
Functioncollate_fn
(batch)
mld/data/humanml/dataset.py:19
Functioncollate_fn
(batch)
mld/data/humanml/data/dataset.py:19
Methodcollate_or_none
(key)
mld/transforms/base.py:10
Functioncolored_material_bsdf
(name, **kwargs)
mld/render/blender/materials.py:82
Functioncolored_material_relection_BSDF
(r, g, b, a=1, roughness=0.127451, saturation_factor=1)
mld/render/blender/materials.py:24
Functioncompose_and_save_img
(img_list, save_dir, img_name, col=4, row=1, img_size=(256, 200))
mld/data/humanml/utils/utils.py:112
Functioncompose_gif_img_list
(img_list, fp_out, duration)
mld/data/humanml/utils/utils.py:86
Methodcompute
(self, sanity_flag)
mld/models/metrics/compute.py:71
Methodcompute
(self, sanity_flag)
mld/models/metrics/stgcn.py:99
Methodcompute
(self, sanity_flag)
mld/models/metrics/gru.py:97
Methodcompute
(self, sanity_flag)
mld/models/metrics/tm2t.py:77
Methodcompute
(self, sanity_flag)
mld/models/metrics/uncond.py:60
Methodcompute
(self, sanity_flag)
mld/models/metrics/mr.py:52
Methodcompute
(self, split)
mld/models/losses/temos.py:152
Methodcompute
(self, split)
mld/models/losses/actor.py:76
Methodcompute
(self, split)
mld/models/losses/mld.py:160
Methodcompute
(self, split)
mld/models/losses/tmost.py:168
Functioncompute_2d_loss
keypoints loss
mld/models/losses/utils.py:18
Functioncompute_glob_loss
(model, batch)
mld/models/losses/utils.py:50
Functioncompute_hp_loss
(model, batch)
mld/models/losses/utils.py:164
Functioncompute_kl_loss
(model, batch)
mld/models/losses/utils.py:169
Functioncompute_limb_loss
(model, batch)
mld/models/losses/utils.py:31
Methodcompute_loss
(self, batch)
mld/models/architectures/uestc_stgcn.py:122
Functioncompute_rc_loss
(model, batch)
mld/models/losses/utils.py:81
Functioncompute_rcverts_loss
(model, batch)
mld/models/losses/utils.py:113
Functioncompute_rcxyz_loss
(model, batch)
mld/models/losses/utils.py:92
Functioncompute_theta_loss
(model, batch)
mld/models/losses/utils.py:64
Functioncompute_vel_loss
(model, batch)
mld/models/losses/utils.py:134
Functioncompute_velxyz_loss
(model, batch)
mld/models/losses/utils.py:149
Methodconfigure_optimizers
(self)
mld/models/modeltype/base.py:129
Functionconv2d
(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1)
mld/models/operator/conv2d_gradfix.py:22
Functionconv_transpose2d
( input, weight, bias=None, stride=1, padding=0, output_padding=0, groups=1, d
mld/models/operator/conv2d_gradfix.py:45
Methodcount_attn
(self, attn)
mld/models/architectures/vision_transformer.py:313
Functioncreate_prior
(prior_type, **kwargs)
mld/transforms/joints2rots/prior.py:35
Methoddecompose
(self)
mld/models/operator/position_encoding.py:32
Methoddetach_or_none
(tensor)
mld/transforms/base.py:62
Functiondetach_to_numpy
(tensor)
mld/utils/temos_utils.py:20
Methodemit
(self, record)
mld/tools/logging.py:34
Functionestimate_translation
This function is borrowed from https://github.com/nkolot/SPIN/utils/geometry.py Find camera translation that brings 3D joints S closest to 2
mld/utils/geometry.py:355
Functioneuclidean_distance_matrix_np
Params: -- matrix1: N1 x D -- matrix2: N2 x D Returns: -- dist: N1 x N2 dist[i, j] == distance(matrix1[i], matrix2[j])
mld/models/metrics/utils.py:44
Functioneuler2quat
Convert Euler angles to quaternions.
mld/data/humanml/common/quaternion.py:169
Functioneuler_angles_to_matrix
Convert rotations given as Euler angles in radians to rotation matrices. Args: euler_angles: Euler angles in radians as tensor of sh
mld/tools/geometry.py:165
Functioneuler_angles_to_matrix
Convert rotations given as Euler angles in radians to rotation matrices. Args: euler_angles: Euler angles in radians as tensor of sh
mld/utils/rotation_conversions.py:150
Functioneuler_angles_to_matrix
Convert rotations given as Euler angles in radians to rotation matrices. Args: euler_angles: Euler angles in radians as tensor of sh
mld/data/a2m/utils/rotation_conversions.py:150
Functioneuler_to_quaternion
Convert Euler angles to quaternions.
mld/data/humanml/common/quaternion.py:233
Methodeval_gt
(self, batch, renoem=True)
mld/models/modeltype/mld.py:962
Functionexpmap_to_quaternion
Convert axis-angle rotations (aka exponential maps) to quaternions. Stable formula from "Practical Parameterization of Rotations Using the Ex
mld/data/humanml/common/quaternion.py:214
Functionexport_fbx
(pkl_path)
mld/utils/demo_utils.py:69
Functionexport_fbx_hand
(pkl_path)
mld/utils/demo_utils.py:54
Functionextend_paths
(path, keyids, *, onesample=True, number_of_samples=1)
render.py:24
Functionextract_features
(positions, feet_thre, n_raw_offsets, kinematic_chain, face_joint_indx, fid_r, fid_l)
mld/data/humanml/scripts/motion_process.py:39
Methodfeats2joints
(self, features)
mld/data/HumanML3D.py:41
Methodfilter
(self, record)
mld/tools/logging.py:9
Methodforward
(self, *args, **kwargs)
mld/transforms/smpl.py:241
Methodforward
(self, s1_input)
mld/transforms/multi_scale_init.py:22
Methodforward
(self, s1_input)
mld/transforms/multi_scale_init.py:54
Methodforward
(self, s1_input)
mld/transforms/multi_scale_init.py:75
Methodforward
(self, s1_input)
mld/transforms/multi_scale_init.py:133
Methodforward
(self, s1_input)
mld/transforms/multi_scale_init.py:156
Methodforward
(self, smpl_data: dict, jointstype: Optional[str] = None, inpu
mld/transforms/rots2joints/smplh.py:45
Methodforward
(self, joints: Tensor)
mld/transforms/joints2jfeats/rifke.py:27
Methodforward
Returns the angle prior loss for the given pose Args: pose: (Bx[23 + 1] * 3) torch tensor with the axis-angle represe
mld/transforms/joints2rots/prior.py:72
Methodforward
(self, module_input, *args)
mld/transforms/joints2rots/prior.py:94
Methodforward
(self, pose, betas)
mld/transforms/joints2rots/prior.py:225
Methodforward
(self, data)
mld/transforms/rots2rfeats/smplvelp.py:28
Methodforward
(self, texts: List[str])
mld/models/architectures/mld_clip.py:50
Methodforward
(self, texts: List[str])
mld/models/architectures/mld_bert.py:162
Methodforward
(self, inputs)
mld/models/architectures/t2m_motionenc.py:21
Methodforward
(self, inputs, m_lens)
mld/models/architectures/t2m_motionenc.py:51
Methodforward
(self, motion)
mld/models/architectures/uestc_stgcn.py:73
Methodforward
(self, x, A)
mld/models/architectures/uestc_stgcn.py:203
← previousnext →601–700 of 912, ranked by callers