DexMimicGen Datasets
收藏DexMimicGen 数据集概述
数据集简介
- 用于模拟环境和数据集的官方发布,服务于ICRA 2025论文“DexMimicGen: Automated Data Generation for Bimanual Dexterous Manipulation via Imitation Learning”。
- 数据集和环境的详细信息可在DexMimicGen官网查看。
- 论文可在arXiv上查看。
使用说明
-
需要安装最新版本的robosuite。 bash git clone https://github.com/ARISE-Initiative/robosuite pip install -e robosuite
-
克隆DexMimicGen仓库并安装。 bash git clone https://github.com/NVlabs/dexmimicgen.git cd dexmimicgen pip install -e .
环境信息
- 环境详细信息见environments.md。
数据集获取
-
数据集可在HuggingFace下载。
-
使用以下脚本下载数据集: bash python scripts/download_hf_datasets.py --path /path/to/save/datasets
-
默认保存路径为
./datasets。 -
播放数据集中的演示: bash python scripts/playback_datasets.py --dataset xxxxx.hdf5 --n 1
训练代码
-
提供了配置和训练代码,用于复现论文中的BC-RNN结果。
-
安装robomimic: bash git clone https://github.com/ARISE-Initiative/robomimic.git -b dexmimicgen cd robomimic pip install -e .
-
生成训练配置文件: bash cd dexmimicgen python scripts/generate_training_config.py --dataset_dir /path/to/datasets --config_dir /path/to/save/config --output_dir /path/to/save/output
-
运行训练脚本: bash cd robomimic python scripts/train.py --config /path/to/config
许可
- 代码遵循NVIDIA Source Code License。
- 数据集遵循CC-BY 4.0。
引用
- 如使用该代码,请引用DexMimicGen论文: bibtex @inproceedings{jiang2024dexmimicen, title = {DexMimicGen: Automated Data Generation for Bimanual Dexterous Manipulation via Imitation Learning}, author = {Jiang, Zhenyu and Xie, Yuqi and Lin, Kevin and Xu, Zhenjia and Wan, Weikang and Mandlekar, Ajay and Fan, Linxi and Zhu, Yuke}, booktitle = {2025 IEEE International Conference on Robotics and Automation (ICRA)}, year = {2025} }




