MCPcopy Create free account

hub / github.com/chengxuxin/expressive-humanoid / functions

Functions1,218 in github.com/chengxuxin/expressive-humanoid

↓ 105 callersMethodnumpy
(self)
ASE/ase/utils/motion_lib.py:48
↓ 77 callersMethodformat
(self, record)
ASE/ase/utils/logger.py:25
↓ 75 callersMethodclone
(self)
ASE/ase/poselib/poselib/skeleton/skeleton3d.py:1038
↓ 75 callersFunctionquat_rotate
Rotate a 3D vector with the 3D rotation
ASE/ase/poselib/poselib/core/rotation3d.py:233
↓ 55 callersFunctionquat_from_angle_axis
Create a 3D rotation from angle and axis of rotation. The rotation is counter-clockwise along the axis. The rotation can be interpreted as
ASE/ase/poselib/poselib/core/rotation3d.py:147
↓ 36 callersMethodupdate
Update the plotter for any updates in the task primitives
ASE/ase/poselib/poselib/visualization/core.py:91
↓ 33 callersFunctionquat_mul
quaternion multiplication
ASE/ase/poselib/poselib/core/rotation3d.py:37
↓ 25 callersMethodfrom_rotation_and_root_translation
Construct a skeleton state from rotation and root translation :param skeleton_tree: the skeleton tree :type skeleton_tree: S
ASE/ase/poselib/poselib/skeleton/skeleton3d.py:652
↓ 23 callersMethodstep
(self, actions: torch.Tensor)
rsl_rl/rsl_rl/env/vec_env.py:50
↓ 20 callersMethodsave
(self, path, infos=None)
rsl_rl/rsl_rl/runners/on_policy_runner.py:529
↓ 18 callersMethodload
(self, params)
ASE/ase/learning/amp_network_builder.py:60
↓ 17 callersMethodindex
get the index of the node :param node_name: the name of the node :type node_name: string :rtype: int
ASE/ase/poselib/poselib/skeleton/skeleton3d.py:231
↓ 16 callersMethodadd_roughness
(self, terrain, difficulty=1)
legged_gym/legged_gym/utils/terrain.py:147
↓ 16 callersMethodfrom_file
Read the object from a file (either .npy or .json) :param path: path of the file :type path: string :param args, kwargs: the
ASE/ase/poselib/poselib/core/backend/abstract.py:116
↓ 16 callersMethodstep
(self, actions)
ASE/ase/env/tasks/vec_task.py:37
↓ 15 callersMethodnum_motions
(self)
ASE/ase/utils/motion_lib.py:140
↓ 15 callersFunctionquat_identity
Construct 3D identity rotation given shape
ASE/ase/poselib/poselib/core/rotation3d.py:136
↓ 14 callersMethodcreate
Create more task primitives from a task for the plotter
ASE/ase/poselib/poselib/visualization/core.py:85
↓ 14 callersMethodfrom_skeleton_state
Construct a skeleton motion from a skeleton state. The velocities are estimated using second order gaussian filter along the last axi
ASE/ase/poselib/poselib/skeleton/skeleton3d.py:1125
↓ 14 callersMethodget_motion_state
(self, motion_ids, motion_times, get_lbp=False)
ASE/ase/utils/motion_lib.py:199
↓ 13 callersFunctionplot_skeleton_motion_interactive
Visualize a skeleton motion along its first dimension interactively. :param skeleton_motion: :param task_name: :type skeleton_motion
ASE/ase/poselib/poselib/visualization/common.py:189
↓ 13 callersFunctionplot_skeleton_state
Visualize a skeleton state :param skeleton_state: :param task_name: :type skeleton_state: SkeletonState :type task_name: string,
ASE/ase/poselib/poselib/visualization/common.py:36
↓ 13 callersMethodto_file
Write the object to a file (either .npy or .json) :param path: path of the file :type path: string
ASE/ase/poselib/poselib/core/backend/abstract.py:136
↓ 13 callersMethodtrain
(self)
ASE/ase/learning/common_agent.py:110
↓ 12 callersMethodclear
(self)
rsl_rl/rsl_rl/storage/rollout_storage.py:121
↓ 10 callersMethodreindex
(self, vec)
legged_gym/legged_gym/envs/base/legged_robot.py:287
↓ 9 callersMethod_preproc_amp_obs
(self, amp_obs)
ASE/ase/learning/amp_agent.py:563
↓ 9 callersMethodnormalize
(self, input)
rsl_rl/rsl_rl/utils/utils.py:118
↓ 9 callersMethodregister
(self, name: str, task_class: VecEnv, env_cfg: LeggedRobotCfg, train_cfg: LeggedRobotCfgPPO)
legged_gym/legged_gym/utils/task_registry.py:51
↓ 8 callersMethodfrom_dict
(cls, dict_repr, *args, **kwargs)
ASE/ase/poselib/poselib/skeleton/skeleton3d.py:158
↓ 8 callersMethodlog
(self, locs, width=80, pad=35)
rsl_rl/rsl_rl/runners/on_policy_runner.py:424
↓ 8 callersFunctionquat_inverse
The inverse of the rotation
ASE/ase/poselib/poselib/core/rotation3d.py:242
↓ 8 callersFunctionquat_mul_norm
Combine two set of 3D rotations together using \**\* operator. The shape needs to be broadcastable
ASE/ase/poselib/poselib/core/rotation3d.py:224
↓ 8 callersFunctiontensor_to_dict
Construct an ordered dictionary from the object :rtype: OrderedDict
ASE/ase/poselib/poselib/core/tensor_utils.py:32
↓ 8 callersFunctiontransform_translation
Get translation from transform
ASE/ase/poselib/poselib/core/rotation3d.py:325
↓ 8 callersMethodupdate
(self, skeleton_state)
ASE/ase/poselib/poselib/visualization/skeleton_plotter_tasks.py:65
↓ 7 callersMethodload
(self, path, load_optimizer=True)
rsl_rl/rsl_rl/runners/on_policy_runner.py:542
↓ 7 callersMethodsample_time
(self, motion_ids, truncate_time=None)
ASE/ase/utils/motion_lib.py:165
↓ 7 callersFunctiontransform_from_rotation_translation
Construct a transform from a quaternion and 3D translation. Only one of them can be None.
ASE/ase/poselib/poselib/core/rotation3d.py:292
↓ 7 callersFunctiontransform_rotation
Get rotation from transform
ASE/ase/poselib/poselib/core/rotation3d.py:319
↓ 6 callersMethod_gather_cur_goals
(self, future=0)
legged_gym/legged_gym/envs/base/legged_robot.py:563
↓ 6 callersMethodclear
(self)
legged_gym/legged_gym/utils/storage.py:24
↓ 6 callersMethodevaluate
(self, critic_observations, **kwargs)
rsl_rl/rsl_rl/modules/actor_critic.py:300
↓ 6 callersFunctionget_activation
(act_name)
rsl_rl/rsl_rl/modules/actor_critic.py:308
↓ 6 callersMethodget_buffer_size
(self)
ASE/ase/learning/replay_buffer.py:49
↓ 6 callersMethodget_observations
(self)
rsl_rl/rsl_rl/env/vec_env.py:56
↓ 6 callersMethodget_scoped_name
(self, name)
ASE/ase/poselib/poselib/visualization/core.py:52
↓ 6 callersMethodinfer_hist_latent
(self, obs)
rsl_rl/rsl_rl/modules/actor_critic.py:201
↓ 6 callersMethodinfer_priv_latent
(self, obs)
rsl_rl/rsl_rl/modules/actor_critic.py:197
↓ 6 callersFunctionquat_normalize
Construct 3D rotation from quaternion (the quaternion needs not to be normalized).
ASE/ase/poselib/poselib/core/rotation3d.py:117
↓ 6 callersMethodreindex_dof_pos_vel
(self, dof_pos, dof_vel)
legged_gym/legged_gym/envs/h1/h1_amp.py:303
↓ 5 callersMethod_calc_disc_rewards
(self, amp_obs)
ASE/ase/learning/amp_agent.py:598
↓ 5 callersMethodact
(self, obs, critic_obs, info, hist_encoding=False)
rsl_rl/rsl_rl/algorithms/ppo.py:175
↓ 5 callersMethodenv_reset
(self, env_ids=None)
ASE/ase/learning/ase_agent.py:338
↓ 5 callersFunctioneuler_from_quaternion
Convert a quaternion into euler angles (roll, pitch, yaw) roll is rotation around x in radians (counterclockwise) pitch is ro
legged_gym/legged_gym/envs/base/legged_robot.py:56
↓ 5 callersMethodeval_disc
(self, amp_obs)
ASE/ase/learning/amp_network_builder.py:109
↓ 5 callersMethodfrom_fbx
Construct a skeleton motion from a fbx file (TODO - generalize this). If the skeleton tree is not given, it will use the first frame
ASE/ase/poselib/poselib/skeleton/skeleton3d.py:1199
↓ 5 callersMethodinsert
Add new states to memory.
rsl_rl/rsl_rl/storage/replay_buffer.py:20
↓ 5 callersMethodrender
(self, sync_frame_time=False)
ASE/ase/env/tasks/humanoid.py:468
↓ 5 callersMethodretarget_to_by_tpose
Retarget the skeleton state to a target skeleton tree. This is a naive retarget implementation with rough approximations. See the me
ASE/ase/poselib/poselib/skeleton/skeleton3d.py:985
↓ 5 callersMethodsample
(self, n)
ASE/ase/learning/replay_buffer.py:79
↓ 5 callersMethodsample_motions
(self, n, max_difficulty=None)
ASE/ase/utils/motion_lib.py:149
↓ 5 callersMethodsave
(self, path, infos=None)
rsl_rl/rsl_rl/runners/on_policy_runner_mimic_amp.py:539
↓ 5 callersMethodsave
(self, path, infos=None)
rsl_rl/rsl_rl/runners/on_policy_runner_mimic.py:488
↓ 5 callersMethodset_camera
Set camera position and direction
legged_gym/legged_gym/envs/base/legged_robot.py:466
↓ 5 callersMethodset_eval
(self)
ASE/ase/learning/amp_agent.py:63
↓ 5 callersMethodset_train
(self)
ASE/ase/learning/amp_agent.py:69
↓ 5 callersMethodupdate
(self)
rsl_rl/rsl_rl/algorithms/ppo.py:242
↓ 4 callersMethod_eval_disc
(self, amp_obs)
ASE/ase/learning/amp_agent.py:575
↓ 4 callersMethod_to_state_vector
(rot, rt)
ASE/ase/poselib/poselib/skeleton/skeleton3d.py:620
↓ 4 callersFunctionbuild_amp_observations
(root_pos, root_rot, root_vel, root_ang_vel, dof_pos, dof_vel, key_body_pos, local
ASE/ase/env/tasks/humanoid_amp.py:311
↓ 4 callersFunctionclass_to_dict
(obj)
legged_gym/legged_gym/utils/helpers.py:41
↓ 4 callersMethoddiscount_values
(self, mb_fdones, mb_values, mb_rewards, mb_next_values)
ASE/ase/learning/common_agent.py:465
↓ 4 callersMethodenv_step
(self, actions)
ASE/ase/learning/hrl_agent.py:73
↓ 4 callersMethodeval_actor
(self, obs)
ASE/ase/learning/amp_network_builder.py:79
↓ 4 callersMethodfeed_forward_generator
(self, num_mini_batch, mini_batch_size)
rsl_rl/rsl_rl/storage/replay_buffer.py:35
↓ 4 callersMethodget_actions_log_prob
(self, actions)
rsl_rl/rsl_rl/modules/actor_critic.py:289
↓ 4 callersFunctionget_args
()
legged_gym/legged_gym/utils/helpers.py:196
↓ 4 callersMethodget_motion_files
(self, motion_ids)
ASE/ase/utils/motion_lib.py:180
↓ 4 callersMethodget_motion_length
(self, motion_ids)
ASE/ase/utils/motion_lib.py:183
↓ 4 callersMethodget_num_amp_obs
(self)
ASE/ase/env/tasks/humanoid_amp.py:89
↓ 4 callersMethodlookat
(self, i)
legged_gym/legged_gym/envs/base/base_task.py:150
↓ 4 callersMethodmake_terrain
(self, choice, difficulty)
legged_gym/legged_gym/utils/terrain.py:152
↓ 4 callersMethodmini_batch_generator
(self, num_mini_batches, num_epochs=8)
rsl_rl/rsl_rl/storage/rollout_storage.py:148
↓ 4 callersMethodprocess_infos
(self, infos, done_indices)
ASE/ase/run.py:105
↓ 4 callersMethodreccurent_mini_batch_generator
(self, num_mini_batches, num_epochs=8)
rsl_rl/rsl_rl/storage/rollout_storage.py:196
↓ 4 callersMethodrender
(self, sync_frame_time=True)
legged_gym/legged_gym/envs/base/base_task.py:155
↓ 4 callersMethodshow
(self)
ASE/ase/poselib/poselib/visualization/plt_plotter.py:80
↓ 3 callersMethod__init__
(self, num_prop, num_demo, text_feat_input_dim, text_feat_
rsl_rl/rsl_rl/modules/actor_critic_mimic.py:89
↓ 3 callersMethod__init__
(self, input_dim, output_dim, hidden_dims=[256, 128, 64],
rsl_rl/rsl_rl/modules/estimator.py:13
↓ 3 callersMethod__init__
(self, obs_size, ase_latent_size, units, activation, initializer)
ASE/ase/learning/ase_network_builder.py:261
↓ 3 callersMethod_actor_loss
(self, old_action_log_probs_batch, action_log_probs, advantage, curr_e_clip)
ASE/ase/learning/common_agent.py:533
↓ 3 callersMethod_compute_torques
(self, actions)
legged_gym/legged_gym/envs/h1/h1_mimic_eval.py:62
↓ 3 callersMethod_critic_loss
(self, value_preds_batch, values, curr_e_clip, return_batch, clip_value)
ASE/ase/learning/common_agent.py:549
↓ 3 callersMethod_enable_enc_grad_penalty
(self)
ASE/ase/learning/ase_agent.py:502
↓ 3 callersMethod_eval_critic
(self, obs_dict, ase_latents)
ASE/ase/learning/ase_agent.py:413
↓ 3 callersMethod_indent
(self, s)
ASE/ase/poselib/poselib/skeleton/skeleton3d.py:137
↓ 3 callersMethod_max
(self, x, y)
ASE/ase/poselib/poselib/visualization/plt_plotter.py:274
↓ 3 callersMethod_min
(self, x, y)
ASE/ase/poselib/poselib/visualization/plt_plotter.py:267
↓ 3 callersMethod_prepare_reward_function
Prepares a list of reward functions, whcih will be called to compute the total reward. Looks for self._reward_<REWARD_NAME>, where <REWAR
legged_gym/legged_gym/envs/base/legged_robot.py:826
next →1–100 of 1,218, ranked by callers