Skip to main content

robosuite: A Modular Simulation Framework and Benchmark for Robot Learning

Project description

robosuite

[Homepage][White Paper][Documentations][ARISE Initiative]


Latest Updates

  • [10/19/2021] v1.3: Ray tracing and physically based rendering tools :sparkles: and access to additional vision modalities 🎥 [video spotlight] [release notes]

  • [02/17/2021] v1.2: Added observable sensor models :eyes: and dynamics randomization :game_die: [release notes]

  • [12/17/2020] v1.1: Refactored infrastructure and standardized model classes for much easier environment prototyping :wrench: [release notes]


robosuite is a simulation framework powered by the MuJoCo physics engine for robot learning. It also offers a suite of benchmark environments for reproducible research. The current release (v1.3) features rendering tools, ground-truth of vision modalities, and camera utilities. This project is part of the broader Advancing Robot Intelligence through Simulated Environments (ARISE) Initiative, with the aim of lowering the barriers of entry for cutting-edge research at the intersection of AI and Robotics.

Data-driven algorithms, such as reinforcement learning and imitation learning, provide a powerful and generic tool in robotics. These learning paradigms, fueled by new advances in deep learning, have achieved some exciting successes in a variety of robot control problems. However, the challenges of reproducibility and the limited accessibility of robot hardware (especially during a pandemic) have impaired research progress. The overarching goal of robosuite is to provide researchers with:

  • a standardized set of benchmarking tasks for rigorous evaluation and algorithm development;
  • a modular design that offers great flexibility to design new robot simulation environments;
  • a high-quality implementation of robot controllers and off-the-shelf learning algorithms to lower the barriers to entry.

This framework was originally developed since late 2017 by researchers in Stanford Vision and Learning Lab (SVL) as an internal tool for robot learning research. Now it is actively maintained and used for robotics research projects in SVL and the UT Robot Perception and Learning Lab (RPL). We welcome community contributions to this project. For details please check out our contributing guidelines.

This release of robosuite contains seven robot models, eight gripper models, six controller modes, and nine standardized tasks. It also offers a modular design of APIs for building new environments with procedural generation. We highlight these primary features below:

  • standardized tasks: a set of standardized manipulation tasks of large diversity and varying complexity and RL benchmarking results for reproducible research;
  • procedural generation: modular APIs for programmatically creating new environments and new tasks as combinations of robot models, arenas, and parameterized 3D objects;
  • robot controllers: a selection of controller types to command the robots, such as joint-space velocity control, inverse kinematics control, operational space control, and 3D motion devices for teleoperation;
  • multi-modal sensors: heterogeneous types of sensory signals, including low-level physical states, RGB cameras, depth maps, and proprioception;
  • human demonstrations: utilities for collecting human demonstrations, replaying demonstration datasets, and leveraging demonstration data for learning. Check out our sister project robomimic;
  • photorealistic rendering: integration with advanced graphics tools that provide real-time photorealistic renderings of simulated scenes.

Citations

Please cite robosuite if you use this framework in your publications:

@inproceedings{robosuite2020,
  title={robosuite: A Modular Simulation Framework and Benchmark for Robot Learning},
  author={Yuke Zhu and Josiah Wong and Ajay Mandlekar and Roberto Mart\'{i}n-Mart\'{i}n},
  booktitle={arXiv preprint arXiv:2009.12293},
  year={2020}
}

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

robosuite-1.3.1.tar.gz (247.0 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

robosuite-1.3.1-py3-none-any.whl (360.3 kB view details)

Uploaded Python 3

File details

Details for the file robosuite-1.3.1.tar.gz.

File metadata

  • Download URL: robosuite-1.3.1.tar.gz
  • Upload date:
  • Size: 247.0 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.5.0 importlib_metadata/4.8.2 pkginfo/1.7.1 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.7.11

File hashes

Hashes for robosuite-1.3.1.tar.gz
Algorithm Hash digest
SHA256 da7c9cbfc54cf48aca351d5d27af0feda4c753d829c65bcf73167d95e94feb1e
MD5 bcfe8e1b8edc326b95c6041cd4454f7d
BLAKE2b-256 3392fbaafc939c34d6757d71574ba3d277a653b1a34136afa19475e5e28406c1

See more details on using hashes here.

File details

Details for the file robosuite-1.3.1-py3-none-any.whl.

File metadata

  • Download URL: robosuite-1.3.1-py3-none-any.whl
  • Upload date:
  • Size: 360.3 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.5.0 importlib_metadata/4.8.2 pkginfo/1.7.1 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.7.11

File hashes

Hashes for robosuite-1.3.1-py3-none-any.whl
Algorithm Hash digest
SHA256 3b4fef9de6d9deb4460194daf2cb43093479c2914e724f9ed40e6813ec0d22f7
MD5 bb6eecb8a5fa94e6b9db747c8fc96b37
BLAKE2b-256 4b5319a35973783affa328e7d84e28126c0afdf6e6ff4e714368d37d55fa8e5c

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page