NVlabs / contact_graspnet

Efficient 6-DoF Grasp Generation in Cluttered Scenes

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Result different during inference

y556zhao opened this issue · comments

Hi,

I was using default parameters to test the model on 7.npy, and the resulting grasps are sparse as shown in the image below. May I ask if I was doing anything wrong?

Here is the command I used: python contact_graspnet/inference.py --np_path=test_data/7.npy --local_regions --filter_grasps

Checkpoint: checkpoints/scene_test_2048_bs3_hor_sigma_001

Thank you!
@MartinSmeyer
grasp

Hi,

This looks like issue #9

Apparently, users of newer GPUs require tensorflow-gpu=2.5 with CUDA >= 11 to run this code. Recompile the tf_ops if you use a newer version. Does this help?

I will update the requirements soon.

Best,

Martin

Yes, it works! Thank you so much for helping!