Skip to content

vikashplus/robohive

Repository files navigation

report report Documentation PyPI PyPI - License Downloads Open In Colab Slack

RoboHive Social Preview RoboHive is a collection of environments/tasks simulated with the MuJoCo physics engine exposed using the OpenAI-Gym API. Its compatible with any gym-compatible agents training framework (Stable Baselines, RLlib, TorchRL, AgentHive, etc)

Getting Started

Getting started with RoboHive is as simple as -

# Install RoboHive
pip install robohive
# Initialize RoboHive
robohive_init
# Demo an environment
python -m robohive.utils.examine_env -e FrankaReachRandom-v0

or, alternatively for editable installation -

# Clone RoboHive
git clone --recursive https://github.com/vikashplus/robohive.git; cd robohive
# Install (editable) RoboHive
pip install -e .
# Demo an environment
python -m robohive.utils.examine_env -e FrankaReachRandom-v0

See detailed installation instructions for options on mujoco-python-bindings and visual-encoders (R3M, RRL, VC), and frequently asked questions for more details.

Suites

RoboHive contains a variety of environments, which are organized as suites. Each suite is a collection of loosely related environments. The following suites are provided at the moment with plans to improve the diversity of the collection.

Hand-Manipulation-Suite (video)
Alt text A collection of environments centered around dexterous manipulation. Standard ADROIT benchmarks introduced in Learning Complex Dexterous Manipulation with Deep Reinforcement Learning and Demonstrations, RSS2018.) are a part of this suite
Arm-Manipulation-Suite
Alt text A collection of environments centered around Arm manipulation.
Myo-Suite (website)
Alt text A collection of environments centered around Musculoskeletal control.
Myo/MyoDM-Suite (Website)
myodm_task_suite A collection of musculoskeletal environments for dexterous manipulation introduced as MyoDM in MyoDeX.
MultiTask Suite
Alt text A collection of environments centered around multi-task. Standard RelayKitchen benchmarks are a part of this suite.

- TCDM Suite (WIP)

This suite contains a collection of environments centered around dexterous manipulation. Standard TCDM benchmarks are a part of this suite

- ROBEL Suite (Coming soon)

This suite contains a collection of environments centered around real-world locomotion and manipulation. Standard ROBEL benchmarks are a part of this suite

Citation

If you find RoboHive useful in your research,

  • please consider supporting the project by providing a star ⭐
  • please consider citing our project by using the following BibTeX entry:
@Misc{RoboHive2020,
  title = {RoboHive -- A Unified Framework for Robot Learning},
  howpublished = {\url{https://sites.google.com/view/robohive}},
  year = {2020},
  url = {https://sites.google.com/view/robohive},
}