jsll / pytorch_6dof-graspnet

MIT License
151 stars 55 forks source link

Mid point computation maybe not correct? #21

Open patselle opened 1 year ago

patselle commented 1 year ago

Hi,

The mid point is obtained in Line 672 in utils.py. In order to compute this value the first two elements from the transformed control points are used (grasp_cps). According to get_control_point_tensor (line 282) the first two elements are always the same. Should the line 672 maybe replaced by (1 to -1)

mid = (grasp_cps[:, 0, :] + grasp_cps[:, -1, :]) / 2.0 ?