MCPcopy Create free account

hub / github.com/amazon-far/ResMimic / functions

Functions982 in github.com/amazon-far/ResMimic

↓ 53 callersMethodformat
(self, record)
pose/pose/utils/logger.py:25
↓ 50 callersMethodnumpy
(self)
pose/pose/utils/motion_lib.py:56
↓ 41 callersMethodclone
(self)
pose/pose/poselib/poselib/skeleton/skeleton3d.py:1036
↓ 30 callersFunctionquat_rotate_inverse
(q, v)
pose/pose/utils/isaacgym_torch_utils.py:72
↓ 22 callersFunctiontorch_rand_float
(lower, upper, shape, device)
pose/pose/utils/isaacgym_torch_utils.py:194
↓ 21 callersMethodreindex
(self, vec)
legged_gym/legged_gym/envs/base/legged_robot.py:206
↓ 19 callersMethodload
(self, path, load_optimizer=True)
rsl_rl/rsl_rl/runners/on_policy_dagger_runner.py:503
↓ 16 callersMethodadd_roughness
(self, terrain, difficulty=1)
legged_gym/legged_gym/gym_utils/terrain.py:149
↓ 14 callersFunctionquat_from_euler_xyz
(roll, pitch, yaw)
pose/pose/utils/isaacgym_torch_utils.py:177
↓ 13 callersMethodclear
(self)
rsl_rl/rsl_rl/storage/rollout_storage.py:121
↓ 13 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
↓ 13 callersMethodfrom_rotation_and_root_translation
Construct a skeleton state from rotation and root translation :param skeleton_tree: the skeleton tree :type skeleton_tree: S
pose/pose/poselib/poselib/skeleton/skeleton3d.py:652
↓ 12 callersFunctionconvert_to_local_root_body_pos
(root_rot, body_pos)
legged_gym/legged_gym/envs/base/humanoid_char.py:524
↓ 12 callersMethodslerp
(self, val0, val1, blend)
rsl_rl/rsl_rl/datasets/motion_loader.py:188
↓ 10 callersMethodlog
(self, locs, width=80, pad=35)
rsl_rl/rsl_rl/runners/on_policy_dagger_runner.py:370
↓ 9 callersFunctionconvert_to_global_root_body_pos
(root_pos, root_rot, body_pos)
legged_gym/legged_gym/envs/base/humanoid_char.py:510
↓ 9 callersMethodevaluate
(self, critic_observations, **kwargs)
rsl_rl/rsl_rl/modules/actor_critic.py:178
↓ 9 callersMethodnum_motions
(self)
pose/pose/utils/motion_lib.py:150
↓ 9 callersFunctionquat_rotate
Rotate a 3D vector with the 3D rotation
pose/pose/poselib/poselib/core/rotation3d.py:233
↓ 9 callersMethodsave
(self, path, infos=None)
rsl_rl/rsl_rl/runners/on_policy_dagger_runner.py:477
↓ 8 callersMethodfrom_dict
(cls, dict_repr, *args, **kwargs)
pose/pose/poselib/poselib/skeleton/skeleton3d.py:158
↓ 8 callersMethodnormalize
(self, input)
rsl_rl/rsl_rl/utils/utils.py:120
↓ 8 callersFunctionquat_inverse
The inverse of the rotation
pose/pose/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
pose/pose/poselib/poselib/core/rotation3d.py:224
↓ 8 callersFunctiontensor_to_dict
Construct an ordered dictionary from the object :rtype: OrderedDict
pose/pose/poselib/poselib/core/tensor_utils.py:32
↓ 8 callersFunctiontransform_translation
Get translation from transform
pose/pose/poselib/poselib/core/rotation3d.py:325
↓ 8 callersMethodupdate
(self, skeleton_state)
pose/pose/poselib/poselib/visualization/skeleton_plotter_tasks.py:65
↓ 7 callersMethodcalc_motion_frame
(self, motion_ids, motion_times)
pose/pose/utils/motion_lib_pkl.py:326
↓ 7 callersFunctionquat_apply
(a, b)
pose/pose/utils/isaacgym_torch_utils.py:49
↓ 7 callersMethodstep
(self, actions: torch.Tensor)
rsl_rl/rsl_rl/env/vec_env.py:50
↓ 7 callersFunctiontransform_from_rotation_translation
Construct a transform from a quaternion and 3D translation. Only one of them can be None.
pose/pose/poselib/poselib/core/rotation3d.py:292
↓ 7 callersFunctiontransform_rotation
Get rotation from transform
pose/pose/poselib/poselib/core/rotation3d.py:319
↓ 6 callersMethod_get_motion_times
(self, env_ids=None)
legged_gym/legged_gym/envs/base/humanoid_mimic.py:151
↓ 6 callersMethodget_actions_log_prob
(self, actions)
rsl_rl/rsl_rl/modules/actor_critic.py:167
↓ 6 callersMethodget_motion_length
(self, motion_ids)
pose/pose/utils/motion_lib.py:193
↓ 6 callersMethodget_scoped_name
(self, name)
pose/pose/poselib/poselib/visualization/core.py:52
↓ 6 callersMethodnormalize_torch
(self, input, device)
rsl_rl/rsl_rl/utils/utils.py:125
↓ 6 callersFunctionquat_mul
(a, b)
pose/pose/utils/isaacgym_torch_utils.py:20
↓ 6 callersFunctionquat_normalize
Construct 3D rotation from quaternion (the quaternion needs not to be normalized).
pose/pose/poselib/poselib/core/rotation3d.py:117
↓ 6 callersFunctionquat_rotate
(q, v)
pose/pose/utils/isaacgym_torch_utils.py:59
↓ 5 callersFunctionclass_to_dict
(obj)
legged_gym/legged_gym/gym_utils/helpers.py:41
↓ 5 callersFunctionget_activation
(act_name)
rsl_rl/rsl_rl/modules/actor_critic.py:189
↓ 5 callersFunctionget_args
()
legged_gym/legged_gym/gym_utils/helpers.py:263
↓ 5 callersMethodindex
get the index of the node :param node_name: the name of the node :type node_name: string :rtype: int
pose/pose/poselib/poselib/skeleton/skeleton3d.py:231
↓ 5 callersMethodload_state_dict
(self, state_dict)
rsl_rl/rsl_rl/utils/scale.py:15
↓ 5 callersMethodmini_batch_generator
(self, num_mini_batches, num_epochs=8)
rsl_rl/rsl_rl/storage/rollout_storage.py:148
↓ 5 callersFunctionquat_conjugate
(a)
pose/pose/utils/isaacgym_torch_utils.py:85
↓ 5 callersFunctionquat_identity
Construct 3D identity rotation given shape
pose/pose/poselib/poselib/core/rotation3d.py:136
↓ 5 callersFunctionquat_to_exp_map
(q)
pose/pose/utils/torch_utils.py:90
↓ 5 callersFunctionset_nested_attr
Set nested attribute using dot notation (e.g., 'env.num_envs')
legged_gym/legged_gym/gym_utils/helpers.py:58
↓ 5 callersFunctionto_torch
(x, dtype=torch.float, device='cuda:0', requires_grad=False)
pose/pose/utils/isaacgym_torch_utils.py:15
↓ 5 callersMethodupdate
Update the plotter for any updates in the task primitives
pose/pose/poselib/poselib/visualization/core.py:91
↓ 4 callersMethod__init__
(self, num_prop, num_actions, obs_context_len, priv_encode
rsl_rl/rsl_rl/modules/actor_critic_transformer.py:158
↓ 4 callersMethod__init__
(self, cfg)
rsl_rl/rsl_rl/utils/layers.py:71
↓ 4 callersMethod_get_mimic_obs
Override to use unified motion sampling for both privileged and future observations.
legged_gym/legged_gym/envs/g1/g1_mimic_future.py:355
↓ 4 callersMethod_push_robots
Random pushes the robots. Emulates an impulse by setting a randomized base velocity.
legged_gym/legged_gym/envs/base/legged_robot.py:568
↓ 4 callersMethod_resample_commands
Randommly select commands of some environments Args: env_ids (List[int]): Environments ids for which new commands are needed
legged_gym/legged_gym/envs/base/legged_robot.py:474
↓ 4 callersMethod_to_state_vector
(rot, rt)
pose/pose/poselib/poselib/skeleton/skeleton3d.py:620
↓ 4 callersMethodact
(self, obs, critic_obs, info, hist_encoding=False)
rsl_rl/rsl_rl/algorithms/ppo.py:111
↓ 4 callersMethodadd_transitions
(self, transition: Transition)
rsl_rl/rsl_rl/storage/rollout_storage.py:88
↓ 4 callersMethodfrom_skeleton_state
Construct a skeleton motion from a skeleton state. The velocities are estimated using second order gaussian filter along the last axi
pose/pose/poselib/poselib/skeleton/skeleton3d.py:1123
↓ 4 callersMethodget_cfgs
(self, name)
legged_gym/legged_gym/gym_utils/task_registry.py:59
↓ 4 callersMethodlookat
(self, i)
legged_gym/legged_gym/envs/base/base_task.py:149
↓ 4 callersMethodmake_terrain
(self, choice, difficulty)
legged_gym/legged_gym/gym_utils/terrain.py:154
↓ 4 callersMethodreccurent_mini_batch_generator
(self, num_mini_batches, num_epochs=8)
rsl_rl/rsl_rl/storage/rollout_storage.py:196
↓ 4 callersFunctionreduce_all
(x, op)
rsl_rl/rsl_rl/utils/utils.py:224
↓ 4 callersFunctionreduce_inplace_all
(x, op)
rsl_rl/rsl_rl/utils/utils.py:263
↓ 4 callersMethodset_camera
Set camera position and direction
legged_gym/legged_gym/envs/base/legged_robot.py:371
↓ 4 callersMethodstate_dict
(self)
rsl_rl/rsl_rl/utils/scale.py:12
↓ 4 callersMethodstep
(self, actions)
legged_gym/legged_gym/envs/g1/g1_hoi.py:491
↓ 3 callersMethod__init__
observation: motion_obs, cur_priop_obs, history_obs
rsl_rl/rsl_rl/modules/actor_critic_teleop.py:125
↓ 3 callersMethod__init__
observation: motion_obs, cur_priop_obs, history_obs
rsl_rl/rsl_rl/modules/actor_critic_tracking.py:125
↓ 3 callersMethod_apply_motion_component_masking
Apply masking to motion state component and return mask indicator
legged_gym/legged_gym/envs/g1/g1_mimic_future.py:339
↓ 3 callersMethod_compute_torques
Compute torques from actions. Actions can be interpreted as position or velocity targets given to a PD controller, or directly as scaled
legged_gym/legged_gym/envs/base/legged_robot.py:487
↓ 3 callersMethod_indent
(self, s)
pose/pose/poselib/poselib/skeleton/skeleton3d.py:137
↓ 3 callersMethod_max
(self, x, y)
pose/pose/poselib/poselib/visualization/plt_plotter.py:274
↓ 3 callersMethod_min
(self, x, y)
pose/pose/poselib/poselib/visualization/plt_plotter.py:267
↓ 3 callersMethod_transfer_to
(self, new_skeleton_tree: SkeletonTree)
pose/pose/poselib/poselib/skeleton/skeleton3d.py:734
↓ 3 callersMethod_update_lim
(self, xs, ys)
pose/pose/poselib/poselib/visualization/plt_plotter.py:103
↓ 3 callersMethod_update_lim
(self, xs, ys, zs)
pose/pose/poselib/poselib/visualization/plt_plotter.py:289
↓ 3 callersMethodadd_terrain_to_map
(self, terrain, row, col)
legged_gym/legged_gym/gym_utils/terrain.py:341
↓ 3 callersMethodcreate_sim
Creates simulation, terrain and evironments
legged_gym/legged_gym/envs/base/legged_robot.py:348
↓ 3 callersMethodcuriculum
(self, random=False, max_difficulty=False)
legged_gym/legged_gym/gym_utils/terrain.py:119
↓ 3 callersMethoddim
(self)
pose/pose/poselib/poselib/visualization/simple_plotter_tasks.py:162
↓ 3 callersFunctionenable_mp
()
rsl_rl/rsl_rl/utils/utils.py:191
↓ 3 callersMethodget_frame_at_time
Returns frame for the given trajectory at the specified time.
rsl_rl/rsl_rl/datasets/motion_loader.py:195
↓ 3 callersMethodget_full_frame_at_time_batch
(self, traj_idxs, times)
rsl_rl/rsl_rl/datasets/motion_loader.py:230
↓ 3 callersMethodget_hidden_states
(self)
rsl_rl/rsl_rl/modules/actor_critic_recurrent.py:89
↓ 3 callersFunctionget_num_procs
()
rsl_rl/rsl_rl/utils/utils.py:173
↓ 3 callersFunctionget_rand_dis_int
(scale)
legged_gym/legged_gym/gym_utils/terrain.py:898
↓ 3 callersMethodget_root_pos_batch
(poses)
rsl_rl/rsl_rl/datasets/motion_loader.py:354
↓ 3 callersMethodget_root_rot
(pose)
rsl_rl/rsl_rl/datasets/motion_loader.py:357
↓ 3 callersMethodget_root_rot_batch
(poses)
rsl_rl/rsl_rl/datasets/motion_loader.py:360
↓ 3 callersFunctionhalf_sloped_terrain
(terrain, wall_width=4, start2center=0.7, max_height=1)
legged_gym/legged_gym/gym_utils/terrain.py:859
↓ 3 callersMethodif_fix_std
(self)
rsl_rl/rsl_rl/modules/actor_critic.py:186
↓ 3 callersMethodmake_alg_runner
Creates the training algorithm either from a registered namme or from the provided config file. Args: env (isaacgym.VecTaskPyth
legged_gym/legged_gym/gym_utils/task_registry.py:105
↓ 3 callersMethodmake_env
Creates an environment either from a registered namme or from the provided config file. Args: name (string): Name of a registere
legged_gym/legged_gym/gym_utils/task_registry.py:66
↓ 3 callersFunctionnormalize
(x, eps: float = 1e-9)
pose/pose/utils/isaacgym_torch_utils.py:44
↓ 3 callersFunctionnormalize_angle
(x)
pose/pose/utils/isaacgym_torch_utils.py:105
↓ 3 callersFunctionquat_apply_yaw
(quat, vec)
legged_gym/legged_gym/gym_utils/math.py:38
next →1–100 of 982, ranked by callers