MCPcopy Create free account

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

Functions616 in github.com/THU-VCLab/HGGD

↓ 161 callersMethodappend
(self, grasp: Grasp)
dataset/grasp.py:292
↓ 41 callersMethodload
Load a mesh from a file. Note ---- If the mesh is not already in .obj format, this requires the installation of meshl
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1204
↓ 24 callersMethodcopy
Return a copy of the mesh. Note ---- This method only copies the vertices and triangles of the mesh.
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:692
↓ 20 callersMethodtransform_pt_obj_to_grid
Transforms points from world frame to grid frame.
customgraspnetAPI/utils/dexnet/grasping/meshpy/sdf.py:108
↓ 18 callersMethodkeys
(self)
customgraspnetAPI/utils/dexnet/grasping/grasp_quality_config.py:83
↓ 14 callersFunction_isArrayLike
(obj)
customgraspnetAPI/graspnet.py:70
↓ 14 callersMethodcreate_line_of_action
Creates a line of action, or the points in space that the grasp traces out, from a point g in world coordinates on an object. Returns
customgraspnetAPI/utils/dexnet/grasping/grasp.py:102
↓ 14 callersMethodfind_contact
Find the point at which a point traveling along a given line of action hits a surface. Parameters ---------- line_of_
customgraspnetAPI/utils/dexnet/grasping/grasp.py:906
↓ 12 callersFunctioncreate_mesh_box
Author: chenxi-wang Create box instance with mesh representation.
customgraspnetAPI/utils/utils.py:413
↓ 10 callersMethodeval
(self)
dataset/base_grasp_dataset.py:67
↓ 10 callersFunctionget_camera_intrinsic
(camera=camera)
dataset/config.py:10
↓ 10 callersMethodgradient
Returns the SDF gradient at the given coordinates, interpolating if necessary. Parameters ---------- coords : :obj:`n
customgraspnetAPI/utils/dexnet/grasping/meshpy/sdf.py:381
↓ 9 callersMethodgetposevectorlist
(self)
customgraspnetAPI/utils/xmlhandler.py:96
↓ 9 callersFunctiontransform_points
Author: chenxi-wang **Input:** - points: numpy array of (N,3), point cloud - trans: numpy array of (4,4), transformation matrix
customgraspnetAPI/utils/utils.py:365
↓ 8 callersMethodgrasp_matrix
Computes the grasp map between contact forces and wrenchs on the object in its reference frame. Parameters ----------
customgraspnetAPI/utils/dexnet/grasping/quality.py:194
↓ 7 callersMethodis_out_of_bounds
Returns True if coords is an out of bounds access. Parameters ---------- coords : :obj:`numpy.ndarray` of int A 2
customgraspnetAPI/utils/dexnet/grasping/meshpy/sdf.py:177
↓ 7 callersFunctionparse_posevector
Author: chenxi-wang Decode posevector to object id and transformation matrix.
customgraspnetAPI/utils/utils.py:397
↓ 6 callersMethod__repr__
(self)
customgraspnetAPI/grasp.py:52
↓ 6 callersFunctionpixel_depth_2_points
**Input:** - pixel_x: numpy array of int of the pixel x coordinate. shape: (-1,) - pixel_y: numpy array of int of the pixle y coordicat
dataset/graspnet_utils.py:101
↓ 6 callersMethodsize
(self)
dataset/grasp.py:229
↓ 6 callersMethodto_6d_grasp_group
(self, depth=0.025)
dataset/grasp.py:682
↓ 6 callersMethodto_tensor
(cls, s)
dataset/base_grasp_dataset.py:261
↓ 5 callersFunctionbatch_rgbdxyz_2_rgbxy_depth
**Input:** - points: np.array(-1,3) of the points in camera frame - camera: string of the camera type **Output:** - coords: f
util.py:77
↓ 5 callersMethodconfiguration_from_params
Convert param list to a configuration vector for the class.
customgraspnetAPI/utils/dexnet/grasping/grasp.py:83
↓ 5 callersFunctionconv_with_dim_reshape
(in_dim, mid_dim, out_dim, bias=True)
models/anchornet.py:134
↓ 5 callersFunctionget_scene_name
**Input:** - num: int of the scene number. **Output:** - string of the scene name.
customgraspnetAPI/utils/eval_utils.py:21
↓ 5 callersMethodmin_norm_vector_in_facet
Finds the minimum norm point in the convex hull of a given facet (aka simplex) by solving a QP. Parameters ----------
customgraspnetAPI/utils/dexnet/grasping/quality.py:882
↓ 5 callersMethodparallel_eval_scenes
**Input:** - scene_ids: list of int of scene index. - dump_folder: string of the folder that saves the npy files.
customgraspnetAPI/graspnet_eval.py:265
↓ 5 callersMethodplot
(self, ax, color=None)
dataset/grasp.py:490
↓ 5 callersFunctionpoints_2_pixel_depth
**Input:** - points: np.array(-1,3) of the points in camera frame **Output:** - coords: float of xy in pixel frame [-1, 2] -
dataset/graspnet_utils.py:14
↓ 5 callersMethodtangents
Returns the direction vector and tangent vectors at a contact point. The direction vector defaults to the *inward-facing* normal vector at
customgraspnetAPI/utils/dexnet/grasping/contacts.py:119
↓ 4 callersMethod__init__
(self, in_dim, planes=16, mode='34')
models/anchornet.py:48
↓ 4 callersMethod__len__
**Output:** - int of the length.
customgraspnetAPI/grasp.py:846
↓ 4 callersMethod_make_layer
make resnet layers(by layer i didnt mean this 'layer' was the same as a neuron netowork layer, ex. conv layer), one layer may
models/resnet_model.py:124
↓ 4 callersMethod_proj_point_to_plane
Project the given point onto the plane containing the three points in tri_verts. Parameters ---------- tri_verts : :o
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1839
↓ 4 callersFunctionanchor_output_process
(loc_map, cls_mask, theta_offset,
dataset/evaluation.py:19
↓ 4 callersFunctionbatch_framexy_depth_2_xyz
**Input:** - pixel_x: numpy array of int of the pixel x coordinate. shape: (-1,) - pixel_y: numpy array of int of the pixle y coordicat
util.py:266
↓ 4 callersFunctionbatch_rgbdxyz_2_rgbxy_depth
**Input:** - points: np.array(-1,3) of the points in camera frame - camera: string of the camera type **Output:** - coords: f
customgraspnetAPI/utils/utils.py:654
↓ 4 callersFunctiondata_process
(points: torch.Tensor, depths: torch.Tensor, rect_ggs: List,
dataset/pc_dataset_tools.py:349
↓ 4 callersFunctiondetect_2d_grasp
detect 2d grasp from GHM heatmaps. Args: loc_map (Tensor): B x 1 x H x W cls_mask (Tensor): B x anchor_k x H_r x W_r thet
dataset/evaluation.py:87
↓ 4 callersFunctioneuclid_distance
(points1: torch.Tensor, points2: torch.Tensor)
dataset/utils.py:50
↓ 4 callersFunctionfeature_fusion
(points, perpoint_features, xyzs, mode='knn')
dataset/pc_dataset_tools.py:29
↓ 4 callersMethodloadScenePointCloud
**Input:** - sceneId: int of the scene index. - camera: string of type of camera, 'realsense' or 'kinect' - annId:
customgraspnetAPI/graspnet.py:541
↓ 4 callersMethodmax_coords
Returns the maximum coordinates of the mesh. Returns ------- :obj:`numpy.ndarray` of float A 3-ndarray of floats
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:271
↓ 4 callersMethodmin_coords
Returns the minimum coordinates of the mesh. Returns ------- :obj:`numpy.ndarray` of float A 3-ndarray of floats
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:260
↓ 4 callersMethodplot_friction_cone
(self, color='y', scale=1.0)
customgraspnetAPI/utils/dexnet/grasping/contacts.py:726
↓ 4 callersMethodstep
(self, epoch=None)
models/anchornet.py:38
↓ 4 callersMethodto_scene_points
(self, rgbs: torch.Tensor, depths: torch.Tensor,
demo.py:83
↓ 4 callersMethodto_xyz_maps
(self, depths)
demo.py:122
↓ 4 callersMethodtransform_pt_grid_to_obj
Transforms points from grid frame to world frame.
customgraspnetAPI/utils/dexnet/grasping/meshpy/sdf.py:113
↓ 3 callersMethod__len__
**Output:** - int of the length.
customgraspnetAPI/grasp.py:251
↓ 3 callersMethod_area_of_tri
Return the area of the given triangle. Parameters ---------- tri : :obj:`numpy.ndarray` of int The triangle for w
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1413
↓ 3 callersMethod_compute_basis
Computes axes for a transformed basis relative to the plane in which input vertices lie. Parameters ---------- face_v
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1527
↓ 3 callersMethod_compute_surface_window_projection
Compute the projection window onto the basis defined by u1 and u2. Params: u1, u2 - orthogonal numpy 3 arrays width -
customgraspnetAPI/utils/dexnet/grasping/contacts.py:352
↓ 3 callersMethod_get_rotation_matrix_y
(theta)
customgraspnetAPI/utils/dexnet/grasping/grasp.py:391
↓ 3 callersMethodadd
**Input:** - element: Grasp instance or GraspGroup instance.
customgraspnetAPI/grasp.py:454
↓ 3 callersFunctionangle_distance
(a: torch.Tensor, b: torch.Tensor)
dataset/utils.py:84
↓ 3 callersFunctionbatch_framexy_depth_2_xyz
**Input:** - pixel_x: numpy array of int of the pixel x coordinate. shape: (-1,) - pixel_y: numpy array of int of the pixle y coordicat
customgraspnetAPI/utils/utils.py:766
↓ 3 callersMethodbounding_box
Returns the mesh's bounding box corners. Returns ------- :obj:`tuple` of :obj:`numpy.ndarray` of float A 2-tuple
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:282
↓ 3 callersMethodcenter_vertices_bb
Center the mesh's vertices at the center of its bounding box. This shifts the mesh without rotating it so that the center of its boun
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:527
↓ 3 callersFunctioncollision_detect
(points_all: torch.Tensor, pred_gg, mode='regnet')
dataset/evaluation.py:343
↓ 3 callersFunctionconvert_2d_to_3d
(x, y, d)
dataset/utils.py:10
↓ 3 callersFunctiondetect_6d_grasp_multi
(rect_gg: RectGraspGroup, pred_view: torch.Tensor, offset:
dataset/evaluation.py:191
↓ 3 callersMethodfind_zero_crossing_quadratic
Find zero crossing using quadratic approximation along 1d line.
customgraspnetAPI/utils/dexnet/grasping/meshpy/sdf.py:776
↓ 3 callersFunctionfocal_loss
(pred, targets, thres=0.99, alpha=0.5, gamma=2,
models/losses.py:14
↓ 3 callersFunctionframexy_depth_2_xyz
**Input:** - pixel_x: int of the pixel x coordinate. - pixel_y: int of the pixle y coordicate. - depth: float of depth. The unit i
customgraspnetAPI/utils/utils.py:741
↓ 3 callersFunctiongenerate_scene_model
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:92
↓ 3 callersFunctiongenerate_views
Author: chenxi-wang View sampling on a sphere using Febonacci lattices. **Input:** - N: int, number of viewpoints. - phi: float, c
customgraspnetAPI/utils/utils.py:62
↓ 3 callersFunctionget_camera_intrinsic
**Input:** - camera: string of type of camera, "realsense" or "kinect". **Output:** - numpy array of shape (3, 3) of the camera in
customgraspnetAPI/utils/utils.py:27
↓ 3 callersFunctionget_camera_parameters
author: Minghao Gou **Input:** - camera: string of type of camera: 'kinect' or 'realsense' **Output:** - open3d.camera.Pinhol
customgraspnetAPI/utils/vis.py:45
↓ 3 callersFunctionjit_euler2quaternion
(euler)
models/losses.py:152
↓ 3 callersMethodloadCollisionLabels
**Input:** - sceneIds: int or list of int of the scene ids. **Output:** - dict of the collision labels.
customgraspnetAPI/graspnet.py:415
↓ 3 callersMethodloadGrasp
**Input:** - sceneId: int of scene id. - annId: int of annotation id. - format: string of grasp format, '6d' or 'r
customgraspnetAPI/graspnet.py:689
↓ 3 callersMethodloadGraspLabels
**Input:** - objIds: int or list of int of the object ids. **Output:** - a dict of grasplabels of each object.
customgraspnetAPI/graspnet.py:333
↓ 3 callersMethodload_from_dict
(self, grasp_dict, min_width=0, min_score=0.8)
dataset/grasp.py:626
↓ 3 callersMethodload_grasp_labels
(self, index)
dataset/base_grasp_dataset.py:328
↓ 3 callersFunctionlog_acc_str
(name, T, F)
train_utils.py:21
↓ 3 callersMethodnms
(self)
dataset/grasp.py:411
↓ 3 callersMethodon_surface
Determines whether or not a point is on the object surface. Parameters ---------- coords : :obj:`numpy.ndarray` of int
customgraspnetAPI/utils/dexnet/grasping/meshpy/sdf.py:157
↓ 3 callersFunctionplot_gripper_pro_max
Author: chenxi-wang **Input:** - center: numpy array of (3,), target point as gripper center - R: numpy array of (3,3), rotation m
customgraspnetAPI/utils/utils.py:494
↓ 3 callersMethodsurface_points
Returns the points on the surface. Returns ------- :obj:`tuple` of :obj:`numpy.ndarray` of int, :obj:`numpy.ndarray` of float
customgraspnetAPI/utils/dexnet/grasping/meshpy/sdf.py:118
↓ 3 callersMethodtrain
(self)
dataset/base_grasp_dataset.py:70
↓ 3 callersMethodtransform_pt_grid_to_obj
Converts a point in grid coords to the world basis. If direction then don't translate. Parameters ---------- x_grid :
customgraspnetAPI/utils/dexnet/grasping/meshpy/sdf.py:737
↓ 3 callersMethodunaug
(self)
dataset/base_grasp_dataset.py:64
↓ 2 callersMethod__init__
(self, config)
customgraspnetAPI/utils/dexnet/grasping/grasp_quality_config.py:59
↓ 2 callersMethod__init__
(self, quality, uncertainty=0.0, quality_config=None)
customgraspnetAPI/utils/dexnet/grasping/grasp_quality_function.py:61
↓ 2 callersMethod__init__
(self, block, num_blocks, in_dim=3, planes=64)
models/resnet_model.py:104
↓ 2 callersMethod__repr__
(self)
dataset/grasp.py:39
↓ 2 callersMethod_compute_bb_center
Computes the center point of the bounding box. Returns ------- :obj:`numpy.ndarray` of float 3-ndarray of floats
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1303
↓ 2 callersMethod_compute_centroid
Computes the centroid (mean) of the mesh's vertices. Returns ------- :obj:`numpy.ndarray` of float 3-ndarray of f
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1339
↓ 2 callersMethod_compute_face_dag
Computes a directed acyclic graph (DAG) specifying the toppling structure of the mesh faces by: 1) Computing the mesh convex hull
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1578
↓ 2 callersMethod_point_in_tri
Is the given point contained in the given triangle? Parameters ---------- tri_verts : :obj:`list` of :obj:`numpy.ndarray` of
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1866
↓ 2 callersMethodappend
(self, grasp: RectGrasp)
dataset/grasp.py:649
↓ 2 callersMethodaxis_from_endpoints
Normalized axis of grasp from endpoints as np 3-arrays.
customgraspnetAPI/utils/dexnet/grasping/grasp.py:292
↓ 2 callersMethodcenter_from_endpoints
Grasp center from endpoints as np 3-arrays.
customgraspnetAPI/utils/dexnet/grasping/grasp.py:286
↓ 2 callersMethodclose_fingers
Finds the contact points by closing on the given object. Parameters ---------- obj : :obj:`GraspableObject3D` obj
customgraspnetAPI/utils/dexnet/grasping/grasp.py:57
↓ 2 callersFunctioncompute_anchor_loss
Compute Anchor Loss. :param pred: A tuple (Location Map, Cls Mask, Offset theta, Offset h, Offset w) :param target: ground truth :param r
models/losses.py:57
↓ 2 callersFunctioncompute_multicls_loss
(pred, offset, gg_labels: list,
models/losses.py:85
↓ 2 callersMethodcontains
Checks whether or not the key is supported.
customgraspnetAPI/utils/dexnet/grasping/grasp_quality_config.py:67
↓ 2 callersMethodfriction_cone
Computes the friction cone and normal for a contact point. Parameters ---------- num_cone_faces : int number of c
customgraspnetAPI/utils/dexnet/grasping/contacts.py:230
next →1–100 of 616, ranked by callers