MCPcopy Create free account

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

Functions982 in github.com/amazon-far/ResMimic

↓ 1 callersMethodupdate
(self, x)
rsl_rl/rsl_rl/utils/scale.py:37
↓ 1 callersMethodupdate
(self)
rsl_rl/rsl_rl/utils/normalizer.py:28
↓ 1 callersMethodupdate
(self, lines)
pose/pose/poselib/poselib/visualization/simple_plotter_tasks.py:89
↓ 1 callersMethodupdate
(self, dots)
pose/pose/poselib/poselib/visualization/simple_plotter_tasks.py:127
↓ 1 callersMethodupdate_counter
(self)
rsl_rl/rsl_rl/algorithms/dagger.py:138
↓ 1 callersMethodupdate_counter
(self)
rsl_rl/rsl_rl/algorithms/ppo.py:254
↓ 1 callersMethodupdate_counter
(self)
rsl_rl/rsl_rl/algorithms/ppo_amp.py:290
↓ 1 callersMethodupdate_distribution
(self, observations)
rsl_rl/rsl_rl/modules/actor_critic.py:150
↓ 1 callersMethodupdate_distribution
(self, observations)
rsl_rl/rsl_rl/modules/actor_critic_teleop.py:274
↓ 1 callersMethodupdate_distribution
(self, observations)
rsl_rl/rsl_rl/modules/actor_critic_mimic.py:212
↓ 1 callersMethodupdate_distribution
(self, observations)
rsl_rl/rsl_rl/modules/actor_critic_tracking.py:275
↓ 1 callersMethodupdate_distribution
(self, observations, hist_encoding)
rsl_rl/rsl_rl/modules/actor_critic_rma.py:229
↓ 1 callersMethodupdate_distribution
(self, observations)
rsl_rl/rsl_rl/modules/actor_critic_future.py:387
↓ 1 callersMethodupdate_distribution
(self, observations, hist_encoding)
rsl_rl/rsl_rl/modules/actor_critic_transformer.py:253
↓ 1 callersMethodupdate_from_moments
(self, batch_mean: np.ndarray, batch_var: np.ndarray, batch_count: int)
rsl_rl/rsl_rl/utils/utils.py:97
↓ 1 callersFunctionwrap_to_pi
(angles)
legged_gym/legged_gym/gym_utils/math.py:45
FunctionIsRotationMatrix
Returns true if the 3x3 submatrix represents a rotation. Args: m: A transformation matrix. Raises: ValueError: If input is not a matrix
rsl_rl/rsl_rl/datasets/pose3d.py:232
FunctionQuaternionNormalize
Normalizes the quaternion to length 1. Divides the quaternion by its magnitude. If the magnitude is too small, returns the quaternion identity v
rsl_rl/rsl_rl/datasets/pose3d.py:83
FunctionQuaternionRandomRotation
Creates a random small rotation around a random axis. Generates a small rotation with the axis vector selected uniformly from the unit sphere and
rsl_rl/rsl_rl/datasets/pose3d.py:188
FunctionQuaternionRotatePoint
Performs a rotation by quaternion. Rotate the point by the quaternion using quaternion multiplication, (q * p * q^-1), without constructing the r
rsl_rl/rsl_rl/datasets/pose3d.py:211
FunctionQuaternionToAxisAngle
Calculates axis and angle of rotation performed by a quaternion. Calculates the axis and angle of the rotation performed by the quaternion. The q
rsl_rl/rsl_rl/datasets/pose3d.py:137
FunctionVector3RandomNormal
Returns a random 3D vector from a normal distribution. Each component is selected independently from a normal distribution. Args: sigma: Sca
rsl_rl/rsl_rl/datasets/pose3d.py:32
FunctionVector3RandomUniform
Returns a 3D vector selected uniformly from the input box. Args: low: The min-value corner of the box. high: The max-value corner of the bo
rsl_rl/rsl_rl/datasets/pose3d.py:49
Method__call__
(self, x, update=False)
rsl_rl/rsl_rl/utils/scale.py:42
Method__del__
(self)
legged_gym/legged_gym/gym_utils/logger.py:135
Method__getattr__
(self, string)
pose/pose/utils/motion_lib.py:96
Method__getitem__
(self, index)
pose/pose/poselib/poselib/visualization/simple_plotter_tasks.py:94
Method__getitem__
(self, index)
pose/pose/poselib/poselib/visualization/simple_plotter_tasks.py:132
Method__getitem__
get the name of the node given the index
pose/pose/poselib/poselib/skeleton/skeleton3d.py:123
Method__init__
(self, dt)
legged_gym/legged_gym/gym_utils/logger.py:37
Method__init__
(self, actor_critic)
legged_gym/legged_gym/gym_utils/helpers.py:344
Method__init__
(self)
legged_gym/legged_gym/gym_utils/task_registry.py:46
Method__init__
(self, cfg: LeggedRobotCfg.terrain, num_robots)
legged_gym/legged_gym/gym_utils/terrain.py:46
Method__init__
(self, cfg: G1HOICfg, sim_params, physics_engine, sim_device, headless)
legged_gym/legged_gym/envs/g1/g1_hoi.py:22
Method__init__
(self, cfg: G1MimicPrivCfg, sim_params, physics_engine, sim_device, headless)
legged_gym/legged_gym/envs/g1/g1_mimic_distill.py:16
Method__init__
(self, cfg: G1MimicStuFutureCfg, sim_params, physics_engine, sim_device, headless)
legged_gym/legged_gym/envs/g1/g1_mimic_future.py:23
Method__init__
Parses the provided config file, calls create_sim() (which creates, simulation, terrain and environments), initilizes pytorch
legged_gym/legged_gym/envs/base/humanoid_char.py:19
Method__init__
Initializes all member classes recursively. Ignores all namse starting with '__' (buit-in methods).
legged_gym/legged_gym/envs/base/base_config.py:34
Method__init__
Parses the provided config file, calls create_sim() (which creates, simulation, terrain and environments), initilizes pytorch
legged_gym/legged_gym/envs/base/legged_robot.py:79
Method__init__
(self, cfg: HumanoidMimicCfg, sim_params, physics_engine, sim_device, headless)
legged_gym/legged_gym/envs/base/humanoid_mimic.py:24
Method__init__
(self, cfg, sim_params, physics_engine, sim_device, headless)
legged_gym/legged_gym/envs/base/base_task.py:41
Method__init__
(self, num_observations, num_motion_observations, num_pri
legged_gym/legged_gym/scripts/save_jit_stu_future.py:39
Method__init__
(self, num_observations, num_motion_observations, num_pri
legged_gym/legged_gym/scripts/save_jit.py:39
Method__init__
(self, num_prop, num_critic_obs, num_priv_latent,
rsl_rl/rsl_rl/modules/actor_critic.py:76
Method__init__
( self, input_dim, amp_reward_coef, hidden_layer_sizes, device, task_reward_lerp=0.0)
rsl_rl/rsl_rl/modules/amp_discriminator.py:10
Method__init__
(self, activation_fn, input_size, tsteps, output_size, tanh_encoder_output=False)
rsl_rl/rsl_rl/modules/actor_critic_teleop.py:42
Method__init__
(self, activation_fn, input_size, tsteps, output_size, tanh_encoder_output=False)
rsl_rl/rsl_rl/modules/actor_critic_teleop.py:84
Method__init__
(self, num_observations, num_critic_observations, num_motion
rsl_rl/rsl_rl/modules/actor_critic_teleop.py:183
Method__init__
(self, activation_fn, input_size, tsteps, output_size, tanh_encoder_output=False)
rsl_rl/rsl_rl/modules/actor_critic_mimic.py:42
Method__init__
(self, num_observations, num_critic_observations, num_motion
rsl_rl/rsl_rl/modules/actor_critic_mimic.py:127
Method__init__
(self, activation_fn, input_size, tsteps, output_size, tanh_encoder_output=False)
rsl_rl/rsl_rl/modules/actor_critic_tracking.py:42
Method__init__
(self, activation_fn, input_size, tsteps, output_size, tanh_encoder_output=False)
rsl_rl/rsl_rl/modules/actor_critic_tracking.py:84
Method__init__
(self, num_observations, num_critic_observations, num_motion
rsl_rl/rsl_rl/modules/actor_critic_tracking.py:184
Method__init__
(self, activation_fn, input_size, tsteps, output_size, tanh_encoder_output=False)
rsl_rl/rsl_rl/modules/actor_critic_rma.py:42
Method__init__
(self, num_prop, num_critic_obs, num_priv_latent,
rsl_rl/rsl_rl/modules/actor_critic_rma.py:151
Method__init__
(self, activation_fn, input_size, tsteps, output_size, attention_heads=4, dropout=0.1, tempo
rsl_rl/rsl_rl/modules/actor_critic_future.py:45
Method__init__
(self, num_observations, num_critic_observations, num_motion
rsl_rl/rsl_rl/modules/actor_critic_future.py:249
Method__init__
(self, activation_fn, input_size, tsteps, output_size, tanh_encoder_output=False)
rsl_rl/rsl_rl/modules/dagger_actor.py:42
Method__init__
(self, num_actor_obs, num_critic_obs, num_actions,
rsl_rl/rsl_rl/modules/actor_critic_recurrent.py:42
Method__init__
(self, latent_dim, num_head, dropout_rate)
rsl_rl/rsl_rl/modules/actor_critic_transformer.py:55
Method__init__
(self, input_dim, output_dim, context_len, latent_dim=32, num_head=2, num_layer=2, dropout_rate=0.0)
rsl_rl/rsl_rl/modules/actor_critic_transformer.py:78
Method__init__
(self, activation_fn, input_size, tsteps, output_size, tanh_encoder_output=False)
rsl_rl/rsl_rl/modules/actor_critic_transformer.py:114
Method__init__
(self, num_prop, num_critic_obs, num_priv_latent,
rsl_rl/rsl_rl/modules/actor_critic_transformer.py:187
Method__init__
(self, modules, **kwargs)
rsl_rl/rsl_rl/utils/layers.py:12
Method__init__
(self, pad=3)
rsl_rl/rsl_rl/utils/layers.py:32
Method__init__
(self)
rsl_rl/rsl_rl/utils/layers.py:58
Method__init__
(self, *args, dropout=0., act=nn.Mish(inplace=True), **kwargs)
rsl_rl/rsl_rl/utils/layers.py:90
Method__init__
Calulates the running mean and std of a data stream https://en.wikipedia.org/wiki/Algorithms_for_calculating_variance#Parallel_algori
rsl_rl/rsl_rl/utils/utils.py:80
Method__init__
(self, input_dim, epsilon=1e-4, clip_obs=10.0)
rsl_rl/rsl_rl/utils/utils.py:115
Method__init__
(self, cfg)
rsl_rl/rsl_rl/utils/scale.py:7
Method__init__
(self, shape, device, init_mean=None, init_std=None, eps=1e-4, clip=np.inf, dtype=torch.float)
rsl_rl/rsl_rl/utils/normalizer.py:7
Method__init__
(self, insize, epsilon=1e-05, per_channel=False, norm_only=False)
rsl_rl/rsl_rl/utils/running_mean_std.py:80
Method__init__
(self, base_policy_path, device)
rsl_rl/rsl_rl/runners/on_policy_dagger_runner.py:71
Method__init__
(self, env: VecEnv, train_cfg, log_dir=None,
rsl_rl/rsl_rl/runners/on_policy_dagger_runner.py:83
Method__init__
Initialize a ReplayBuffer object. Arguments: buffer_size (int): maximum size of buffer
rsl_rl/rsl_rl/storage/replay_buffer.py:8
Method__init__
Initialize a ReplayBuffer object. Arguments: buffer_size (int): maximum size of buffer
rsl_rl/rsl_rl/storage/replay_buffer.py:39
Method__init__
(self, num_envs, num_transitions_per_env, obs_shape, privileged_obs_shape, actions_shape, device='cpu')
rsl_rl/rsl_rl/storage/rollout_storage.py:52
Method__init__
Expert dataset provides AMP observations from Dog mocap dataset. time_between_frames: Amount of time in seconds between transition.
rsl_rl/rsl_rl/datasets/motion_loader.py:50
Method__init__
(self, env, teacher_actor, dagger_actor,
rsl_rl/rsl_rl/algorithms/dagger.py:43
Method__init__
(self, env, actor_critic, teacher_actor_critic,
rsl_rl/rsl_rl/algorithms/dagger_ppo.py:54
Method__init__
(self, env, actor_critic, num_learning_epochs=1,
rsl_rl/rsl_rl/algorithms/ppo.py:44
Method__init__
(self, env, actor_critic, discriminator, a
rsl_rl/rsl_rl/algorithms/ppo_amp.py:44
Method__init__
(self, motion_file, object_motion_file, device, motion_decompose=False, mo
pose/pose/utils/motion_lib_hoi.py:38
Method__init__
(self, obj, device)
pose/pose/utils/motion_lib.py:65
Method__init__
(self, motion_file, dof_body_ids, dof_offsets, key_body_ids, device, no_keybody=False)
pose/pose/utils/motion_lib.py:102
Method__init__
(self, path=None)
pose/pose/utils/logger.py:59
Method__init__
(self, name, real=None)
pose/pose/utils/motion_lib_pkl.py:13
Method__init__
(self, task_name: str, task_type: str)
pose/pose/poselib/poselib/visualization/core.py:40
Method__init__
(self, tasks)
pose/pose/poselib/poselib/visualization/core.py:62
Method__init__
(self, task: BasePlotterTask)
pose/pose/poselib/poselib/visualization/core.py:77
Method__init__
( self, task_name: str, lines: np.ndarray, color: str = "blue", line_w
pose/pose/poselib/poselib/visualization/simple_plotter_tasks.py:45
Method__init__
(self, total_num_frames)
pose/pose/poselib/poselib/visualization/common.py:96
Method__init__
(self, task: "BasePlotterTask")
pose/pose/poselib/poselib/visualization/plt_plotter.py:238
Method__init__
( self, task_name: str, skeleton_motion, frame_index=None, joints_colo
pose/pose/poselib/poselib/visualization/skeleton_plotter_tasks.py:96
Method__init__
(self, skeleton_motion_tasks)
pose/pose/poselib/poselib/visualization/skeleton_plotter_tasks.py:205
Method__init__
(self, tensor_backend, skeleton_tree, is_local)
pose/pose/poselib/poselib/skeleton/skeleton3d.py:360
Method__init__
(self, tensor_backend, skeleton_tree, is_local, fps, *args, **kwargs)
pose/pose/poselib/poselib/skeleton/skeleton3d.py:1032
Method__init__
(self, file_path: str, device)
pose/pose/util_funcs/kinematics_model.py:8
Method__iter__
Should override this function to return a list of task primitives
pose/pose/poselib/poselib/visualization/core.py:55
Method__iter__
(self)
pose/pose/poselib/poselib/visualization/core.py:65
← previousnext →401–500 of 982, ranked by callers