Open AI Gym Environment for the Dobot Magician Robotic Arm. Based on the fetch environments provided by gym.
Currently consists of -
- DobotPickAndPlaceEnv
- DobotPushEnv
- DobotReachEnv
- DobotClutterPickAndPlaceEnv
- DobotClutterPushEnv
The ClutterEnv variations add additonal blocks as clutter. The number of blocks can be changed by setting clutter_num
while creating the environment.(Max=40,Default=20)
Requires -
- python 3 (Tested on Python 3.6)
- gym > 0.10.3
- mujoco_py > 1.5
- mujoco - mjpro150
git clone https://github.com/WarrG3X/gym-dobot
cd gym-dobot
pip install -e .
from gym_dobot.envs import DobotPickAndPlaceEnv
env = DobotPickAndPlaceEnv()
First follow the steps for Basic Installation.
To access these environments from within your existing gym installation, they must first be registered.
If you have installed gym from their git repository then you can directly register the environments by simply applying the patch provided in this repository.
# Go to gym repository
cd gym
git apply /path/to/repo/gym-dobot/gym_install.patch
Otherwise, follow the instructions stated here.
import gym
env = gym.make("DobotPickAndPlaceEnv-v1")
This is much more convenient as other packages such as openai/baselines can be directly used with these dobot environments.
# Basic Test
python -m gym_dobot.run_env
#To see additional options
python -m gym_dobot.run_env --help
#Advanced Usage Example
python -m gym_dobot.run_env --env=DobotClutterPickAndPlaceEnv --steps=200 --clutter=40