MCPcopy Create free account

hub / github.com/daerduoCarey/where2act / functions

Functions253 in github.com/daerduoCarey/where2act

↓ 39 callersMethodclose
(self)
code/env.py:238
↓ 39 callersMethodnorm
L2 norm of the quaternion 4-vector. This should be 1.0 for a unit quaternion (versor) Slow but accurate. If speed is a concern, consi
code/pyquaternion/quaternion.py:473
↓ 20 callersMethodrender
(self)
code/env.py:174
↓ 20 callersMethodwait_n_steps
(self, n: int)
code/robots/panda_robot.py:148
↓ 18 callersMethodget_metadata
(self)
code/camera.py:119
↓ 12 callersMethodget_material
(self, static_friction, dynamic_friction, restitution)
code/env.py:171
↓ 12 callersMethodget_object_qpos
(self)
code/env.py:154
↓ 12 callersMethodmove_to_target_pose
Move the robot hand dynamically to a given target pose Args: target_ee_pose: (4, 4) transformation of robot hand in robot
code/robots/panda_robot.py:115
↓ 12 callersMethodstep
(self)
code/env.py:182
↓ 11 callersMethodget_observation
(self)
code/camera.py:51
↓ 9 callersMethod_validate_number_sequence
Validate a sequence to be of a certain length and ensure it's a numpy array of floats. Raises: ValueError: Invalid length or non-
code/pyquaternion/quaternion.py:137
↓ 8 callersMethodclose_gripper
(self)
code/robots/panda_robot.py:136
↓ 8 callersMethodget_target_part_pose
(self)
code/env.py:161
↓ 8 callersMethodget_target_part_qpos
(self)
code/env.py:157
↓ 8 callersMethodrandom
Generate a random unit quaternion. Uniformly distributed across the rotation space As per: http://planning.cs.uiuc.edu/node198.html
code/pyquaternion/quaternion.py:261
↓ 7 callersMethodcompute_XYZA_matrix
(id1, id2, pts, size1, size2)
code/camera.py:71
↓ 7 callersFunctionprintout
(flog, strout)
code/utils.py:24
↓ 6 callersMethod_fast_normalise
Normalise the object to a unit quaternion using a fast approximation method if appropriate. Object is guaranteed to be a quaternion of approx
code/pyquaternion/quaternion.py:498
↓ 6 callersMethod_normalise
Object is guaranteed to be a unit quaternion after calling this operation UNLESS the object is equivalent to Quaternion(0)
code/pyquaternion/quaternion.py:489
↓ 6 callersMethodget_movable_link_mask
(self, link_ids)
code/camera.py:86
↓ 6 callersMethodload_object
(self, urdf, material, state='closed')
code/env.py:84
↓ 6 callersMethodset_controller_camera_pose
(self, x, y, z, yaw, pitch)
code/env.py:79
↓ 5 callersMethodcompute_camera_XYZA
(self, depth)
code/camera.py:61
↓ 4 callersMethodbgs
(self, d6s)
code/models/model_3d.py:128
↓ 4 callersMethodinference_whole_pc
(self, feats, dirs1, dirs2)
code/models/model_3d.py:239
↓ 4 callersMethodload_data
(self, data_list)
code/data.py:46
↓ 4 callersMethodopen_gripper
(self)
code/robots/panda_robot.py:140
↓ 4 callersMethodset_target_object_part_actor_id
(self, actor_id)
code/env.py:134
↓ 4 callersMethodstart_checking_contact
(self, robot_hand_actor_id, robot_gripper_actor_ids, strict)
code/env.py:164
↓ 4 callersMethodwait_to_start
(self)
code/env.py:231
↓ 3 callersMethod__init__
(self, feat_dim, rv_dim)
code/models/model_3d_legacy.py:117
↓ 3 callersMethod__init__
(self, feat_dim, rv_dim)
code/models/model_3d.py:112
↓ 3 callersMethod_sum_of_squares
(self)
code/pyquaternion/quaternion.py:443
↓ 3 callersMethodbgs
(self, d6s)
code/models/model_3d_legacy.py:134
↓ 3 callersMethodget_metadata_json
(self)
code/camera.py:134
↓ 3 callersMethodget_normal_map
(self)
code/camera.py:77
↓ 3 callersMethodjoin_all
(self)
code/datagen.py:93
↓ 3 callersFunctionqrot
Rotate vector(s) v about the rotation described by quaternion(s) q. Expects a tensor of shape (*, 4) for q and a tensor of shape (*, 3) for v
code/blender_utils/quaternion.py:33
↓ 3 callersFunctionsave_h5
(fn, data)
code/utils.py:243
↓ 3 callersMethodset_object_joint_angles
(self, joint_angles)
code/env.py:131
↓ 3 callersMethodstart_all
(self)
code/datagen.py:77
↓ 2 callersMethod__div__
(self, other)
code/pyquaternion/quaternion.py:397
↓ 2 callersMethod_q_matrix
Matrix representation of quaternion for multiplication purposes.
code/pyquaternion/quaternion.py:565
↓ 2 callersMethod_rotate_quaternion
Rotate a quaternion vector using the stored rotation. Params: q: The vector to be rotated, in quaternion form (0 + xi + yj + kz)
code/pyquaternion/quaternion.py:583
↓ 2 callersMethodadd_one_recollect_job
(self, src_data_dir, dir1, dir2, recollect_record_name, tar_data_dir, x, y)
code/datagen.py:35
↓ 2 callersMethodcalculate_twist
(self, time_to_target, target_ee_pose)
code/robots/panda_robot.py:107
↓ 2 callersMethodexp
Quaternion Exponential. Find the exponential of a quaternion amount. Params: q: the input quaternion/argument as a Quat
code/pyquaternion/quaternion.py:626
↓ 2 callersFunctionforward
(batch, data_features, network, conf, \ is_val=False, step=None, epoch=None, batch_ind=0, num_batch=1,
code/train_3d.py:305
↓ 2 callersFunctionforward
(batch, data_features, network, conf, \ is_val=False, step=None, epoch=None, batch_ind=0, num_batch=1,
code/train_3d_critic.py:287
↓ 2 callersMethodget_ce_loss
(self, pred_logits, gt_labels)
code/models/model_3d.py:174
↓ 2 callersFunctionget_global_position_from_camera
This function is provided only to show how to convert camera observation to world space coordinates. It can be removed if not needed.
code/utils.py:154
↓ 2 callersFunctionget_random_number
(l, r)
code/utils.py:240
↓ 2 callersMethodis_unit
Determine whether the quaternion is of unit length to within a specified tolerance value. Params: tolerance: [optional] maximum a
code/pyquaternion/quaternion.py:554
↓ 2 callersMethodlog
Quaternion Logarithm. Find the logarithm of a quaternion amount. Params: q: the input quaternion/argument as a Quaterni
code/pyquaternion/quaternion.py:649
↓ 2 callersFunctionobj_centened_camera_pos
(dist, azimuth_deg, elevation_deg)
code/blender_utils/render_blender.py:103
↓ 2 callersFunctionprocess_angle_limit
(x)
code/utils.py:233
↓ 2 callersFunctionqeuler
Convert quaternion(s) q to Euler angles. Expects a tensor of shape (*, 4), where * denotes any number of dimensions. Returns a tensor of
code/blender_utils/quaternion.py:53
↓ 2 callersFunctionskew
(vec)
code/utils.py:198
↓ 1 callersMethod__idiv__
(self, other)
code/pyquaternion/quaternion.py:404
↓ 1 callersMethod__init__
(self, feat_dim)
code/models/model_3d_critic_legacy.py:100
↓ 1 callersMethod__init__
(self, feat_dim)
code/models/model_3d_critic.py:97
↓ 1 callersMethod__rdiv__
(self, other)
code/pyquaternion/quaternion.py:407
↓ 1 callersMethod_from_axis_angle
Initialise from axis and angle representation Create a Quaternion by specifying the 3-vector rotation axis and rotation angle (in rad
code/pyquaternion/quaternion.py:238
↓ 1 callersMethod_from_matrix
Initialise from matrix representation Create a Quaternion by specifying the 3x3 rotation or 4x4 transformation matrix (as a numpy arr
code/pyquaternion/quaternion.py:159
↓ 1 callersMethod_q_bar_matrix
Matrix representation of quaternion for multiplication purposes.
code/pyquaternion/quaternion.py:574
↓ 1 callersMethod_vector_conjugate
(self)
code/pyquaternion/quaternion.py:440
↓ 1 callersMethod_wrap_angle
Helper method: Wrap any angle to lie between -pi and pi Odd multiples of pi are wrapped to +pi (as opposed to -pi)
code/pyquaternion/quaternion.py:1018
↓ 1 callersMethodadd_one_collect_job
(self, data_dir, shape_id, category, cnt_id, primact_type, trial_id)
code/datagen.py:27
↓ 1 callersFunctionadjoint_matrix
(pose)
code/utils.py:203
↓ 1 callersMethodbgdR
(self, Rgts, Rps)
code/models/model_3d_legacy.py:143
↓ 1 callersMethodbgdR
(self, Rgts, Rps)
code/models/model_3d.py:137
↓ 1 callersMethodcheck_contact_is_valid
(self)
code/env.py:190
↓ 1 callersMethodcheck_success
(self, result_data, primact_type)
code/data.py:94
↓ 1 callersMethodclear_velocity_command
(self)
code/robots/panda_robot.py:144
↓ 1 callersMethodcompute_joint_velocity_from_twist
This function is a kinematic-level calculation which do not consider dynamics. Pay attention to the frame of twist, is it spatial twi
code/robots/panda_robot.py:53
↓ 1 callersFunctionexport_label
(out, l)
code/utils.py:99
↓ 1 callersFunctionexport_pts
(out, v)
code/utils.py:94
↓ 1 callersFunctionexport_pts_label
(out, v, l)
code/utils.py:104
↓ 1 callersFunctionfunc
(output_folder, num_per_page, input_main_folder, input_folder_list_comma, input_title_list_comma, first_hierac
code/gen_html_hierachy_local.py:22
↓ 1 callersMethodget_6d_rot_loss
(self, pred_6d, gt_6d)
code/models/model_3d_legacy.py:152
↓ 1 callersMethodget_6d_rot_loss
(self, pred_6d, gt_6d)
code/models/model_3d.py:146
↓ 1 callersMethodget_axis
Get the axis or vector about which the quaternion rotation occurs For a null rotation (a purely real quaternion), the rotation angle will
code/pyquaternion/quaternion.py:1027
↓ 1 callersMethodget_ce_loss
(self, pred_logits, gt_labels)
code/models/model_3d_legacy.py:181
↓ 1 callersMethodget_handle_mask
(self)
code/camera.py:95
↓ 1 callersMethodinference_action_score
(self, pcs)
code/models/model_3d.py:257
↓ 1 callersMethodinference_actor
(self, pcs)
code/models/model_3d.py:270
↓ 1 callersMethodinference_critic
(self, pcs, dirs1, dirs2, abs_val=False)
code/models/model_3d.py:284
↓ 1 callersMethodinternal_controller
Control the robot dynamically to execute the given twist for one time step This method will try to execute the joint velocity using the inter
code/robots/panda_robot.py:85
↓ 1 callersFunctionload_obj
(fn)
code/blender_utils/render_using_blender.py:9
↓ 1 callersMethodlog_map
Quaternion logarithm map. Find the logarithm map on the quaternion Riemannian manifold. Params: q: the base point at wh
code/pyquaternion/quaternion.py:720
↓ 1 callersFunctionplot_figure
(idx, up, forward, result)
code/visu_action_heatmap_proposals.py:202
↓ 1 callersFunctionplot_figure
(up, forward)
code/visu_critic_heatmap.py:180
↓ 1 callersFunctionpose2exp_coordinate
Compute the exponential coordinate corresponding to the given SE(3) matrix Note: unit twist is not a unit vector Args: pose
code/utils.py:210
↓ 1 callersFunctionqmul
Multiply quaternion(s) q with quaternion(s) r. Expects two equally-sized tensors of shape (*, 4), where * denotes any number of dimensions.
code/blender_utils/quaternion.py:13
↓ 1 callersFunctionqmul_np
(q, r)
code/blender_utils/quaternion.py:101
↓ 1 callersFunctionquaternionFromYawPitchRoll
(yaw, pitch, roll)
code/blender_utils/render_blender.py:33
↓ 1 callersFunctionrot2so3
(rotation)
code/utils.py:187
↓ 1 callersMethodslerp
Spherical Linear Interpolation between quaternions. Implemented as described in https://en.wikipedia.org/wiki/Slerp Find a valid quat
code/pyquaternion/quaternion.py:832
↓ 1 callersMethodsym_log_map
Quaternion symmetrized logarithm map. Find the symmetrized logarithm map on the quaternion Riemannian manifold. Params:
code/pyquaternion/quaternion.py:737
↓ 1 callersFunctiontag_tr
(k, data, table_title)
code/gen_html_hierachy_local.py:5
next →1–100 of 253, ranked by callers