Motion Before Action: Diffusing Object Motion as Manipulation Condition

Authors: Yue Su*, Xinyu Zhan*, Hongjie Fang, Yong-Lu Li, Cewu Lu, Lixin Yang†

🛫 Getting Started

💻 Installation

We use RISE as our real robot baseline, Please following the installation guide to install the mba conda environments and the dependencies, as well as the real robot environments. Also, remember to adjust the constant parameters in dataset/constants.py and utils/constants.py according to your own environment.

📷 Calibration

Please calibrate the camera(s) with the robot before data collection and evaluation to ensure correct spatial transformations between camera(s) and the robot. Please refer to calibration guide for more details.

🛢️ Data Collection

Data will be released soon.

🧑🏻‍💻 Training

The training scripts are saved in script.

conda activate mba
bash script/command_train__place__mba.sh # Add MBA module
bash script/command_train__place.sh # For baseline

🤖 Evaluation

Please follow the deployment guide to modify the evaluation script.

Modify the arguments in command_eval.sh, then

conda activate mba
bash command_eval.sh

✍️ Citation

@article{su2024motion,
  title={Motion before action: Diffusing object motion as manipulation condition},
  author={Su, Yue and Zhan, Xinyu and Fang, Hongjie and Li, Yong-Lu and Lu, Cewu and Yang, Lixin},
  journal={arXiv preprint arXiv:2411.09658},
  year={2024}
}

📃 License

MBA is licensed under CC BY-NC-SA 4.0