MCPcopy Index your code

hub / github.com/UWRobotLearning/rmp2 / functions

Functions177 in github.com/UWRobotLearning/rmp2

↓ 8 callersFunctiongradient
compute the gradient with default value of unconnected_gradients=0
rmp2/utils/tf_utils.py:97
↓ 8 callersFunctiontiming
(msg, verbose=True)
rmp2/utils/python_utils.py:9
↓ 7 callersMethod__init__
(self, feature_keys=None, name='rmp', dtype=tf.float32)
rmp2/rmpgraph/rmps/rmps.py:35
↓ 7 callersFunctionmerge_dicts
(original, new_dict)
rmp2/utils/python_utils.py:20
↓ 6 callersMethodget_observation
joint angles and velocities of the robot. for CLOSED_LOOP (NOT RECOMMENDED): both joint angles and velocities are given
rmp2/envs/robot_sim.py:113
↓ 6 callersFunctionip
inner product between vectors or list(s) of tensors
rmp2/utils/tf_utils.py:9
↓ 6 callersFunctionregister_envs_and_models
()
rmp2/utils/rllib_utils.py:11
↓ 6 callersMethodreset
reset the robot to initial configuration and velocity
rmp2/envs/robot_sim.py:71
↓ 4 callersMethodforward_mapping
forward mapping from root node to leaf nodes given environment features -------------------------------------------- :param q
rmp2/rmpgraph/rmpgraph.py:26
↓ 4 callersFunctionipm
vector-matrix-vector product x^T M y for tensors or lists of tensors
rmp2/utils/tf_utils.py:47
↓ 4 callersMethodobs_to_policy_input
params obs: environment observation return policy_input: input to the policy
rmp2/utils/env_wrappers.py:27
↓ 4 callersMethodrmp_evals
evaluate the geometry at leaf nodes given environment features -------------------------------------------- :param xs: list o
rmp2/rmpgraph/rmpgraph.py:35
↓ 4 callersMethodseed
(self, seed=None)
rmp2/envs/robot_env.py:235
↓ 4 callersMethodstep
apply velocity control to the robot :param action: joint accelerations
rmp2/envs/robot_sim.py:90
↓ 3 callersMethodget_extended_observation
get observation array :return obs: an nd array containing sin(joint angles), cos(joint angles), joint velocit
rmp2/envs/robot_env.py:239
↓ 2 callersMethod_collision
check whether the robot is in collision with obstacles :param buffer: buffer for collision checking
rmp2/envs/robot_env.py:419
↓ 2 callersFunctionadd_goal
(bullet_client, position, radius=0.05, color=[0.0, 1.0, 0.0, 1])
rmp2/utils/bullet_utils.py:7
↓ 2 callersFunctionadd_obstacle_ball
(bullet_client, center, radius=0.1, color=[0.4, 0.4, 0.4, 1])
rmp2/utils/bullet_utils.py:27
↓ 2 callersFunctionadd_obstacle_cylinder
(bullet_client, center, radius=0.1, length=0.1, color=[0.4, 0.4, 0.4, 1])
rmp2/utils/bullet_utils.py:38
↓ 2 callersFunctionbatch_jacobian
compute the batch_gradient with default value of unconnected_gradients=0
rmp2/utils/tf_utils.py:127
↓ 2 callersFunctiondist2balls
compute the distance between control point balls and obstacle balls :param control_points: batch_size x num_control_points x dimension, contr
rmp2/rmpgraph/taskmaps/distance_taskmaps.py:18
↓ 2 callersFunctionget_policy_network
(x_shape, y_shape, config, dtype, name='policy_network')
rmp2/policies/policy_networks.py:9
↓ 2 callersFunctionget_rmp
return an rmp instance given type and params :param rmp_type: str, rmp type :param params: dict, rmp parameters :param dtype: str, e.
rmp2/rmpgraph/rmps/rmps.py:8
↓ 2 callersFunctionget_robot_urdf_path
(robot_name)
rmp2/utils/robot_config_utils.py:29
↓ 2 callersMethodobs_to_value_input
params obs: environment observation return value_input: input to the value function
rmp2/utils/env_wrappers.py:35
↓ 2 callersFunctionpolicy
(state)
examples/rmp2/rmp2_3link.py:27
↓ 2 callersFunctionpolicy
(state)
examples/rmp2/rmp2_franka.py:28
↓ 2 callersMethodreset
reset time and simulator
rmp2/envs/robot_env.py:187
↓ 2 callersFunctionsample_from_torus_2d
(random_generator, angle_center, angle_range, major_radius, minor_radius)
rmp2/utils/np_utils.py:23
↓ 2 callersFunctionsample_from_torus_3d
(random_generator, angle_center, angle_range, major_radius, minor_radius, height)
rmp2/utils/np_utils.py:7
↓ 2 callersFunctionsolve
solution to linear equation Ax=b in batch
rmp2/utils/tf_utils.py:90
↓ 2 callersMethodsolve
(self, q, qd, method='rmp2', mode='value', **features)
rmp2/rmpgraph/rmpgraph.py:62
↓ 2 callersMethodtransformation
homogeneous transformation matrices for the joint :param q: joint configuration :return frame: homogenous transformation matr
rmp2/kinematics/tf_fk.py:45
↓ 1 callersFunctionT_prismatic
(xyz, rpy, axis, qi)
rmp2/utils/tf_transform_utils.py:37
↓ 1 callersFunctionT_revolute
(xyz, rpy, axis, qi)
rmp2/utils/tf_transform_utils.py:88
↓ 1 callersFunctionT_rpy
Homogeneous transformation matrix with roll pitch yaw.
rmp2/utils/tf_transform_utils.py:29
↓ 1 callersMethod__del__
(self)
rmp2/envs/robot_env.py:232
↓ 1 callersMethod__init__
(self, x_shape, y_shape, rmp_config, dtype='float32', name='flat_rmpnet', **kwargs)
rmp2/policies/policy_networks.py:56
↓ 1 callersMethod_calc_damping_gain_divisor
(self, x)
rmp2/rmpgraph/rmps/rmps.py:292
↓ 1 callersMethod_clear_goal_and_obstacles
clear the goal and obstacle objects in pybullet
rmp2/envs/robot_env.py:453
↓ 1 callersMethod_forward_kinematics
recursively compute forward kinematics ------------------------ :param base_link: the name of the current base link :
rmp2/kinematics/tf_fk.py:127
↓ 1 callersMethod_generate_random_goal
randomly generate a goal for the end effector
rmp2/envs/robot_env.py:487
↓ 1 callersMethod_generate_random_initial_config
generate a random initial configuration and joint velocities for the robot
rmp2/envs/robot_env.py:469
↓ 1 callersMethod_generate_random_obstacles
randomly generate obstacles in the environment
rmp2/envs/robot_env.py:494
↓ 1 callersMethod_get_goal_reward
the part of reward function for going to goal
rmp2/envs/robot_env.py:403
↓ 1 callersMethod_get_obstacle_reward
the part of reward function for avoiding obstacles
rmp2/envs/robot_env.py:388
↓ 1 callersMethod_get_reward
the reward function :return reward: the current reward
rmp2/envs/robot_env.py:357
↓ 1 callersMethod_goal_obstacle_collision
check whether the (potentially randomly generated) goal and obstacles are in collision :param buffer: buffer for collision ch
rmp2/envs/robot_env.py:437
↓ 1 callersMethod_length_scale_normalized_repulsion_distance
(self, x)
rmp2/rmpgraph/rmps/rmps.py:289
↓ 1 callersMethod_policy
(self, input_dict)
rmp2/policies/gaussian_policy.py:64
↓ 1 callersMethod_set_joint_indices
set the joint limits for the robot
rmp2/envs/robot_sim.py:139
↓ 1 callersMethod_smooth_activation_gate
(self, x)
rmp2/rmpgraph/rmps/rmps.py:283
↓ 1 callersMethod_sort_joint_names
sort the joint names recursively
rmp2/kinematics/tf_fk.py:166
↓ 1 callersMethod_termination
check wether the current episode is terminated due to either out of steps or collision
rmp2/envs/robot_env.py:336
↓ 1 callersMethod_value
(self, input_dict)
rmp2/policies/gaussian_policy.py:70
↓ 1 callersFunctionadd_collision_goal
(bullet_client, position, radius=0.05, color=[0.0, 1.0, 0.0, 1])
rmp2/utils/bullet_utils.py:17
↓ 1 callersFunctiondist2boxes
compute the distance between control point balls and obstacle boxes :param control_points: batch_size x num_control_points x dimension, contr
rmp2/rmpgraph/taskmaps/distance_taskmaps.py:38
↓ 1 callersFunctiondistmap
(type, control_points, control_point_radii, **params)
rmp2/rmpgraph/taskmaps/distance_taskmaps.py:10
↓ 1 callersMethodforward_kinematics
compute the forward kinematics --------------------------- :param q: configuration space coordinate :return x: dictio
rmp2/kinematics/tf_fk.py:104
↓ 1 callersFunctionget_control_points
(link_positions, arm_collision_controllers)
rmp2/rmpgraph/robotics.py:23
↓ 1 callersFunctionget_robot_eef_uid
(robot_name)
rmp2/utils/robot_config_utils.py:39
↓ 1 callersFunctionload_env_wrapper
(envid, dtype=tf.float32)
rmp2/utils/env_wrappers.py:9
↓ 1 callersFunctionload_robot_config
(robot_name=None, config_path=None)
rmp2/utils/robot_config_utils.py:8
↓ 1 callersFunctionpdist2
pairwise distance between x and y in batch
rmp2/utils/tf_utils.py:76
↓ 1 callersMethodpolicy_output_to_action
params actions: output of the policy return env_actions: action to the environment
rmp2/utils/env_wrappers.py:42
↓ 1 callersMethodrmp_eval
evaluating the rmp node :param x: tf.Tensor, position :param xd: tf.Tensor, velocity :param rmp_type: str, canonical
rmp2/rmpgraph/rmps/rmps.py:49
↓ 1 callersMethodrmp_eval_canonical
canonical rmp eval :param x: tf.Tensor position :param xd: tf.Tensor velocity :param features: dict, all selected fea
rmp2/rmpgraph/rmps/rmps.py:65
↓ 1 callersMethodrmp_eval_canonical
(self, x, xd, **features)
rmp2/rmpgraph/rmps/rmps.py:106
↓ 1 callersMethodrmp_eval_canonical
(self, x, xd, **features)
rmp2/rmpgraph/rmps/rmps.py:147
↓ 1 callersMethodrmp_eval_canonical
(self, x, xd, **features)
rmp2/rmpgraph/rmps/rmps.py:180
↓ 1 callersMethodrmp_eval_canonical
(self, x, xd, **features)
rmp2/rmpgraph/rmps/rmps.py:222
↓ 1 callersMethodrmp_eval_canonical
(self, x, xd, **features)
rmp2/rmpgraph/rmps/rmps.py:296
↓ 1 callersMethodrmp_eval_canonical
(self, x, xd, **features)
rmp2/rmpgraph/rmps/rmps.py:373
↓ 1 callersMethodrmp_eval_natural
natural rmp eval :param x: tf.Tensor position :param xd: tf.Tensor velocity :param features: dict, all selected featu
rmp2/rmpgraph/rmps/rmps.py:76
↓ 1 callersFunctionrotation_rpy
Returns a rotation matrix from roll pitch yaw. ZYX convention.
rmp2/utils/tf_transform_utils.py:15
↓ 1 callersMethodset_joint_names
topologically sort the joint names
rmp2/kinematics/tf_fk.py:155
↓ 1 callersMethodsolve_direct
(self, q, qd, mode='value', **features)
rmp2/rmpgraph/rmpgraph.py:168
↓ 1 callersMethodsolve_rmp2
(self, q, qd, mode='value', **features)
rmp2/rmpgraph/rmpgraph.py:71
Method__call__
(self, x)
rmp2/utils/tf_utils.py:179
Method__call__
(self, q, qd, **features)
rmp2/rmpgraph/rmpgraph.py:243
Method__call__
(self, x, xd, rmp_type='canonical', **features)
rmp2/rmpgraph/rmps/rmps.py:86
Method__call__
(self, x, xd, **features)
rmp2/policies/policy_networks.py:46
Method__call__
(self, x)
rmp2/policies/policy_networks.py:64
Method__init__
(self, style=None, colormap=None)
rmp2/utils/plot_configs.py:65
Method__init__
(self, x_shape, y_shape, hidden_units=(), activation='tanh', hidden_layer_init_scale=1.0, output_layer
rmp2/utils/tf_utils.py:142
Method__init__
(self, dtype=tf.float32)
rmp2/utils/env_wrappers.py:51
Method__init__
(self, dtype='float32')
rmp2/utils/env_wrappers.py:66
Method__init__
(self, dtype='float32')
rmp2/utils/env_wrappers.py:95
Method__init__
(self, dtype=tf.float32)
rmp2/utils/env_wrappers.py:125
Method__init__
(self, dtype='float32')
rmp2/utils/env_wrappers.py:140
Method__init__
(self, dtype='float32')
rmp2/utils/env_wrappers.py:168
Method__init__
:param urdf_joint: joint class from urdf_parser_py :param dtype: data type
rmp2/kinematics/tf_fk.py:15
Method__init__
:param urdf_path: path to urdf file :param link_set: list of links to compute fk :param workspace_dim: workspace dimension
rmp2/kinematics/tf_fk.py:69
Method__init__
(self, robot_name=None, config_path=None, config=None, workspace_dim=3, dtype=tf.float32, rmp_type='canonical'
rmp2/rmpgraph/robotics.py:41
Method__init__
(self, rmps, rmp_type='canonical', timed=False, offset=1e-3, dtype=tf.float32, name='rmpgraph', **kwargs)
rmp2/rmpgraph/rmpgraph.py:15
Method__init__
(self, metric_scalar, position_gain, damping_gain, robust_position_term_thresh, inertia, name=
rmp2/rmpgraph/rmps/rmps.py:95
Method__init__
( self, metric_scalar, metric_length_scale, metric_exploder_eps, metric_velocity_gate_
rmp2/rmpgraph/rmps/rmps.py:128
Method__init__
( self, max_velocity, velocity_damping_region, damping_gain, metric_weight, eps=1e-6,
rmp2/rmpgraph/rmps/rmps.py:166
Method__init__
( self, accel_p_gain, accel_d_gain, accel_norm_eps, metric_alpha_length_scale,
rmp2/rmpgraph/rmps/rmps.py:203
Method__init__
( self, margin, damping_gain, damping_std_dev, damping_robustness_eps,
rmp2/rmpgraph/rmps/rmps.py:255
next →1–100 of 177, ranked by callers