graspnet / graspnetAPI

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

in grasp_eval, when k+1 > len(score_list) #20

Closed cww97 closed 3 years ago

cww97 commented 3 years ago

https://github.com/graspnet/graspnetAPI/blob/master/graspnetAPI/graspnet_eval.py#L195

                    if k+1 > len(score_list):
                        grasp_accuracy[k,fric_idx] = np.sum(((score_list<=fric) & (score_list>0)).astype(int))/(k+1)

why the denominator here is (k+1) instead of the length of score_list, or should this grids removed from grasp_accuracy

GouMinghao commented 3 years ago

Thanks for your question, The denominator is (k+1) because we require the algorithm to output at least k grasp poses.
This line has the same result with adding several zero-score grasp poses until there are k grasp poses.