- Python 3
- PyTorch 1.8
- Open3d 0.8
- TensorBoard 2.3
- NumPy
- SciPy
- Pillow
- tqdm
Install graspnetAPI for using GraspNet-1Billion dataset.
git clone https://github.com/graspnet/graspnetAPI.git
cd graspnetAPI
pip install .
python vis_graspability.py
Training and testing code will be available upon paper acceptance.