Skip to content

Open AI Gym Environment for the Dobot Magician Robotic Arm

Notifications You must be signed in to change notification settings

yucnet/gym-dobot

Repository files navigation

gym-dobot

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

Basic Installation

git clone https://github.com/WarrG3X/gym-dobot
cd gym-dobot
pip install -e .

Usage

from gym_dobot.envs import DobotPickAndPlaceEnv
env = DobotPickAndPlaceEnv()

Gym Installation

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.

Usage

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.

Test

# 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 

About

Open AI Gym Environment for the Dobot Magician Robotic Arm

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages