MCPcopy Create free account

hub / github.com/Improbable-AI/walk-these-ways / functions

Functions411 in github.com/Improbable-AI/walk-these-ways

↓ 1 callersMethod_get_heights
Samples heights of the terrain at required points around each robot. The points are offset by the base's position and rotated by the base
go1_gym/envs/base/legged_robot.py:1772
↓ 1 callersMethod_get_noise_scale_vec
Sets a vector used to scale the noise added to the observations. [NOTE]: Must be adapted when changing the observations structure
go1_gym/envs/base/legged_robot.py:1053
↓ 1 callersMethod_init_buffers
Initialize torch tensors which will contain simulation states and processed quantities
go1_gym/envs/base/legged_robot.py:1123
↓ 1 callersMethod_init_command_distribution
(self, env_ids)
go1_gym/envs/base/legged_robot.py:1299
↓ 1 callersMethod_init_custom_buffers__
(self)
go1_gym/envs/base/legged_robot.py:1260
↓ 1 callersMethod_init_height_points
Returns points at which the height measurments are sampled (in base frame) Returns: [torch.Tensor]: Tensor of shape (num_envs, s
go1_gym/envs/base/legged_robot.py:1756
↓ 1 callersMethod_post_physics_step_callback
Callback called before computing terminations, rewards, and observations Default behaviour: Compute ang vel command based on target and h
go1_gym/envs/base/legged_robot.py:675
↓ 1 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
go1_gym/envs/base/legged_robot.py:1385
↓ 1 callersMethod_process_dof_props
Callback allowing to store/change/randomize the DOF properties of each environment. Called During environment creation. Base
go1_gym/envs/base/legged_robot.py:581
↓ 1 callersMethod_process_rigid_body_props
(self, props, env_id)
go1_gym/envs/base/legged_robot.py:667
↓ 1 callersMethod_process_rigid_shape_props
Callback allowing to store/change/randomize the rigid shape properties of each environment. Called During environment creation.
go1_gym/envs/base/legged_robot.py:563
↓ 1 callersMethod_render_headless
(self)
go1_gym/envs/base/legged_robot.py:1622
↓ 1 callersMethod_step_contact_targets
(self)
go1_gym/envs/base/legged_robot.py:826
↓ 1 callersMethodadd_command_profile
(self, command_profile)
go1_gym_deploy/utils/deployment_runner.py:61
↓ 1 callersMethodadd_control_agent
(self, agent, name)
go1_gym_deploy/utils/deployment_runner.py:47
↓ 1 callersMethodadd_policy
(self, policy)
go1_gym_deploy/utils/deployment_runner.py:58
↓ 1 callersMethodadd_transitions
(self, transition: Transition)
go1_gym_learn/ppo_cse/rollout_storage.py:57
↓ 1 callersMethodadd_transitions
(self, transition: Transition)
go1_gym_learn/ppo/rollout_storage.py:57
↓ 1 callersFunctionbuild_mlp
(in_dim, units, layers, out_dim, act='relu', layer_norm=False, act_final=False)
scripts/actuator_net/utils.py:66
↓ 1 callersMethodcheck_termination
Check if environments need to be reset
go1_gym/envs/base/legged_robot.py:138
↓ 1 callersFunctionclass_to_dict
(obj)
go1_gym_deploy/utils/logger.py:8
↓ 1 callersFunctionclass_to_dict
(obj)
go1_gym_deploy/envs/lcm_agent.py:13
↓ 1 callersMethodcompute_observations
Computes observations
go1_gym/envs/base/legged_robot.py:302
↓ 1 callersMethodcompute_returns
(self, last_critic_obs, last_critic_privileged_obs)
go1_gym_learn/ppo_cse/ppo.py:93
↓ 1 callersMethodcompute_returns
(self, last_critic_obs, last_critic_privileged_obs)
go1_gym_learn/ppo/ppo.py:90
↓ 1 callersMethodcompute_reward
Compute rewards Calls each reward function which had a non-zero scale (processed in self._prepare_reward_function()) adds eac
go1_gym/envs/base/legged_robot.py:263
↓ 1 callersMethodcreate_sim
Creates simulation, terrain and evironments
go1_gym/envs/base/legged_robot.py:493
↓ 1 callersMethodcurriculum
(self, cfg)
go1_gym/utils/terrain.py:90
↓ 1 callersFunctionget_activation
(act_name)
go1_gym_learn/ppo_cse/actor_critic.py:149
↓ 1 callersFunctionget_activation
(act_name)
go1_gym_learn/ppo/actor_critic.py:176
↓ 1 callersMethodget_body_angular_vel
(self)
go1_gym_deploy/utils/cheetah_state_estimator.py:133
↓ 1 callersMethodget_body_linear_vel
(self)
go1_gym_deploy/utils/cheetah_state_estimator.py:129
↓ 1 callersMethodget_camera_bottom
(self)
go1_gym_deploy/utils/cheetah_state_estimator.py:248
↓ 1 callersMethodget_camera_front
(self)
go1_gym_deploy/utils/cheetah_state_estimator.py:245
↓ 1 callersMethodget_camera_left
(self)
go1_gym_deploy/utils/cheetah_state_estimator.py:254
↓ 1 callersMethodget_camera_rear
(self)
go1_gym_deploy/utils/cheetah_state_estimator.py:251
↓ 1 callersMethodget_camera_right
(self)
go1_gym_deploy/utils/cheetah_state_estimator.py:257
↓ 1 callersMethodget_contact_state
(self)
go1_gym_deploy/utils/cheetah_state_estimator.py:142
↓ 1 callersMethodget_dof_pos
(self)
go1_gym_deploy/utils/cheetah_state_estimator.py:226
↓ 1 callersMethodget_dof_vel
(self)
go1_gym_deploy/utils/cheetah_state_estimator.py:230
↓ 1 callersMethodget_gravity_vector
(self)
go1_gym_deploy/utils/cheetah_state_estimator.py:138
↓ 1 callersFunctionget_likely_iface
()
go1_gym_deploy/utils/network_config_unitree.py:18
↓ 1 callersFunctionget_rotation_matrix_from_rpy
Get rotation matrix from the given quaternion. Args: q (np.array[float[4]]): quaternion [w,x,y,z] Returns: np.array[float
go1_gym_deploy/utils/cheetah_state_estimator.py:23
↓ 1 callersFunctionget_saved_interface_name
()
go1_gym_deploy/utils/network_config_unitree.py:8
↓ 1 callersMethodget_yaw
(self)
go1_gym_deploy/utils/cheetah_state_estimator.py:236
↓ 1 callersMethodinit
go1_gym_deploy/unitree_legged_sdk_bin/lcm_position.cpp:63
↓ 1 callersMethodinit_storage
(self, num_envs, num_transitions_per_env, actor_obs_shape, privileged_obs_shape, obs_history_shape,
go1_gym_learn/ppo_cse/ppo.py:54
↓ 1 callersMethodinit_storage
(self, num_envs, num_transitions_per_env, actor_obs_shape, privileged_obs_shape, obs_history_shape,
go1_gym_learn/ppo/ppo.py:51
↓ 1 callersMethodinitialize_terrains
(self)
go1_gym/utils/terrain.py:67
↓ 1 callersFunctionis_met
(scale, l2_err, threshold)
go1_gym/envs/base/curriculum.py:6
↓ 1 callersMethodlearn
(self, num_learning_iterations, init_at_random_ep_len=False, eval_freq=100, eval_expert=False)
go1_gym_learn/ppo/__init__.py:92
↓ 1 callersFunctionlin_vel_rmsd
(env, actor_critic, obs)
go1_gym_learn/eval_metrics/metrics.py:8
↓ 1 callersFunctionlin_vel_x
(env, actor_critic, obs)
go1_gym_learn/eval_metrics/metrics.py:16
↓ 1 callersFunctionload_and_run_policy
(label, experiment_name, max_vel=1.0, max_yaw_vel=1.0)
go1_gym_deploy/scripts/deploy_policy.py:15
↓ 1 callersMethodload_cfgs
(self)
go1_gym/utils/terrain.py:38
↓ 1 callersFunctionload_env
(label, headless=False)
scripts/play.py:32
↓ 1 callersFunctionload_policy
(logdir)
go1_gym_deploy/scripts/deploy_policy.py:57
↓ 1 callersFunctionload_policy
(logdir)
scripts/play.py:17
↓ 1 callersMethodlog_video
(self, it)
go1_gym_learn/ppo_cse/__init__.py:277
↓ 1 callersMethodlog_video
(self, it)
go1_gym_learn/ppo/__init__.py:267
↓ 1 callersFunctionmain
()
go1_gym_deploy/utils/network_config_unitree.py:54
↓ 1 callersMethodmini_batch_generator
(self, num_mini_batches, num_epochs=8)
go1_gym_learn/ppo_cse/rollout_storage.py:100
↓ 1 callersMethodmini_batch_generator
(self, num_mini_batches, num_epochs=8)
go1_gym_learn/ppo/rollout_storage.py:100
↓ 1 callersFunctionplay_go1
(headless=True)
scripts/play.py:89
↓ 1 callersFunctionpolicy
(obs, info={})
scripts/play.py:22
↓ 1 callersMethodpoll
(self, cb=None)
go1_gym_deploy/utils/cheetah_state_estimator.py:375
↓ 1 callersMethodpost_physics_step
check terminations, compute observations and rewards calls self._post_physics_step_callback() for common computations calls
go1_gym/envs/base/legged_robot.py:90
↓ 1 callersFunctionpower_consumption
(env, actor_critic, obs)
go1_gym_learn/eval_metrics/metrics.py:35
↓ 1 callersMethodprocess_env_step
(self, rewards, dones, infos)
go1_gym_learn/ppo_cse/ppo.py:79
↓ 1 callersMethodprocess_env_step
(self, rewards, dones, infos)
go1_gym_learn/ppo/ppo.py:76
↓ 1 callersMethodpublish_30Hz
(self)
go1_gym_deploy/tests/check_camera_msgs.py:116
↓ 1 callersMethodpublish_action
(self, action, hard_reset=False)
go1_gym_deploy/envs/lcm_agent.py:189
↓ 1 callersMethodrandomized_terrain
(self, cfg)
go1_gym/utils/terrain.py:80
↓ 1 callersMethodrender_gui
(self, sync_frame_time=True)
go1_gym/envs/base/base_task.py:108
↓ 1 callersMethodreset
(self)
go1_gym/envs/go1/velocity_tracking/__init__.py:46
↓ 1 callersMethodreset
(self)
go1_gym/envs/wrappers/history_wrapper.py:37
↓ 1 callersMethodreset
(self, env_ids: Union[list, torch.Tensor])
go1_gym_learn/env/vec_env.py:30
↓ 1 callersMethodreset_gait_indices
(self)
go1_gym_deploy/envs/lcm_agent.py:225
↓ 1 callersMethodreset_idx
Reset some environments. Calls self._reset_dofs(env_ids), self._reset_root_states(env_ids), and self._resample_commands(env_ids) and
go1_gym/envs/base/legged_robot.py:150
↓ 1 callersMethodrun
(self, num_log_steps=1000000000, max_steps=100000000, logging=True)
go1_gym_deploy/utils/deployment_runner.py:125
↓ 1 callersFunctionrun_env
(render=False, headless=False)
scripts/test.py:14
↓ 1 callersMethodsample_bins
default to uniform
go1_gym/envs/base/curriculum.py:67
↓ 1 callersMethodsample_uniform_from_cell
(self, centroids)
go1_gym/envs/base/curriculum.py:82
↓ 1 callersMethodselected_terrain
(self, cfg)
go1_gym/utils/terrain.py:99
↓ 1 callersMethodset_camera
Set camera position and direction
go1_gym/envs/base/legged_robot.py:518
↓ 1 callersMethodset_to
(self, low, high, value=1.0)
go1_gym/envs/base/curriculum.py:18
↓ 1 callersMethodspin
(self)
go1_gym_deploy/utils/cheetah_state_estimator.py:393
↓ 1 callersMethodstep
(self, action)
go1_gym/envs/wrappers/history_wrapper.py:18
↓ 1 callersMethodstep
(self, actions)
go1_gym/envs/base/base_task.py:105
↓ 1 callersFunctiontrain_actuator_network
(xs, ys, actuator_network_path)
scripts/actuator_net/utils.py:78
↓ 1 callersFunctiontrain_go1
(headless=True)
scripts/train.py:1
↓ 1 callersMethodupdate
(self)
go1_gym_learn/ppo_cse/ppo.py:97
↓ 1 callersMethodupdate_distribution
(self, observation_history)
go1_gym_learn/ppo_cse/actor_critic.py:113
↓ 1 callersMethodupdate_distribution
(self, observations, privileged_observations)
go1_gym_learn/ppo/actor_critic.py:137
FunctionCoT
(env, actor_critic, obs)
go1_gym_learn/eval_metrics/metrics.py:40
MethodCustom
go1_gym_deploy/unitree_legged_sdk_bin/lcm_position.cpp:24
MethodRobotControl
go1_gym_deploy/unitree_legged_sdk_bin/lcm_position.cpp:131
MethodUDPRecv
go1_gym_deploy/unitree_legged_sdk_bin/lcm_position.cpp:98
MethodUDPSend
go1_gym_deploy/unitree_legged_sdk_bin/lcm_position.cpp:103
Method__getattr__
(self, name)
go1_gym_deploy/envs/history_wrapper.py:50
← previousnext →101–200 of 411, ranked by callers