NVlabs / contact_graspnet

Efficient 6-DoF Grasp Generation in Cluttered Scenes
Other
324 stars 111 forks source link

Issue fo using my own data #31

Closed henry890112 closed 1 year ago

henry890112 commented 1 year ago

Hello, I am using my own data to train contact graspnet. However, I am unable to obtain thicker grasping poses in the output. Could this be due to a problem with my data or do I need to modify the test_data/*.npy files? I have attached my input and output data for reference. Thank you!

Screenshot from 2023-02-14 17-01-42

Screenshot from 2023-02-14 17-03-33

Screenshot from 2023-02-14 16-54-49

Screenshot from 2023-02-14 16-55-06

MartinSmeyer commented 1 year ago

Hey @henry890112,

What robot / gripper model do you use? If your gripper width is wider than the 8cm of the Franka Panda Gripper, then check #28

MartinSmeyer commented 1 year ago

In general thicker grasping poses are a bit less confident, because they are less certain to result in successful grasps. To force such predictions, you can lower the confidence threshold and filter for grasps with higher predicted grasp widths.

henry890112 commented 1 year ago

I apologize for the poor description of my issue, but my question is: I am unable to achieve the grasping pose that is highlighted in green, just like which I have attached below. Thank you for your patience. image

MartinSmeyer commented 1 year ago

Sorry still unclear, that is not a question

henry890112 commented 1 year ago

in my own data I can't get the grasping pose that is highlighted in green, is this Highest Scoring Gripping Pose?and why I can't get it in my own data. Thank you!

MartinSmeyer commented 1 year ago

Yes this is the highest scoring grasp pose. Ah ok, it's just about the visualization, the most confident grasps are only shown if you have more than one object in your mask image:

https://github.com/NVlabs/contact_graspnet/blob/a12b7e3a1d7adf38762bfae1c8b84b4550059a6f/contact_graspnet/visualization_utils.py#L99-L105

henry890112 commented 1 year ago

Thank for your answer!