graspnet / graspnetAPI

Toolbox for our GraspNet-1Billion dataset.
https://graspnet.net/index.html
203 stars 52 forks source link

您好,请问grasp_nms文件和nms_grasp方法有给出吗 #53

Closed biosbeta closed 5 months ago

biosbeta commented 5 months ago

nms抓取非极大值抑制的具体实现在项目文件中未找到
def nms(self, translation_thresh = 0.03, rotation_thresh = 30.0 / 180.0 * np.pi): ''' Input:

    - translation_thresh: float of the translation threshold.

    - rotation_thresh: float of the rotation threshold.

    **Output:**

    - GraspGroup instance after nms.
    '''
    from grasp_nms import nms_grasp
    return GraspGroup(nms_grasp(self.grasp_group_array, translation_thresh, rotation_thresh))