Berkeley AUTOLAB’s Dex-Net Package¶
The dex-net package contains a Python API for managing HDF5 databases of 3D object models, parallel-jaw grasps, and grasp robustness metrics from used in the Dex-Net 2.0 paper. The package supports reading the Dex-Net 2.0 HDF5 Database containing parallel-jaw grasps for the ABB YuMi across 1,500 object models from the Dex-Net Object Mesh Dataset v1.1, and also supports the creation of databases with custom datasets of object models. The package is part of the Dexterity Network (Dex-Net) project and is maintained by the Berkeley AUTOLAB.
The goals of this project are to facilitate:
- Replicability of results from the Dex-Net 2.0 paper.
- Research extensions on using synthetic datasets of 3D object models with physics-based robustness metrics as a prior for robot grasping and manipulation.
Grasp robustness metrics from the dex-net are only known to predict success for lifting and transporting objects on a physical robot with:
- An ABB YuMi robot with a custom parallel-jaw gripper with silicone tips.
- Test objects from the Dex-Net 2.0 paper, which (a) have available grasps in every resting pose (e.g. not flat against the table), (b) weigh less than 0.25kg, the payload of the YuMi, and (c) are partially opaque to enable measurement with a depth sensor.
We are currently researching how to generate databases that can generalize across robots and additional objects. As of Jan 29, 2018, custom training datasets can be generated from existing databases using the script tools/generate_gqcnn_dataset.py. If you are interested in expanding the existing functionality, please email Jeff Mahler (firstname.lastname@example.org) with the subject line “Interested in GQ-CNN Dataset Generation.”
The package is currently under active development. Installation has been tested on Ubuntu 12.04, 14.04, and 16.04.
Please raise all bugs, feature requests, and other issues under the Github Issues. For other questions or concerns, please contact Jeff Mahler (email@example.com) with the subject line starting with “dex-net development: “
Active developers include:
- Jeff Mahler
- Alan Li
- Matt Matl
- Vishal Satish
- Jacky Liang
- Xinyu Liu
If you use the code, datasets, or models in a publication, please cite the Dex-Net 2.0 paper. If you use the 3D object models, then we recommend that you also cite the original sources of the data: