OstrichRL
This is the repository accompanying the paper OstrichRL: A Musculoskeletal Ostrich Simulation to Study Bio-mechanical Locomotion.
It contains models of a musculoskeletal ostrich and of the Cassie robot, motion capture data, and tasks for the dm_control suite:
- ostrich-run
- ostrich-mocap_tracking
- ostrich-foraging
- ostrich-run_torque
- cassie-run
- cassie-mocap_tracking
Instructions
To install
Download and install OstrichRL:
git clone https://github.com/vittorione94/ostrichrl
pip install -e ostrichrl/
To launch experiments
We recommend using the Tonic RL library.
Download and install Tonic:
git clone https://github.com/fabiopardo/tonic
pip install -e tonic/
To play with the environments and random actions
python -m tonic.play --header "import ostrichrl" \
--env "tonic.environments.ControlSuite('ostrich-run')" \
--agent "tonic.agents.UniformRandom()"
python -m tonic.play --header "import ostrichrl" \
--env "tonic.environments.ControlSuite('ostrich-mocap_tracking', task_kwargs=dict(clip='cyclic', test=True, play=True))" \
--agent "tonic.agents.UniformRandom()"
python -m tonic.play --header "import ostrichrl" \
--env "tonic.environments.ControlSuite('ostrich-foraging')" \
--agent "tonic.agents.UniformRandom()"
python -m tonic.play --header "import ostrichrl" \
--env "tonic.environments.ControlSuite('ostrich-run_torque')" \
--agent "tonic.agents.UniformRandom()"
python -m tonic.play --header "import ostrichrl" \
--env "tonic.environments.ControlSuite('cassie-run')" \
--agent "tonic.agents.UniformRandom()"
python -m tonic.play --header "import ostrichrl" \
--env "tonic.environments.ControlSuite('cassie-mocap_tracking', task_kwargs=dict(clip='0100', test=True, play=True))" \
--agent "tonic.agents.UniformRandom()"
To train
python -m tonic.train --header "import ostrichrl, tonic.tensorflow" \
--env "tonic.environments.ControlSuite('ostrich-mocap_tracking', task_kwargs=dict(clip='0047'))" \
--test_env "tonic.environments.ControlSuite('ostrich-mocap_tracking', task_kwargs=dict(clip='0047', test=True))" \
--agent "tonic.tensorflow.agents.TD4()" \
--parallel 8 \
--sequential 5
2 May 19, 2022
110 Dec 20, 2022
209 Jan 05, 2023
37 Dec 27, 2022
4 Nov 14, 2022
3 Feb 25, 2022
59 Dec 26, 2022
253 Dec 07, 2022
124 Nov 06, 2022
57 Nov 21, 2022
5.2k Jan 05, 2023
2 Jul 16, 2022
65 Dec 01, 2022
2 Dec 13, 2021
2 Nov 21, 2021
2 Dec 21, 2021
1.3k Dec 30, 2022
24 Dec 20, 2022
11 Nov 30, 2022
29 Nov 16, 2022