MCPcopy Create free account

hub / github.com/THU-VCLab/HGGD / functions

Functions616 in github.com/THU-VCLab/HGGD

↓ 2 callersMethodfrom_bb
(cls, points)
dataset/grasp.py:468
↓ 2 callersMethodfrom_list
(cls, grasp_list)
dataset/grasp.py:233
↓ 2 callersFunctiongenerate_scene_pointcloud
Author: chenxi-wang **Input:** - dataset_root: str, graspnet dataset root - scene_name: str, name of scene folder, e.g. scene_0000
customgraspnetAPI/utils/utils.py:160
↓ 2 callersFunctionget_6d_key_points
**Input:** - centers: np.array(-1,3) of the translation - Rs: np.array(-1,3,3) of the rotation matrix - widths: np.array(-1) of th
dataset/graspnet_utils.py:45
↓ 2 callersFunctionget_batch_key_points
**Input:** - centers: np.array(-1,3) of the translation - Rs: np.array(-1,3,3) of the rotation matrix - widths: np.array(-1) of th
util.py:108
↓ 2 callersFunctionget_camera_intrinsic
**Input:** - camera: string of type of camera, "realsense" or "kinect". **Output:** - numpy array of shape (3, 3) of the camera in
util.py:56
↓ 2 callersFunctionget_center_group_label
(local_center: List, grasp_labels: List, local_grasp_num)
dataset/pc_dataset_tools.py:233
↓ 2 callersMethodget_gtbb
(self, rect_gg, rot=0, zoom=1.0)
dataset/base_grasp_dataset.py:245
↓ 2 callersFunctionget_model_grasps
Author: chenxi-wang Load grasp labels from .npz files.
customgraspnetAPI/utils/utils.py:385
↓ 2 callersFunctionget_ori_grasp_label
(grasppath)
dataset/pc_dataset_tools.py:212
↓ 2 callersFunctiongetposevectorlist
(objectidlist, is_resume, num_frame, frame_number, xml_dir)
customgraspnetAPI/utils/xmlhandler.py:137
↓ 2 callersMethodgrasp_quality
Computes the quality of a two-finger point grasps on a given object using a quasi-static model. Parameters ----------
customgraspnetAPI/utils/dexnet/grasping/quality.py:64
↓ 2 callersFunctioninference
(view_points, xyzs, x, ori_depth, vis_heatmap=False,
demo.py:135
↓ 2 callersFunctioniter_anchors
(anchor_ranges, anchors, angles)
dataset/utils.py:117
↓ 2 callersFunctionjit_qmul
(q0, q1)
models/losses.py:144
↓ 2 callersMethodloadBGR
**Input:** - sceneId: int of the scene index. - camera: string of type of camera, 'realsense' or 'kinect' - annId:
customgraspnetAPI/graspnet.py:464
↓ 2 callersMethodloadSceneModel
**Input:** - sceneId: int of the scene index. - camera: string of type of camera, 'realsense' or 'kinect' - annId:
customgraspnetAPI/graspnet.py:638
↓ 2 callersMethodnormal_force_magnitude
Returns the component of the force that the contact would apply along the normal direction. Returns ------- float
customgraspnetAPI/utils/dexnet/grasping/contacts.py:213
↓ 2 callersMethodparams_from_configuration
Convert configuration vector to a set of params for the class.
customgraspnetAPI/utils/dexnet/grasping/grasp.py:78
↓ 2 callersMethodrandom_sample
**Input:** - numGrasp: int of the number of sampled grasps. **Output:** - GraspGroup instance of sample grasps.
customgraspnetAPI/grasp.py:528
↓ 2 callersMethodread
Reads in the .obj file and returns a Mesh3D representation of that mesh. Returns ------- :obj:`Mesh3D` A
customgraspnetAPI/utils/dexnet/grasping/meshpy/obj_file.py:52
↓ 2 callersFunctionrotation_distance
(a: torch.Tensor, b: torch.Tensor, symmetry=True)
dataset/utils.py:62
↓ 2 callersMethodsave_npy
**Input:** - npy_file_path: string of the file path.
customgraspnetAPI/grasp.py:491
↓ 2 callersMethodsort_by_score
**Input:** - reverse: bool of order, if False, from high to low, if True, from low to high. **Output:** - no outpu
customgraspnetAPI/grasp.py:511
↓ 2 callersMethodsurface_points
Returns the points on the surface. Parameters ---------- grid_basis : bool If False, the surface points are trans
customgraspnetAPI/utils/dexnet/grasping/meshpy/sdf.py:582
↓ 2 callersMethodto_open3d_geometry
(self, color=None)
dataset/grasp.py:122
↓ 2 callersMethodto_open3d_geometry_list
(self, color=None)
dataset/grasp.py:399
↓ 2 callersMethodto_open3d_geometry_list
**Output:** - list of open3d.geometry.Geometry of the grippers.
customgraspnetAPI/grasp.py:499
↓ 2 callersMethodtransform
Transform by a delta transform. Parameters ---------- delta_T : :obj:`RigidTransform` the transformation from the
customgraspnetAPI/utils/dexnet/grasping/graspable_object.py:183
↓ 2 callersMethodtransform
Return a copy of the mesh that has been transformed by T. Parameters ---------- T : :obj:`RigidTransform` The Rig
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:754
↓ 2 callersFunctiontransform_points
**Input:** - points: (N, 3) - trans: (4, 4) **Output:** - points_trans: (N, 3)
customgraspnetAPI/utils/eval_utils.py:90
↓ 2 callersFunctionviewpoint_params_to_matrix
Author: chenxi-wang **Input:** - towards: numpy array towards vector with shape (3,). - angle: float of in-plane rotation. **
customgraspnetAPI/utils/utils.py:309
↓ 2 callersFunctionvis6D
**Input:** - dataset_root: str, graspnet dataset root - scene_name: str, name of scene folder, e.g. scene_0000 - anno_idx: int, fr
customgraspnetAPI/utils/vis.py:234
↓ 2 callersFunctionvisObjGrasp
Author: chenxi-wang **Input:** - dataset_root: str, graspnet dataset root - obj_idx: int, index of object model - num_grasp:
customgraspnetAPI/utils/vis.py:308
↓ 2 callersMethodwrench_in_positive_span
Check whether a target can be exerted by positive combinations of wrenches in a given basis with L1 norm fonger force limit limit. Pa
customgraspnetAPI/utils/dexnet/grasping/quality.py:813
↓ 1 callersMethod__init__
(self, sdf, mesh, key='', model_name='',
customgraspnetAPI/utils/dexnet/grasping/graspable_object.py:60
↓ 1 callersMethod__init__
(self, k=3)
models/pointnet.py:9
↓ 1 callersMethod__len__
(self)
dataset/grasp.py:225
↓ 1 callersMethod_angle_aligned_with_stable_pose
Returns the y-axis rotation angle that'd allow the current pose to align with stable pose.
customgraspnetAPI/utils/dexnet/grasping/grasp.py:1010
↓ 1 callersMethod_angle_aligned_with_table
Returns the y-axis rotation angle that'd allow the current pose to align with the table normal.
customgraspnetAPI/utils/dexnet/grasping/grasp.py:1081
↓ 1 callersMethod_center_of_tri
Return the center of the given triangle. Parameters ---------- tri : :obj:`numpy.ndarray` of int The triangle for
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1370
↓ 1 callersMethod_closer_segment
Compute which segment is closer to a given point by seeing which side of the midline between the two segments the point falls on. Par
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1941
↓ 1 callersMethod_closest_segment
Returns the finite line segment(s) the least distance from the input point. Parameters ---------- point : :obj:`numpy
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1905
↓ 1 callersMethod_compute_com_uniform
Computes the center of mass using a uniform mass distribution assumption. Returns ------- :obj:`numpy.ndarray` of flo
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1316
↓ 1 callersMethod_compute_gradients
Computes the gradients of the SDF. Returns ------- :obj:`list` of :obj:`numpy.ndarray` of float A list of ndarray
customgraspnetAPI/utils/dexnet/grasping/meshpy/sdf.py:194
↓ 1 callersMethod_compute_inertia
Computes the mesh inertia matrix. Note ---- Only works for watertight meshes. Returns ------- :o
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1288
↓ 1 callersMethod_compute_mass
Computes the mesh mass. Note ---- Only works for watertight meshes. Returns ------- float
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1274
↓ 1 callersMethod_compute_normal
Compute outward facing normal at contact, according to in_direction. Indexes into the SDF grid coordinates to lookup the normal info.
customgraspnetAPI/utils/dexnet/grasping/contacts.py:94
↓ 1 callersMethod_compute_prob_map
Creates a map from faces to static stability probabilities. Parameters ---------- vertices : :obj:`list` of :obj:`_GraphVerte
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1998
↓ 1 callersMethod_compute_proj_area
Projects vertices onto the unit sphere from the center of mass and computes the projected area. Parameters ----------
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1469
↓ 1 callersMethod_compute_surface_points
(self)
customgraspnetAPI/utils/dexnet/grasping/meshpy/sdf.py:572
↓ 1 callersMethod_contains_proj_point
Is the given 3D point (assumed to be on the line that contains the segment) actually within the segment? Parameters
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1713
↓ 1 callersMethod_covariance_of_tri
Return the covariance matrix of the given triangle. Parameters ---------- tri : :obj:`numpy.ndarray` of int The t
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1389
↓ 1 callersMethod_max_edge_length
Compute the maximum edge length of a triangle. Parameters ---------- tri : :obj:`numpy.ndarray` of int The triang
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1814
↓ 1 callersMethod_ordered_tuple
Returns an ordered tuple that represents the segment. The points within are ordered lexicographically. Returns -
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1735
↓ 1 callersMethod_rand_point_on_tri
Return a random point on the given triangle. Parameters ---------- tri : :obj:`numpy.ndarray` of int The triangle
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1450
↓ 1 callersMethod_read_2d
Reads in a 2D SDF file and returns a Sdf object. Returns ------- :obj:`Sdf2D` A 2DSdf created from the data in th
customgraspnetAPI/utils/dexnet/grasping/meshpy/sdf_file.py:102
↓ 1 callersMethod_read_3d
Reads in a 3D SDF file and returns a Sdf object. Returns ------- :obj:`Sdf3D` A 3DSdf created from the data in th
customgraspnetAPI/utils/dexnet/grasping/meshpy/sdf_file.py:69
↓ 1 callersMethod_setup
Sets up common variables for grasp quality evaluations.
customgraspnetAPI/utils/dexnet/grasping/grasp_quality_function.py:98
↓ 1 callersMethod_setup
(self)
customgraspnetAPI/utils/dexnet/grasping/grasp_quality_function.py:179
↓ 1 callersMethod_signed_distance
Returns the signed distance at the given coordinates, interpolating if necessary. Parameters ---------- coords : :obj
customgraspnetAPI/utils/dexnet/grasping/meshpy/sdf.py:293
↓ 1 callersMethod_signed_volume_of_tri
Return the signed volume of the given triangle. Parameters ---------- tri : :obj:`numpy.ndarray` of int The trian
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1350
↓ 1 callersMethod_tri_area_percentages
Return a list of the percent area each triangle contributes to the mesh's surface area. Returns ------- :obj:`list` o
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1431
↓ 1 callersMethodadd_edge
Connects this vertex to the input child vertex. Parameters ---------- child : :obj:`_GraphVertex`
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1800
↓ 1 callersMethodaddobject
(self, pose, objname, objpath, objid)
customgraspnetAPI/utils/xmlhandler.py:26
↓ 1 callersFunctionbatch_get_key_points
**Output:** - center, open_point, upper_point, each of them is a numpy array of shape (2,)
util.py:291
↓ 1 callersMethodbatch_get_key_points
**Output:** - center, open_point, upper_point, each of them is a numpy array of shape (2,)
customgraspnetAPI/grasp.py:1067
↓ 1 callersFunctionbatch_key_point_2_rotation
**Input:** - centers_xyz: numpy array of the center points of shape (-1, 3). - open_points_xyz: numpy array of the open points of shape
util.py:164
↓ 1 callersFunctionbatch_key_point_2_rotation
**Input:** - centers_xyz: numpy array of the center points of shape (-1, 3). - open_points_xyz: numpy array of the open points of shape
customgraspnetAPI/utils/utils.py:857
↓ 1 callersFunctionbatch_key_points_2_tuple
**Input:** - key_points: np.array(-1,4,3) of grasp key points, definition is shown in key_points.png - scores: numpy array of batch gra
util.py:136
↓ 1 callersFunctionbatch_key_points_2_tuple
**Input:** - key_points: np.array(-1,4,3) of grasp key points, definition is shown in key_points.png - scores: numpy array of batch gra
customgraspnetAPI/utils/utils.py:713
↓ 1 callersFunctionbatch_viewpoint_params_to_matrix
**Input:** - towards: numpy array towards vectors with shape (n, 3). - angle: numpy array of in-plane rotations (n, ). **Output:**
customgraspnetAPI/utils/rotation.py:90
↓ 1 callersFunctioncalculate_6d_match
(pred_grasp: torch.Tensor, gg_ori_label, threshold_dis,
dataset/evaluation.py:247
↓ 1 callersFunctioncalculate_coverage
(pred_grasp: torch.Tensor, gg_ori_label: torch.Tensor, threshold
dataset/evaluation.py:294
↓ 1 callersFunctioncalculate_iou_match
(gs, gt_bbs, thre=0.25)
dataset/evaluation.py:333
↓ 1 callersFunctioncenter2dtopc
(rect_ggs: List, center_num, depths: torch.Tensor, output_s
dataset/pc_dataset_tools.py:132
↓ 1 callersMethodcheck_valid
Raise an exception if the config is missing required keys.
customgraspnetAPI/utils/dexnet/grasping/grasp_quality_config.py:87
↓ 1 callersFunctioncollision_detection
**Input:** - grasp_list: [(k1, 17), (k2, 17), ..., (kn, 17)] in camera coordinate - model_list: [(N1, 3), (N2, 3), ..., (Nn, 3)] in cam
customgraspnetAPI/utils/eval_utils.py:197
↓ 1 callersFunctioncompute_closest_points
**Input:** - A: (N, 3) - B: (M, 3) **Output:** - indices: (N,) closest point index in B for each point in A
customgraspnetAPI/utils/eval_utils.py:124
↓ 1 callersFunctioncompute_point_distance
**Input:** - A: (N, 3) - B: (M, 3) **Output:** - dists: (N, M)
customgraspnetAPI/utils/eval_utils.py:108
↓ 1 callersFunctioncompute_point_dists
Compute pair-wise point distances in two matrices. Input: A: [np.ndarray, (N,3), np.float32] point cloud A B: [np.nda
util.py:13
↓ 1 callersMethodconvex_hull
Return a 3D mesh that represents the convex hull of the mesh.
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:918
↓ 1 callersMethodcovariance
Return the total covariance of the mesh's triangles. Returns ------- float The total covariance of the mesh's tri
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:448
↓ 1 callersMethodcreate_config
Automatically create a quality config from a dictionary. Parameters ---------- config : :obj:`dict` dictionary ma
customgraspnetAPI/utils/dexnet/grasping/grasp_quality_config.py:192
↓ 1 callersFunctioncreate_table_cloud
Author: chenxi-wang **Input:** - width/height/depth: float, table width/height/depth along x/z/y-axis in meters - dx/dy/dz: float,
customgraspnetAPI/utils/vis.py:15
↓ 1 callersFunctioncreate_table_points
**Input:** - lx: - ly: - lz: **Output:** - numpy array of the points with shape (-1, 3).
customgraspnetAPI/utils/eval_utils.py:32
↓ 1 callersMethoddetect
(self, grasp_group, approach_dist=0.05)
dataset/collision_detector.py:48
↓ 1 callersMethoddist_to_point
Computes the distance from the segment to the given 3D point. Parameters ---------- point : :obj:`numpy.ndarray`
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1690
↓ 1 callersMethoddraw_gaussian
Convert GraspRectangles to the Location Map for training. :param rect_points: rect_points :return: Location Map
dataset/base_grasp_dataset.py:162
↓ 1 callersFunctiondraw_umich_gaussian
(heatmap, center, radius, k=1)
dataset/base_grasp_dataset.py:27
↓ 1 callersFunctionempty_pose_vector
(objectid)
customgraspnetAPI/utils/xmlhandler.py:124
↓ 1 callersFunctionempty_pose_vector_list
(objectidlist)
customgraspnetAPI/utils/xmlhandler.py:130
↓ 1 callersFunctioneval_grasp
**Input:** - grasp_group: GraspGroup instance for evaluation. - models: in model coordinate - dexnet_models: models in dexnet form
customgraspnetAPI/utils/eval_utils.py:333
↓ 1 callersMethodeval_scene_lr
(self, dump_folder, l, r, proc=2)
customgraspnetAPI/graspnet_eval.py:295
↓ 1 callersFunctionevaluate
()
test_graspnet.py:329
↓ 1 callersFunctionfast_sample
(num_samples, sample_size)
dataset/utils.py:177
↓ 1 callersFunctionfaster_get_local_labels
(eulers, thetas, label_eulers, label_offsets)
models/losses.py:142
↓ 1 callersMethodforce_closure
Checks force closure using the antipodality trick. Parameters ---------- c1 : :obj:`Contact3D` first contact poi
customgraspnetAPI/utils/dexnet/grasping/quality.py:253
↓ 1 callersMethodfrom_npy
**Input:** - npy_file_path: string of the file path.
customgraspnetAPI/grasp.py:482
↓ 1 callersFunctiongaussian2D
(shape, sigma=1)
dataset/base_grasp_dataset.py:18
← previousnext →101–200 of 616, ranked by callers