Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/amazon-far/ResMimic
/ functions
Functions
982 in github.com/amazon-far/ResMimic
⨍
Functions
982
◇
Types & classes
215
↓ 53 callers
Method
format
(self, record)
pose/pose/utils/logger.py:25
↓ 50 callers
Method
numpy
(self)
pose/pose/utils/motion_lib.py:56
↓ 41 callers
Method
clone
(self)
pose/pose/poselib/poselib/skeleton/skeleton3d.py:1036
↓ 30 callers
Function
quat_rotate_inverse
(q, v)
pose/pose/utils/isaacgym_torch_utils.py:72
↓ 22 callers
Function
torch_rand_float
(lower, upper, shape, device)
pose/pose/utils/isaacgym_torch_utils.py:194
↓ 21 callers
Method
reindex
(self, vec)
legged_gym/legged_gym/envs/base/legged_robot.py:206
↓ 19 callers
Method
load
(self, path, load_optimizer=True)
rsl_rl/rsl_rl/runners/on_policy_dagger_runner.py:503
↓ 16 callers
Method
add_roughness
(self, terrain, difficulty=1)
legged_gym/legged_gym/gym_utils/terrain.py:149
↓ 14 callers
Function
quat_from_euler_xyz
(roll, pitch, yaw)
pose/pose/utils/isaacgym_torch_utils.py:177
↓ 13 callers
Method
clear
(self)
rsl_rl/rsl_rl/storage/rollout_storage.py:121
↓ 13 callers
Function
euler_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 callers
Method
from_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 callers
Function
convert_to_local_root_body_pos
(root_rot, body_pos)
legged_gym/legged_gym/envs/base/humanoid_char.py:524
↓ 12 callers
Method
slerp
(self, val0, val1, blend)
rsl_rl/rsl_rl/datasets/motion_loader.py:188
↓ 10 callers
Method
log
(self, locs, width=80, pad=35)
rsl_rl/rsl_rl/runners/on_policy_dagger_runner.py:370
↓ 9 callers
Function
convert_to_global_root_body_pos
(root_pos, root_rot, body_pos)
legged_gym/legged_gym/envs/base/humanoid_char.py:510
↓ 9 callers
Method
evaluate
(self, critic_observations, **kwargs)
rsl_rl/rsl_rl/modules/actor_critic.py:178
↓ 9 callers
Method
num_motions
(self)
pose/pose/utils/motion_lib.py:150
↓ 9 callers
Function
quat_rotate
Rotate a 3D vector with the 3D rotation
pose/pose/poselib/poselib/core/rotation3d.py:233
↓ 9 callers
Method
save
(self, path, infos=None)
rsl_rl/rsl_rl/runners/on_policy_dagger_runner.py:477
↓ 8 callers
Method
from_dict
(cls, dict_repr, *args, **kwargs)
pose/pose/poselib/poselib/skeleton/skeleton3d.py:158
↓ 8 callers
Method
normalize
(self, input)
rsl_rl/rsl_rl/utils/utils.py:120
↓ 8 callers
Function
quat_inverse
The inverse of the rotation
pose/pose/poselib/poselib/core/rotation3d.py:242
↓ 8 callers
Function
quat_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 callers
Function
tensor_to_dict
Construct an ordered dictionary from the object :rtype: OrderedDict
pose/pose/poselib/poselib/core/tensor_utils.py:32
↓ 8 callers
Function
transform_translation
Get translation from transform
pose/pose/poselib/poselib/core/rotation3d.py:325
↓ 8 callers
Method
update
(self, skeleton_state)
pose/pose/poselib/poselib/visualization/skeleton_plotter_tasks.py:65
↓ 7 callers
Method
calc_motion_frame
(self, motion_ids, motion_times)
pose/pose/utils/motion_lib_pkl.py:326
↓ 7 callers
Function
quat_apply
(a, b)
pose/pose/utils/isaacgym_torch_utils.py:49
↓ 7 callers
Method
step
(self, actions: torch.Tensor)
rsl_rl/rsl_rl/env/vec_env.py:50
↓ 7 callers
Function
transform_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 callers
Function
transform_rotation
Get rotation from transform
pose/pose/poselib/poselib/core/rotation3d.py:319
↓ 6 callers
Method
_get_motion_times
(self, env_ids=None)
legged_gym/legged_gym/envs/base/humanoid_mimic.py:151
↓ 6 callers
Method
get_actions_log_prob
(self, actions)
rsl_rl/rsl_rl/modules/actor_critic.py:167
↓ 6 callers
Method
get_motion_length
(self, motion_ids)
pose/pose/utils/motion_lib.py:193
↓ 6 callers
Method
get_scoped_name
(self, name)
pose/pose/poselib/poselib/visualization/core.py:52
↓ 6 callers
Method
normalize_torch
(self, input, device)
rsl_rl/rsl_rl/utils/utils.py:125
↓ 6 callers
Function
quat_mul
(a, b)
pose/pose/utils/isaacgym_torch_utils.py:20
↓ 6 callers
Function
quat_normalize
Construct 3D rotation from quaternion (the quaternion needs not to be normalized).
pose/pose/poselib/poselib/core/rotation3d.py:117
↓ 6 callers
Function
quat_rotate
(q, v)
pose/pose/utils/isaacgym_torch_utils.py:59
↓ 5 callers
Function
class_to_dict
(obj)
legged_gym/legged_gym/gym_utils/helpers.py:41
↓ 5 callers
Function
get_activation
(act_name)
rsl_rl/rsl_rl/modules/actor_critic.py:189
↓ 5 callers
Function
get_args
()
legged_gym/legged_gym/gym_utils/helpers.py:263
↓ 5 callers
Method
index
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 callers
Method
load_state_dict
(self, state_dict)
rsl_rl/rsl_rl/utils/scale.py:15
↓ 5 callers
Method
mini_batch_generator
(self, num_mini_batches, num_epochs=8)
rsl_rl/rsl_rl/storage/rollout_storage.py:148
↓ 5 callers
Function
quat_conjugate
(a)
pose/pose/utils/isaacgym_torch_utils.py:85
↓ 5 callers
Function
quat_identity
Construct 3D identity rotation given shape
pose/pose/poselib/poselib/core/rotation3d.py:136
↓ 5 callers
Function
quat_to_exp_map
(q)
pose/pose/utils/torch_utils.py:90
↓ 5 callers
Function
set_nested_attr
Set nested attribute using dot notation (e.g., 'env.num_envs')
legged_gym/legged_gym/gym_utils/helpers.py:58
↓ 5 callers
Function
to_torch
(x, dtype=torch.float, device='cuda:0', requires_grad=False)
pose/pose/utils/isaacgym_torch_utils.py:15
↓ 5 callers
Method
update
Update the plotter for any updates in the task primitives
pose/pose/poselib/poselib/visualization/core.py:91
↓ 4 callers
Method
__init__
(self, num_prop, num_actions, obs_context_len, priv_encode
rsl_rl/rsl_rl/modules/actor_critic_transformer.py:158
↓ 4 callers
Method
__init__
(self, cfg)
rsl_rl/rsl_rl/utils/layers.py:71
↓ 4 callers
Method
_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 callers
Method
_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 callers
Method
_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 callers
Method
_to_state_vector
(rot, rt)
pose/pose/poselib/poselib/skeleton/skeleton3d.py:620
↓ 4 callers
Method
act
(self, obs, critic_obs, info, hist_encoding=False)
rsl_rl/rsl_rl/algorithms/ppo.py:111
↓ 4 callers
Method
add_transitions
(self, transition: Transition)
rsl_rl/rsl_rl/storage/rollout_storage.py:88
↓ 4 callers
Method
from_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 callers
Method
get_cfgs
(self, name)
legged_gym/legged_gym/gym_utils/task_registry.py:59
↓ 4 callers
Method
lookat
(self, i)
legged_gym/legged_gym/envs/base/base_task.py:149
↓ 4 callers
Method
make_terrain
(self, choice, difficulty)
legged_gym/legged_gym/gym_utils/terrain.py:154
↓ 4 callers
Method
reccurent_mini_batch_generator
(self, num_mini_batches, num_epochs=8)
rsl_rl/rsl_rl/storage/rollout_storage.py:196
↓ 4 callers
Function
reduce_all
(x, op)
rsl_rl/rsl_rl/utils/utils.py:224
↓ 4 callers
Function
reduce_inplace_all
(x, op)
rsl_rl/rsl_rl/utils/utils.py:263
↓ 4 callers
Method
set_camera
Set camera position and direction
legged_gym/legged_gym/envs/base/legged_robot.py:371
↓ 4 callers
Method
state_dict
(self)
rsl_rl/rsl_rl/utils/scale.py:12
↓ 4 callers
Method
step
(self, actions)
legged_gym/legged_gym/envs/g1/g1_hoi.py:491
↓ 3 callers
Method
__init__
observation: motion_obs, cur_priop_obs, history_obs
rsl_rl/rsl_rl/modules/actor_critic_teleop.py:125
↓ 3 callers
Method
__init__
observation: motion_obs, cur_priop_obs, history_obs
rsl_rl/rsl_rl/modules/actor_critic_tracking.py:125
↓ 3 callers
Method
_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 callers
Method
_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 callers
Method
_indent
(self, s)
pose/pose/poselib/poselib/skeleton/skeleton3d.py:137
↓ 3 callers
Method
_max
(self, x, y)
pose/pose/poselib/poselib/visualization/plt_plotter.py:274
↓ 3 callers
Method
_min
(self, x, y)
pose/pose/poselib/poselib/visualization/plt_plotter.py:267
↓ 3 callers
Method
_transfer_to
(self, new_skeleton_tree: SkeletonTree)
pose/pose/poselib/poselib/skeleton/skeleton3d.py:734
↓ 3 callers
Method
_update_lim
(self, xs, ys)
pose/pose/poselib/poselib/visualization/plt_plotter.py:103
↓ 3 callers
Method
_update_lim
(self, xs, ys, zs)
pose/pose/poselib/poselib/visualization/plt_plotter.py:289
↓ 3 callers
Method
add_terrain_to_map
(self, terrain, row, col)
legged_gym/legged_gym/gym_utils/terrain.py:341
↓ 3 callers
Method
create_sim
Creates simulation, terrain and evironments
legged_gym/legged_gym/envs/base/legged_robot.py:348
↓ 3 callers
Method
curiculum
(self, random=False, max_difficulty=False)
legged_gym/legged_gym/gym_utils/terrain.py:119
↓ 3 callers
Method
dim
(self)
pose/pose/poselib/poselib/visualization/simple_plotter_tasks.py:162
↓ 3 callers
Function
enable_mp
()
rsl_rl/rsl_rl/utils/utils.py:191
↓ 3 callers
Method
get_frame_at_time
Returns frame for the given trajectory at the specified time.
rsl_rl/rsl_rl/datasets/motion_loader.py:195
↓ 3 callers
Method
get_full_frame_at_time_batch
(self, traj_idxs, times)
rsl_rl/rsl_rl/datasets/motion_loader.py:230
↓ 3 callers
Method
get_hidden_states
(self)
rsl_rl/rsl_rl/modules/actor_critic_recurrent.py:89
↓ 3 callers
Function
get_num_procs
()
rsl_rl/rsl_rl/utils/utils.py:173
↓ 3 callers
Function
get_rand_dis_int
(scale)
legged_gym/legged_gym/gym_utils/terrain.py:898
↓ 3 callers
Method
get_root_pos_batch
(poses)
rsl_rl/rsl_rl/datasets/motion_loader.py:354
↓ 3 callers
Method
get_root_rot
(pose)
rsl_rl/rsl_rl/datasets/motion_loader.py:357
↓ 3 callers
Method
get_root_rot_batch
(poses)
rsl_rl/rsl_rl/datasets/motion_loader.py:360
↓ 3 callers
Function
half_sloped_terrain
(terrain, wall_width=4, start2center=0.7, max_height=1)
legged_gym/legged_gym/gym_utils/terrain.py:859
↓ 3 callers
Method
if_fix_std
(self)
rsl_rl/rsl_rl/modules/actor_critic.py:186
↓ 3 callers
Method
make_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 callers
Method
make_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 callers
Function
normalize
(x, eps: float = 1e-9)
pose/pose/utils/isaacgym_torch_utils.py:44
↓ 3 callers
Function
normalize_angle
(x)
pose/pose/utils/isaacgym_torch_utils.py:105
↓ 3 callers
Function
quat_apply_yaw
(quat, vec)
legged_gym/legged_gym/gym_utils/math.py:38
next →
1–100 of 982, ranked by callers