search

SonYeongGwang / dexnet3.0_pytorch

1 stars 1 forks source link

requirements (PyTorch version) #2

Open jiangSeu opened 4 months ago

jiangSeu commented 4 months ago

When running the code, I found that it doesn't work, likely due to an issue with the PyTorch version. Could you please provide the specific PyTorch version being used, and could you also provide the requirements?

SonYeongGwang commented 3 months ago

I remember this had been tested with CUDA 11.3. You may find how to install the required version in https://pytorch.org/get-started/previous-versions/

This repo provides a classification model (GQCNN) if one or more candidates are sampled from the user. Try to sample grasp candidates (several) on an object surface. In my case, I sampled the candidate using normal vectors from point cloud to find smooth surface. Then cropped the image (32x32) which has the grasp candidate at the center.