MCPcopy Create free account

hub / github.com/OpenMOSS/FRoM-W1 / functions

Functions453 in github.com/OpenMOSS/FRoM-W1

⨍ Functions 453◇ Types & classes 54
Functionaxis_angle_to
(newtype, rotations)
H-GPT/hGPT/utils/temos_utils.py:42
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,
H-GPT/hGPT/utils/rotation_conversions.py:416
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,
H-GPT/hGPT/utils/geometry_conver.py:85
Functionaxisangle2matrots
:param axisangle: N*num_joints*3 :return: N*num_joints*9
H-GPT/hGPT/utils/geometry_conver.py:139
Functionbatch_rodrigues
(axisang)
H-GPT/hGPT/utils/geometry_conver.py:161
Functionbuild_callbacks
(cfg, phase="test", logger=None, **kwargs)
H-GPT/hGPT/callback.py:73
Functioncalculate_activation_statistics
Params: -- activation: num_samples x dim_feat Returns: -- mu: dim_feat -- sigma: dim_feat x dim_feat
H-GPT/hGPT/metrics/utils.py:77
Functioncalculate_diversity
(activation, diversity_times)
H-GPT/hGPT/metrics/utils.py:216
Functioncalculate_diversity_multimodality
( activations, labels, num_labels, diversity_times=200, multimodality_times=20 )
H-GPT/hGPT/metrics/utils.py:393
Functioncalculate_fid
(statistics_1, statistics_2)
H-GPT/hGPT/metrics/utils.py:436
Functioncalculate_kid
(real_activations, generated_activations)
H-GPT/hGPT/metrics/utils.py:591
Functioncfg_mean_nsamples_resolution
(cfg)
H-GPT/hGPT/utils/sample_utils.py:7
Functioncleanexit
()
H-GPT/hGPT/utils/misc.py:20
Functioncollate
(batch)
H-GPT/hGPT/utils/tensors.py:25
Functioncollate_data3d
(batch)
H-GPT/hGPT/utils/tensors.py:65
Functioncollate_data3d_slow
(batch)
H-GPT/hGPT/utils/tensors.py:50
Methodcompute
(self, sanity_flag)
H-GPT/hGPT/metrics/t2m.py:139
Methodcompute
(self, sanity_flag)
H-GPT/hGPT/metrics/m2m.py:42
Methodcompute
(self, sanity_flag)
H-GPT/hGPT/metrics/m2t.py:217
Methodcompute
(self, sanity_flag)
H-GPT/hGPT/metrics/mr.py:36
Methodcompute
Compute the losses and return a dictionary with the losses.
H-GPT/hGPT/losses/base.py:43
Methodconfigure_optimizers
(self)
H-GPT/hGPT/models/base.py:28
Functioncont6d_to_quat
(cont6d)
H-GPT/hGPT/data/motionx/common/quaternion.py:546
Functiondetach_to_numpy
(tensor)
H-GPT/hGPT/utils/temos_utils.py:21
Functiondraw_to_batch
(smpl_joints_batch, title_batch=None, outname=None)
H-GPT/hGPT/data/motionx/visualization/plot_3d_global.py:395
Functiondraw_to_batch_smplh_folder
(kinematic_chain, input_folder)
H-GPT/hGPT/data/motionx/visualization/plot_3d_global.py:437
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
H-GPT/hGPT/utils/geometry_conver.py:473
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])
H-GPT/hGPT/metrics/utils.py:43
Functioneuler2quat
Convert Euler angles to quaternions.
H-GPT/hGPT/data/motionx/common/quaternion.py:178
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
H-GPT/hGPT/utils/geometry_tools.py:164
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
H-GPT/hGPT/utils/rotation_conversions.py:149
Functioneuler_to_quaternion
Convert Euler angles to quaternions.
H-GPT/hGPT/data/motionx/common/quaternion.py:251
Functionexpmap_to_quaternion
Convert axis-angle rotations (aka exponential maps) to quaternions. Stable formula from "Practical Parameterization of Rotations Using the Ex
H-GPT/hGPT/data/motionx/common/quaternion.py:232
Functionexport_fbx
(pkl_path)
H-GPT/hGPT/utils/demo_utils.py:70
Functionexport_fbx_hand
(pkl_path)
H-GPT/hGPT/utils/demo_utils.py:57
FunctionfindAllFile
Recursively find all files in the specified directory. Args: base (str): The base directory to start the search. Returns:
H-GPT/hGPT/data/motionx/scripts/raw_pose_processing.py:61
Methodforward
(self, x)
H-GPT/hGPT/models/evaluator.py:16
Methodforward
(self, x)
H-GPT/hGPT/models/evaluator.py:67
Methodforward
(self, x)
H-GPT/hGPT/models/evaluator.py:117
Methodforward
(self, output1, output2, label)
H-GPT/hGPT/models/evaluator.py:131
Methodforward
(self, pos)
H-GPT/hGPT/models/evaluator.py:198
Methodforward
(self, inputs)
H-GPT/hGPT/models/evaluator.py:217
Methodforward
(self, x)
H-GPT/hGPT/models/evaluator.py:257
Methodforward
(self, inputs)
H-GPT/hGPT/models/evaluator.py:281
Methodforward
(self, inputs, last_pred, hidden, p)
H-GPT/hGPT/models/evaluator.py:338
Methodforward
(self, inputs, hidden, p)
H-GPT/hGPT/models/evaluator.py:387
Methodforward
query (batch, query_dim) key (batch, seq_len, key_dim)
H-GPT/hGPT/models/evaluator.py:416
Methodforward
(self, word_embs, pos_onehot, cap_lens)
H-GPT/hGPT/models/evaluator.py:458
Methodforward
(self, word_embs, pos_onehot, cap_lens)
H-GPT/hGPT/models/evaluator.py:513
Methodforward
(self, inputs, m_lens)
H-GPT/hGPT/models/evaluator.py:559
Methodforward
(self, word_embs, pos_onehot, cap_lens)
H-GPT/hGPT/models/evaluator.py:615
Methodforward
( self, texts: List[str], motion_tokens: Tensor, motion_tokens_len: List[int],
H-GPT/hGPT/models/archs/hgpt_lm.py:153
Methodforward
(self, features)
H-GPT/hGPT/models/archs/hgpt_vq.py:76
Methodforward
(self, x)
H-GPT/hGPT/models/archs/hgpt_vq.py:152
Methodforward
(self, x)
H-GPT/hGPT/models/archs/hgpt_vq.py:196
Methodforward
(self, input: Tensor)
H-GPT/hGPT/models/archs/utils/token_emb.py:55
Methodforward
(self, x)
H-GPT/hGPT/models/archs/utils/quantize_cnn.py:109
Methodforward
(self, z)
H-GPT/hGPT/models/archs/utils/quantize_cnn.py:152
Methodforward
(self, x)
H-GPT/hGPT/models/archs/utils/quantize_cnn.py:296
Methodforward
(self, x)
H-GPT/hGPT/models/archs/utils/quantize_cnn.py:421
Methodforward
(self, x)
H-GPT/hGPT/models/archs/utils/resnet.py:9
Methodforward
(self, x)
H-GPT/hGPT/models/archs/utils/resnet.py:60
Methodforward
(self, x)
H-GPT/hGPT/models/archs/utils/resnet.py:110
Methodforward
(self, commit, commit2, **kwargs)
H-GPT/hGPT/losses/hgpt.py:15
Methodforward_kinematics
( self, quat_params, root_pos, skel_joints=None, do_root_R=True )
H-GPT/hGPT/data/motionx/common/skeleton.py:163
Methodforward_kinematics_cont6d_np
( self, cont6d_params, root_pos, skel_joints=None, do_root_R=True )
H-GPT/hGPT/data/motionx/common/skeleton.py:224
Methodget_init_hidden
(self, latent)
H-GPT/hGPT/models/evaluator.py:333
Methodget_init_hidden
(self, latent)
H-GPT/hGPT/models/evaluator.py:380
Methodget_metrics
(self, trainer, model)
H-GPT/hGPT/callback.py:12
Functionget_padding_mask
(batch_size, seq_len, cap_lens)
H-GPT/hGPT/models/evaluator.py:175
Functionget_path
( sample_path: Path, is_amass: bool, gender: str, split: str, onesample: bool, mean: b
H-GPT/hGPT/utils/sample_utils.py:17
Functionget_quaternion
Computes quaternion parameters, root linear velocity, and root angular velocity based on the input motion capture data. Args
H-GPT/hGPT/data/motionx/scripts/motion_process.py:223
Functionget_root_idx
(joinstype)
H-GPT/hGPT/utils/joints.py:421
Methodget_sample_set
(self, overrides={})
H-GPT/hGPT/data/base.py:34
Functioninit_weight
(m)
H-GPT/hGPT/models/evaluator.py:141
Methodkinematic_tree
(self)
H-GPT/hGPT/data/motionx/common/skeleton.py:71
Functionl2_norm
(x1, x2, dim)
H-GPT/hGPT/metrics/utils.py:8
Functionlengths_to_mask
( lengths: List[int], device: torch.device, max_len: int = None )
H-GPT/hGPT/utils/temos_utils.py:10
Functionlerp
(p0, p1, t)
H-GPT/hGPT/data/motionx/common/quaternion.py:442
Functionload_example_input
(txt_path)
H-GPT/hGPT/utils/demo_utils.py:6
Functionload_pretrained
(cfg, model, logger=None, phase="train")
H-GPT/hGPT/utils/load_checkpoint.py:6
Functionload_pretrained_lora_and_merge
(cfg, model, logger=None, phase="test")
H-GPT/hGPT/utils/load_checkpoint.py:29
Functionload_pretrained_vae
(cfg, model, logger=None)
H-GPT/hGPT/utils/load_checkpoint.py:75
Functionmatrix_of_angles
(cos, sin, inv=False, dim=2)
H-GPT/hGPT/utils/geometry_tools.py:37
Functionmatrix_of_angles
(cos, sin, inv=False, dim=2)
H-GPT/hGPT/utils/geometry_conver.py:101
Functionmatrix_to_axis_angle
Convert rotations given as rotation matrices to axis/angle. Args: matrix: Rotation matrices as tensor of shape (..., 3, 3). Ret
H-GPT/hGPT/utils/rotation_conversions.py:432
Functionmatrix_to_euler_angles
Convert rotations given as rotation matrices to Euler angles in radians. Args: matrix: Rotation matrices as tensor of shape (..., 3,
H-GPT/hGPT/utils/geometry_tools.py:231
Functionmatrix_to_euler_angles
Convert rotations given as rotation matrices to Euler angles in radians. Args: matrix: Rotation matrices as tensor of shape (..., 3,
H-GPT/hGPT/utils/rotation_conversions.py:216
Functionmatrix_to_rotation_6d
Converts rotation matrices to 6D rotation representation by Zhou et al. [1] by dropping the last row. Note that 6D representation is not uniq
H-GPT/hGPT/utils/rotation_conversions.py:535
Functionmatrot2axisangle
:param matrots: N*num_joints*9 :return: N*num_joints*3
H-GPT/hGPT/utils/geometry_conver.py:115
Functionmld_collate
(batch)
H-GPT/hGPT/data/motionx/dataset.py:77
Methodmm_mode
(self, mm_on=True)
H-GPT/hGPT/data/MotionX.py:191
Functionmotionx_collate
(batch)
H-GPT/hGPT/data/MotionX.py:21
Functionnfeats_of
(rottype)
H-GPT/hGPT/utils/easyconvert.py:16
Functionnfeats_of
(rottype)
H-GPT/hGPT/utils/temos_utils.py:29
Methodnjoints
(self)
H-GPT/hGPT/data/motionx/common/skeleton.py:62
Methodoffset
(self)
H-GPT/hGPT/data/motionx/common/skeleton.py:65
Methodon_test_epoch_end
(self)
H-GPT/hGPT/models/base.py:81
Methodon_train_end
( self, trainer: Trainer, pl_module: LightningModule, **kwargs )
H-GPT/hGPT/callback.py:38
Methodon_train_epoch_end
( self, trainer: Trainer, pl_module: LightningModule, padding=False, **kwargs )
H-GPT/hGPT/callback.py:49
← previousnext →301–400 of 453, ranked by callers