jingxixu/kitchen2d---stir

Learning for task and motion planning in a 2D kitchen.

Kitchen2D

This is a code repository of learning for task and motion planning in a 2D kitchen. This repository is associated with the paper Active model learning and diverse action sampling for task and motion planning. Our project page is here. If you have any question, please email me at ziw 'at' mit.edu, or create an issue here.

We developed our code building upon several existing packages:

• pybox2d, a 2D physics engine for Python based on the Box2D library;
• GPy, a Gaussian process framework in Python;
• motion-planners, including basic motion planning functionals such as rapidly-exploring random trees;
• pddlstream, lightweight implementation of STRIPStream, which builds upon the Fast Downward planner.
• numpy, version 1.13.3 or higher.
• scipy, version 0.19.1 or higher.
• sklearn, version 0.18.1 or higher.

In particular, motion-planners and pddlstream are included as submodules in this repository.

System Requirement

We tested our code with Python 2.7.6 on Ubuntu 14.04 LTS (64-bit) and Mac OS X. To install pybox2d, GPy and Fast Downward, follow the following steps.

1. Install numpy, scipy and sklearn, following the instructions here and here.

2. Follow the instructions here to install pybox2d.

3. To run the learning examples, follow the instructions here to install GPy.

4. To run the planning examples, follow the instructions here to obtain Fast Downward.

Quick Start

Once you confirm the system requirements are satisfied, make a copy of this repository with your favorite method, e.g.

git clone [email protected]:YOUR_USERNAME/Kitchen2D.git
cd Kitchen2D

Initialize and update the submodules by

git submodule init
git submodule update

Now you should be able to run the examples below.

Examples

Example of Primitives

An example of using the primitives is in primitive_example.py. Try

python primitive_example.py

Example of Learning

We show an example of both learning and sampling the scooping action in learn_example.py. We adopt an active learning view to learn the feasible region of the pre-conditions of the primitives. In order to plan with the learned pre-conditions, we need to be able to sample from its feasible regions. The detailed algorithm and setups we used can be found in Section IV.A of the accompanying paper. Try

python learn_example.py

Example of Planning

plan_example.py is an example of planning with learned pouring and scooping actions. We use STRIPSream as the backend planner. The goal of the task in plan_example.py is to “serve” a cup of coffee with cream and sugar by placing it on the green coaster near the edge of the table. Click here for vidoes of plans. Try

python plan_example.py

Citation

Please cite our work if you would like to use the code.

@article{wang2018active,
  title={Active model learning and diverse action sampling for task and motion planning},
  author={Wang, Zi and Garrett, Caelan Reed and Kaelbling, Leslie Pack and Lozano-P{\'e}rez, Tom{\'a}s},
  journal={arXiv preprint arXiv:1803.00967},
  year={2018}
}

References

• Wang Z, Garrett CR, Kaelbling LP, Lozano-Pérez T. Active model learning and diverse action sampling for task and motion planning. arXiv preprint arXiv:1803.00967. 2018 Mar 2.