CMU Pushing dataset

Tool for downloading and using the pushing dataset. In this dataset, there are ~7500 push attempts on 8 simple block objects.

Getting started with the data

cd ~
git https://github.com/Dhiraj100892/pushing_dataset.git
cd pushing_dataset
wget https://www.dropbox.com/s/ftostshszsbpehm/data.tar.gz
tar -xvzf data.tar.gz
rm data.tar.gz

Visualize the data

python vis_data.py

Citation

@article{ye2019ocmpc,
  title={Object-centric Forward Modeling for Model Predictive Control},
  author={Ye, Yufei and  Gandhi, Dhiraj and Gupta, Abhinav and Tulsiani, Shubham},
  year={2019},
  booktitle={CoRL}
}