MCPcopy Create free account

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

Functions616 in github.com/THU-VCLab/HGGD

↓ 1 callersFunctiongaussian2D
(shape, sigma=1)
dataset/utils.py:22
↓ 1 callersMethodgetObjIds
**Input:** - sceneIds: int or list of int of the scene ids. **Output:** - a list of int of the object ids in the g
customgraspnetAPI/graspnet.py:283
↓ 1 callersFunctionget_2d_key_points
**Output:** - center, open_point, upper_point, each of them is a numpy array of shape (2,)
dataset/graspnet_utils.py:125
↓ 1 callersMethodget_6d_width
(self)
dataset/grasp.py:662
↓ 1 callersMethodget_T_surface_obj
Gets the transformation that puts the object resting exactly on the z=delta plane. Parameters ---------- T_obj_surfac
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:836
↓ 1 callersMethodget_anchor_map
(self, rect_points, rect_gg: RectGraspGroup,
dataset/base_grasp_dataset.py:179
↓ 1 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
customgraspnetAPI/utils/utils.py:685
↓ 1 callersFunctionget_config
- return the config dict
customgraspnetAPI/utils/config.py:1
↓ 1 callersMethodget_depth
(self, index, rot=0, zoom=1.0)
dataset/base_grasp_dataset.py:79
↓ 1 callersFunctionget_grasp_score
(grasp, obj, fc_list, force_closure_quality_config)
customgraspnetAPI/utils/eval_utils.py:178
↓ 1 callersFunctionget_group_pc
(pc: torch.Tensor, local_centers: List, group_num, grasp_wi
dataset/pc_dataset_tools.py:53
↓ 1 callersMethodget_key_points
**Output:** - center, open_point, upper_point, each of them is a numpy array of shape (2,)
customgraspnetAPI/grasp.py:764
↓ 1 callersFunctionget_mat
Calls get_mat() to get the 4x4 matrix.
customgraspnetAPI/utils/trans3d.py:18
↓ 1 callersMethodget_mat_4x4
**Output:** - Convert self.x, self.y, self.z, self.alpha, self.beta and self.gamma into mat_4x4 pose
customgraspnetAPI/utils/pose.py:61
↓ 1 callersMethodget_model_poses
**Input:** - scene_id: int of the scen index. - ann_id: int of the annotation index. **Output:** - obj_li
customgraspnetAPI/graspnet_eval.py:63
↓ 1 callersFunctionget_obj_pose_list
(camera_pose, pose_vectors)
customgraspnetAPI/utils/utils.py:637
↓ 1 callersFunctionget_optimizer
(args, params)
train_utils.py:282
↓ 1 callersMethodget_quat
**Output:** - Convert self.alpha, self.beta, self.gamma into self.quat
customgraspnetAPI/utils/pose.py:51
↓ 1 callersMethodget_rgb
(self, index, rot=0, zoom=1.0)
dataset/base_grasp_dataset.py:121
↓ 1 callersMethodget_scene_models
return models in model coordinate.
customgraspnetAPI/graspnet_eval.py:33
↓ 1 callersMethodget_translation
**Output:** - Convert self.x, self.y, self.z into self.translation
customgraspnetAPI/utils/pose.py:43
↓ 1 callersFunctiongrasp2rect
(_6d_grasp)
convert_grasp.py:47
↓ 1 callersMethodgtbb_processing
(self, points, rot=0, zoom=1.0)
dataset/base_grasp_dataset.py:144
↓ 1 callersFunctioninference
()
test_graspnet.py:78
↓ 1 callersMethodiou
(self, rect_grasp, angle_threshold=np.pi / 6)
dataset/grasp.py:495
↓ 1 callersFunctionjit_detect_2d
(loc_map, local_max, anchor_clss, grid_points, theta_offset, depth_offset, width_offset, rot
dataset/evaluation.py:44
↓ 1 callersFunctionjit_detect_6d
(pred_gammas, pred_betas, offset, scores, local_centers, anchor_idxs, thetas, widths, k, sca
dataset/evaluation.py:158
↓ 1 callersFunctionjit_rotation_distance
(a, b)
models/losses.py:169
↓ 1 callersFunctionkey_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
dataset/graspnet_utils.py:148
↓ 1 callersFunctionkey_point_2_rotation
**Input:** - center_xyz: numpy array of the center point. - open_point_xyz: numpy array of the open point. - upper_point_xyz: nump
customgraspnetAPI/utils/utils.py:831
↓ 1 callersMethodloadDepth
(self, sceneId, annId=0)
dataset/base_grasp_dataset.py:362
↓ 1 callersMethodloadDepth
**Input:** - sceneId: int of the scene index. - camera: string of type of camera, 'realsense' or 'kinect' - annId:
customgraspnetAPI/graspnet.py:482
↓ 1 callersMethodloadGrasp
(self, sceneId, annId=0)
dataset/base_grasp_dataset.py:338
↓ 1 callersMethodloadMask
**Input:** - sceneId: int of the scene index. - camera: string of type of camera, 'realsense' or 'kinect' - annId:
customgraspnetAPI/graspnet.py:500
↓ 1 callersMethodloadRGB
(self, sceneId, annId=0)
dataset/base_grasp_dataset.py:355
↓ 1 callersMethodloadRGB
**Input:** - sceneId: int of the scene index. - camera: string of type of camera, 'realsense' or 'kinect' - annId:
customgraspnetAPI/graspnet.py:444
↓ 1 callersMethodloadWorkSpace
**Input:** - sceneId: int of the scene index. - camera: string of type of camera, 'realsense' or 'kinect' - annId:
customgraspnetAPI/graspnet.py:518
↓ 1 callersFunctionload_dexnet_model
**Input:** - data_path: path to load .obj & .sdf files **Output:** - obj: dexnet model
customgraspnetAPI/utils/eval_utils.py:73
↓ 1 callersFunctionlog_anchor_loss
(epoch, batch_idx, loss, anchor_loss, anchor_losses, batch_cnt)
train_utils.py:469
↓ 1 callersFunctionlog_and_save
(args, tb, results, epoch, anchornet,
train_utils.py:322
↓ 1 callersFunctionlog_match_result
(results, dis_criterion, rot_criterion)
train_utils.py:305
↓ 1 callersFunctionmask_grasp_group
(_6d_grasp_group, threshold)
util.py:241
↓ 1 callersFunctionmatrix_to_dexnet_params
Author: chenxi-wang **Input:** - numpy array of shape (3, 3) of the rotation matrix. **Output:** - binormal: numpy array of s
customgraspnetAPI/utils/utils.py:277
↓ 1 callersMethodmoment_arm
Computes the moment arm to a point x. Parameters ---------- x : 3x1 :obj:`numpy.ndarray` point to get moment arm
customgraspnetAPI/utils/dexnet/grasping/graspable_object.py:140
↓ 1 callersMethodnms
**Input:** - translation_thresh: float of the translation threshold. - rotation_thresh: float of the rotation threshold.
customgraspnetAPI/grasp.py:586
↓ 1 callersMethodnormalize_vertices
Normalize the mesh's orientation along its principal axes. Transforms the vertices and normals of the mesh such that the origin of th
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:547
↓ 1 callersFunctionoffset_reg_loss
Regression loss for offsets (z, w, theta) in Multi-Grasp Generator. :param regs: offset [B, anchor_k, H/r, W/r] :param gt_regs: ground truth
models/losses.py:41
↓ 1 callersFunctionparse_args
()
train_utils.py:28
↓ 1 callersMethodplot_rect_grasp_group
(self, rgb, numGrasp, shuffle=True)
dataset/grasp.py:750
↓ 1 callersFunctionpoints_2_2d_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
dataset/graspnet_utils.py:73
↓ 1 callersFunctionpos_quat_to_pose_4x4
pose = pos_quat_to_pose_4x4(pos, quat) Convert pos and quat into pose, 4x4 format Args: pos: length-3 position quat: length-4 quaternion
customgraspnetAPI/utils/trans3d.py:30
↓ 1 callersFunctionpose_4x4_to_pos_quat
Convert pose, 4x4 format into pos and quat. Args: pose: numpy array, 4x4 Returns: pos: length-3 position quat: le
customgraspnetAPI/utils/trans3d.py:49
↓ 1 callersFunctionpose_from_pose_vector
**Input:** - pose_vector: A list in the format of [id,x,y,z,alpha,beta,gamma] **Output:** - A pose class instance
customgraspnetAPI/utils/pose.py:72
↓ 1 callersFunctionpose_list_from_pose_vector_list
**Input:** - Pose vector list defined in xmlhandler.py **Output:** - list of poses.
customgraspnetAPI/utils/pose.py:91
↓ 1 callersFunctionprepare_torch_and_logger
(args, mode='train')
train_utils.py:228
↓ 1 callersMethodprincipal_dims
Returns the maximal span of the mesh's coordinates. The maximal span is the maximum coordinate value minus the minimal coordinate val
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:326
↓ 1 callersMethodquality
Compute grasp quality. Parameters ---------- grasp : :obj:`GraspableObject3D` grasp to quality quality on
customgraspnetAPI/utils/dexnet/grasping/grasp_quality_function.py:103
↓ 1 callersMethodread
Reads in the SDF file and returns a Sdf object. Returns ------- :obj:`Sdf` A Sdf created from the data in the fil
customgraspnetAPI/utils/dexnet/grasping/meshpy/sdf_file.py:56
↓ 1 callersFunctionrect2grasp
(rect_gg, widths_2d, center_z_depths, thetas_rad, gammas_rad, betas_rad)
convert_grasp.py:82
↓ 1 callersMethodremove
**Input:** - index: list of the index of grasp
customgraspnetAPI/grasp.py:471
↓ 1 callersFunctionremove_invisible_grasp_points
Remove invisible part of object model according to scene point cloud. Input: cloud: [np.ndarray, (N,3), np.float32] scene poi
util.py:32
↓ 1 callersMethodremove_unreferenced_vertices
Remove any vertices that are not part of a triangular face. Note ---- This method will fail if any bad triangles are present,
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:472
↓ 1 callersFunctionrun
()
train_graspnet.py:537
↓ 1 callersMethodscene_list
(self)
dataset/base_grasp_dataset.py:73
↓ 1 callersFunctionselect_2d_center
(loc_maps, center_num, reduce='max', grid_size=8)
dataset/pc_dataset_tools.py:295
↓ 1 callersFunctionselect_area
(loc_map, top, bottom, left, right, grid_size, overlap)
dataset/pc_dataset_tools.py:284
↓ 1 callersMethodsetaug
(self)
dataset/base_grasp_dataset.py:61
↓ 1 callersFunctionshift_anchors
(gg_labels, anchors, iter_times=1)
dataset/utils.py:144
↓ 1 callersMethodsort
(self)
dataset/grasp.py:407
↓ 1 callersMethodsurface_normal
Returns the sdf surface normal at the given coordinates by computing the tangent plane using SDF interpolation. Parameters --
customgraspnetAPI/utils/dexnet/grasping/meshpy/sdf.py:487
↓ 1 callersMethodsurface_window_projection
Projects the local surface onto the tangent plane at a contact point. Parameters ---------- width : float
customgraspnetAPI/utils/dexnet/grasping/contacts.py:513
↓ 1 callersFunctionto_grasp_group
**Input:** - camera: string of type of camera, 'kinect' or 'realsense'. - depths: numpy array of the depths image. - depth_method:
util.py:335
↓ 1 callersMethodto_open3d_geometry
**Input:** - color: optional, tuple of shape (3) denotes (r, g, b), e.g., (1,0,0) for red **Ouput:** - list of ope
customgraspnetAPI/grasp.py:211
↓ 1 callersFunctionto_open3d_geometry_list
**Output:** - list of open3d.geometry.Geometry of the grippers.
util.py:228
↓ 1 callersFunctionto_opencv_image
**input:** - opencv_rgb: numpy array of opencv BGR format. - numGrasp: int of the number of grasp, 0 for all. **Output:** - n
util.py:385
↓ 1 callersMethodto_opencv_image
**input:** - opencv_rgb: numpy array of opencv BGR format. **Output:** - numpy array of opencv RGB format that sho
customgraspnetAPI/grasp.py:734
↓ 1 callersMethodto_our_gripper
(self, color=None)
dataset/grasp.py:48
↓ 1 callersFunctionto_rect_grasp_group
**Input:** - camera: string of type of camera, 'realsense' or 'kinect'. **Output:** - RectGraspGroup instance or None.
util.py:198
↓ 1 callersMethodto_rect_grasp_group
(self, threshold=0, donms=False)
dataset/grasp.py:305
↓ 1 callersMethodtorques
Get the torques that can be applied by a set of force vectors at the contact point. Parameters ---------- forces : 3x
customgraspnetAPI/utils/dexnet/grasping/contacts.py:289
↓ 1 callersMethodtotal_volume
Return the total volume of the mesh. Returns ------- float The total volume of the mesh.
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:430
↓ 1 callersFunctiontrain
train one epoch. Args: epoch (int): epoch idx anchornet (nn.Module): anchornet (GHM) localnet (nn.Module): localnet (NMG)
train_graspnet.py:271
↓ 1 callersMethodtransform
Returns a new SDF transformed by similarity tf.
customgraspnetAPI/utils/dexnet/grasping/meshpy/sdf.py:103
↓ 1 callersMethodtri_centers
Returns an array of the triangle centers as 3D points. Returns ------- :obj:`numpy.ndarray` of :obj:`numpy.ndarray` of float
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:361
↓ 1 callersMethodtri_normals
Returns a list of the triangle normals. Parameters ---------- align_to_hull : bool If true, we re-orient the norm
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:376
↓ 1 callersFunctionvalidate
(epoch, anchornet: nn.Module, localnet: nn.Module, val_data: torch.utils.data.DataLoader, anchors
train_graspnet.py:34
↓ 1 callersFunctionvector_distance
(a: torch.Tensor, b: torch.Tensor)
dataset/utils.py:107
↓ 1 callersFunctionvisAnno
Author: chenxi-wang **Input:** - dataset_root: str, graspnet dataset root - scene_name: str, name of scene folder, e.g. scene_0000
customgraspnetAPI/utils/vis.py:68
↓ 1 callersFunctionvoxel_sample_points
**Input:** - points: (N, 3) **Output:** - points: (n, 3)
customgraspnetAPI/utils/eval_utils.py:141
↓ 1 callersMethodwidth_from_endpoints
Width of grasp from endpoints.
customgraspnetAPI/utils/dexnet/grasping/grasp.py:300
↓ 1 callersMethodwrite
Writes a Mesh3D object out to a .obj file format. Parameters ---------- mesh : :obj:`Mesh3D` The Mesh3D object to
customgraspnetAPI/utils/dexnet/grasping/meshpy/obj_file.py:115
MethodT_grasp_obj
Rigid transformation from grasp frame to object frame. Rotation matrix is X-axis along approach direction, Y axis pointing between the
customgraspnetAPI/utils/dexnet/grasping/grasp.py:374
MethodT_obj_table
(self)
customgraspnetAPI/utils/dexnet/grasping/meshpy/stable_pose.py:76
MethodT_obj_world
Return pose.
customgraspnetAPI/utils/dexnet/grasping/meshpy/mesh.py:1266
MethodT_obj_world
(self)
customgraspnetAPI/utils/dexnet/grasping/meshpy/stable_pose.py:82
Method__call__
(self, grasp)
customgraspnetAPI/utils/dexnet/grasping/grasp_quality_function.py:94
Method__copy__
Makes a copy of the config.
customgraspnetAPI/utils/dexnet/grasping/grasp_quality_config.py:127
Method__copy__
Makes a copy of the config.
customgraspnetAPI/utils/dexnet/grasping/grasp_quality_config.py:173
Method__eq__
Check equivalence by rotation about the z axis.
customgraspnetAPI/utils/dexnet/grasping/meshpy/stable_pose.py:59
Method__getattr__
(self, key)
customgraspnetAPI/utils/dexnet/grasping/grasp_quality_config.py:73
← previousnext →201–300 of 616, ranked by callers