Simulation environments for basic manipulation tasks.
Project description
MuJoCo Robotics Environments 🤖
Software for creating and experimenting with robot workspaces in MuJoCo.
Installation
PyPI
Install from package from PyPI with:
pip install mujoco_robot_environments
Local Development
Clone this GitHub repository and ensure you have python version 3.10.6 using pyenv. Following this install the python virtual environment for this project with poetry through running:
poetry install
Activate the virtual environment with the following command:
poetry shell
Example Usage
Instantiating shipped environments
from mujoco_robot_environments.tasks.rearrangement import RearrangementEnv
env = RearrangementEnv(viewer=True)
env.reset()
...
Overriding Environment Configurations
Most environments ship with a default set of logic for scene initialization and other settings you may wish to override/customise.
In order to override the default config you need use the hydra override and follow the hierarchical structure of the configuration files seen shipped with the repository. As an example to change the sampling settings for the RearrangementEnv
one can follow the below syntax:
# clear hydra global state to avoid conflicts with other hydra instances
hydra.core.global_hydra.GlobalHydra.instance().clear()
# read hydra config
initialize(version_base=None, config_path=<relative path to config yaml files>, job_name="rearrangement")
# add task configs
COLOR_SEPARATING_CONFIG = compose(
config_name="rearrangement",
overrides=[
"arena/props=colour_splitter",
]
)
# instantiate color separation task
env = RearrangementEnv(viewer=True, cfg=COLOR_SEPARATING_CONFIG)
# expert demonstration
_, _, _, obs = env.reset()
Where in the above example <relative path to config yaml files>
points towards a directory of yaml files of the same structure as those shipped with this repository but containing your custom configuration files.
Task Environments
Colour Sorting Task Environment
An toy environment for reasoning about visual semantics through rearranging objects of varying shapes/colours into target locations.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for mujoco_robot_environments-0.0.9.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 49c31afb32ce98dd980c384f1f6147d46baed82579563328acb98719746ba9f5 |
|
MD5 | 7aeec779966a9fb0e9c8e9244fbf9787 |
|
BLAKE2b-256 | 3d982cb36710b972e5398f2c63a18615c892a9a1b61ff3c4f93f2da0e1ef4e9c |
Hashes for mujoco_robot_environments-0.0.9-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c1cb84c77d1c9aed8723c909770c2d843e65688f1eec7d39b33622bde22023de |
|
MD5 | 2bd07db8cba18a2252326a0144503e3a |
|
BLAKE2b-256 | 6dd5f8ea3758452316cd85701fa8dcc69ddf0c2fd905b9d3b45de49c40895ff1 |