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
4.5k Jan 07, 2023
565 Dec 30, 2022
235 Dec 19, 2022
48 Nov 15, 2022
218 Oct 29, 2022
1 May 10, 2022
32 Nov 13, 2022
1 Feb 01, 2022
186 Dec 18, 2022
277 Dec 28, 2022
10 Feb 10, 2022
2 Sep 13, 2022
4.9k Dec 31, 2022
16 Nov 01, 2022
2 Mar 07, 2022
45 Jan 09, 2023
515 Dec 13, 2022
64 Dec 29, 2022
11 Nov 08, 2022
148 Dec 28, 2022