#55 · Primary category: Robotics & Embodied AI Simulation

MuJoCo_RL_UR5

computer-vision gym-environment mujoco pick-and-place reinforcement-learning robotics

A MuJoCo/Gym environment for robot control using Reinforcement Learning. The task of agents in this environment is pixel-wise prediction of grasp success chances.

Project last updated:11/21/22

GitHub Stars

685

Forks

70

Contributors

1

License

MIT

Why we included this project

Robotic pick and place is usually trained on joint states, but this repository sets up the harder version: a MuJoCo Gym environment where the agent sees a camera image instead of a state vector, wrapping a UR5 arm and Robotiq gripper. That visual observation brings the setup closer to a real deployment, and the multi-discrete action space with a binary grasp-success reward is a concrete example of structuring a difficult manipulation task. The MJ_Controller class is useful on its own for basic arm control and for resetting the robot between episodes. This is a thesis companion project and no longer actively maintained, so treat the code as a teaching example or starting point rather than production software. For anyone prototyping grasp prediction or pick-and-place RL, it is a handy reference.

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