#46 · Primary category: Robotics & Embodied AI Simulation

visual-pushing-grasping

3d artificial-intelligence computer-vision deep-learning deep-reinforcement-learning grasping manipulation pushing robotics vision

Train robotic agents to learn to plan pushing and grasping actions for manipulation with deep reinforcement learning.

Project last updated:05/11/21

GitHub Stars

1.1K

Forks

328

Contributors

3

License

BSD-2-Clause

Why we included this project

For roboticists working on pick-and-place, this is the PyTorch reference implementation behind an IROS 2018 paper on learning pushing and grasping together. Instead of hand-coded control logic, two fully convolutional networks take RGB-D images and learn through trial and error in a Q-learning framework, so a UR5 arm gradually figures out how to push cluttered objects into clearer positions and then grasp them. The repo includes training and testing scripts that run both in simulation and on real hardware, which makes it a practical entry point for experimenting with self-supervised manipulation. Because the published paper includes its results and the code matches it, the project also works as a reproducible baseline for comparing a new learning-based manipulation approach.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category