#46 · Primary category: Robotics & Embodied AI Simulation
visual-pushing-grasping
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
genesis-world
Simulation platform for general-purpose robotics & embodied AI learning.
lerobot
🤗 LeRobot: Making AI for Robotics more accessible with end-to-end learning
ml-agents
The Unity Machine Learning Agents Toolkit (ML-Agents) is an open-source project that enables games and simulations to serve as environments for training intelligent agents using deep reinforcement learning and imitation learning.
AirSim
Open source simulator for autonomous vehicles built on Unreal Engine / Unity, from Microsoft AI & Research
carla
Open-source simulator for autonomous driving research.