/RL-robot-arm3D

3-dimensional spatial robotic arm control using reinforcement learning algorithms

Primary LanguagePythonMIT LicenseMIT

RL_robot_arm3D

3-dimensional spatial robotic arm control using reinforcement learning algorithms

If it helps you with your research, please star it. 🎈

Requirements and Dependencies

  • pytorch 1.9.0
  • python 3.6

Run

-Training run train.py

-Visualization interface run vis_draw.py (Use the right mouse button to click on the x-y screen and the x-z screen)

Training process

Reinforcement learning algorithms

  • [√] PPO

Control interface visualization