BimanGrasp-Dataset
收藏BimanGrasp-Dataset
简介
BimanGrasp-Dataset 是一个大规模的合成数据集,包含一对影子机器人手抓取各种物体的数据。所有抓取动作均通过 Isaac Gym 模拟器验证,并通过了渗透测试(阈值:1.5 mm)。该数据集提供了抓取姿态数据、物体网格和其他资产。所有抓取动作均可使用 plotly 包进行可视化。
安装
建议使用 conda/mamba 环境来加载和可视化双手抓取姿态。
环境设置
bash conda create -n bimangrasp python=3.7 conda activate bimangrasp
依赖安装
bash conda install pytorch==1.12.1 torchvision==0.13.1 cudatoolkit=11.3 -c pytorch pip install trimesh plotly numpy argparse transforms3d
第三方包安装
bash cd third_party && wget https://github.com/facebookresearch/pytorch3d/archive/refs/tags/v0.7.1.tar.gz && tar -xzvf v0.7.1.tar.gz && cd v0.7.1.tar.gz && pip install -e. && cd ../.. cd third_party/pytorch_kinematics && pip install -e . cd ../third_party/torchSDF && pip install -e . && cd ../..
可视化
首先,从 https://github.com/Tsunami-kun/BimanGrasp-Dataset/releases 下载数据集和物体网格,并将其解压到仓库的根目录。
可视化命令
bash python visualization.py --object_name <object_name> --num <num>
示例
可视化示例(包括2D截图和3D HTML可视化)位于 examples/ 目录中。
致谢
感谢以下仓库的作者,我们在代码中参考了他们的工作:
引用
如果您在研究中使用了此仓库,请引用以下文献:
@article{shao2024bimanual, title={Bimanual Grasp Synthesis for Dexterous Robot Hands}, author={Shao, Yanming and Xiao, Chenxi}, journal={IEEE Robotics and Automation Letters}, year={2024}, publisher={IEEE} }




