Improve model card
Browse filesThis PR improves the model card for this Diffusion Policy baseline.
Specifically, it:
- Adds the `robotics` pipeline tag to the metadata.
- Includes direct links to the paper, project page, and GitHub repository.
- Organizes the content for better readability, including installation and rollout instructions.
- Adds the BibTeX citation for the RoboManipBaselines paper.
README.md
CHANGED
|
@@ -1,15 +1,22 @@
|
|
| 1 |
---
|
| 2 |
license: bsd-2-clause
|
|
|
|
| 3 |
---
|
| 4 |
-
# Diffusion Policy
|
| 5 |
|
| 6 |
-
|
| 7 |
|
| 8 |
-
|
| 9 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
|
| 11 |
-
## Policy rollout
|
| 12 |
-
Run a trained policy:
|
| 13 |
```console
|
| 14 |
# Go to the top directory of this repository
|
| 15 |
$ cd robo_manip_baselines
|
|
@@ -17,9 +24,10 @@ $ python ./bin/Rollout.py DiffusionPolicy MujocoUR5eCable --checkpoint ./checkpo
|
|
| 17 |
```
|
| 18 |
|
| 19 |
## Technical Details
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
|
|
|
| 23 |
author = {Chi, Cheng and Feng, Siyuan and Du, Yilun and Xu, Zhenjia and Cousineau, Eric and Burchfiel, Benjamin and Song, Shuran},
|
| 24 |
title = {Diffusion Policy: Visuomotor Policy Learning via Action Diffusion},
|
| 25 |
booktitle = {Proceedings of Robotics: Science and Systems},
|
|
@@ -29,5 +37,14 @@ For more information on the technical details, please see the following paper:
|
|
| 29 |
}
|
| 30 |
```
|
| 31 |
|
| 32 |
-
##
|
| 33 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: bsd-2-clause
|
| 3 |
+
pipeline_tag: robotics
|
| 4 |
---
|
|
|
|
| 5 |
|
| 6 |
+
# Diffusion Policy - RoboManipBaselines
|
| 7 |
|
| 8 |
+
This repository contains a Diffusion Policy model trained using the [RoboManipBaselines](https://github.com/isri-aist/RoboManipBaselines) framework. The model was trained with the [MujocoUR5eCable dataset](https://huggingface.co/datasets/RoboManipBaselines/MujocoUR5eCable).
|
| 9 |
+
|
| 10 |
+
- **Paper:** [RoboManipBaselines: A Unified Framework for Imitation Learning in Robotic Manipulation across Real and Simulation Environments](https://huggingface.co/papers/2509.17057)
|
| 11 |
+
- **Project Page:** [RoboManipBaselines Project Page](https://isri-aist.github.io/RoboManipBaselines-ProjectPage/)
|
| 12 |
+
- **Repository:** [GitHub - isri-aist/RoboManipBaselines](https://github.com/isri-aist/RoboManipBaselines)
|
| 13 |
+
|
| 14 |
+
## Installation
|
| 15 |
+
For installation instructions, please refer to the [Diffusion Policy installation guide](https://github.com/isri-aist/RoboManipBaselines/blob/master/doc/install.md#Diffusion-policy) in the official repository.
|
| 16 |
+
|
| 17 |
+
## Policy Rollout
|
| 18 |
+
To run the trained policy, use the following command from the top directory of the `robo_manip_baselines` repository:
|
| 19 |
|
|
|
|
|
|
|
| 20 |
```console
|
| 21 |
# Go to the top directory of this repository
|
| 22 |
$ cd robo_manip_baselines
|
|
|
|
| 24 |
```
|
| 25 |
|
| 26 |
## Technical Details
|
| 27 |
+
This model implements the Diffusion Policy as described in:
|
| 28 |
+
|
| 29 |
+
```bibtex
|
| 30 |
+
@INPROCEEDINGS{DiffusionPolicy_RSS23,
|
| 31 |
author = {Chi, Cheng and Feng, Siyuan and Du, Yilun and Xu, Zhenjia and Cousineau, Eric and Burchfiel, Benjamin and Song, Shuran},
|
| 32 |
title = {Diffusion Policy: Visuomotor Policy Learning via Action Diffusion},
|
| 33 |
booktitle = {Proceedings of Robotics: Science and Systems},
|
|
|
|
| 37 |
}
|
| 38 |
```
|
| 39 |
|
| 40 |
+
## Citation
|
| 41 |
+
If you use this model or the RoboManipBaselines framework in your research, please cite:
|
| 42 |
+
|
| 43 |
+
```bibtex
|
| 44 |
+
@article{RoboManipBaselines_Murooka_2025,
|
| 45 |
+
title={RoboManipBaselines: A Unified Framework for Imitation Learning in Robotic Manipulation across Real and Simulation Environments},
|
| 46 |
+
author={Murooka, Masaki and Motoda, Tomohiro and Nakajo, Ryoichi and Oh, Hanbit and Makihara, Koshi and Shirai, Keisuke and Ogata, Tetsuya and Domae, Yukiyasu},
|
| 47 |
+
journal={arXiv preprint arXiv:2509.17057},
|
| 48 |
+
year={2025}
|
| 49 |
+
}
|
| 50 |
+
```
|