├── .gitignore ├── .travis.yml ├── LICENSE ├── README.md ├── examples └── train_quadrotor2d_ppo2.py ├── gym_reinmav ├── __init__.py ├── config.yaml ├── controller │ ├── __init__.py │ ├── quat_pid_controller.py │ └── rpy_pid_controller.py ├── envs │ ├── mujoco │ │ ├── __init__.py │ │ ├── assets │ │ │ ├── quadrotor_ground.xml │ │ │ ├── quadrotor_hovering.xml │ │ │ └── quadrotor_quat.xml │ │ ├── mujoco_quad.py │ │ ├── mujoco_quad_hovering.py │ │ └── mujoco_quad_quat.py │ └── native │ │ ├── __init__.py │ │ ├── quadrotor2d.py │ │ ├── quadrotor2d_slungload.py │ │ ├── quadrotor3d.py │ │ ├── quadrotor3d_slungload.py │ │ └── reinmav_env.py ├── example │ ├── __init__.py │ └── mujoco │ │ ├── __init__.py │ │ ├── control_quat.py │ │ ├── control_rpy.py │ │ └── train_hovering.py ├── plots │ ├── position_x.png │ ├── vel_x.png │ └── yaw.png ├── resources │ ├── mujoco_quad.gif │ ├── native_slungload.gif │ └── reinmav-gym-mujoco.gif ├── run.py └── wrapper │ └── __init__.py ├── requirements.txt ├── setup.py └── test ├── test_quadrotor2d.py ├── test_quadrotor2dslungload.py ├── test_quadrotor3d.py ├── test_quadrotor3dslungload.py └── test_reinmav.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/.gitignore -------------------------------------------------------------------------------- /.travis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/.travis.yml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/README.md -------------------------------------------------------------------------------- /examples/train_quadrotor2d_ppo2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/examples/train_quadrotor2d_ppo2.py -------------------------------------------------------------------------------- /gym_reinmav/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/__init__.py -------------------------------------------------------------------------------- /gym_reinmav/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/config.yaml -------------------------------------------------------------------------------- /gym_reinmav/controller/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/controller/__init__.py -------------------------------------------------------------------------------- /gym_reinmav/controller/quat_pid_controller.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gym_reinmav/controller/rpy_pid_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/controller/rpy_pid_controller.py -------------------------------------------------------------------------------- /gym_reinmav/envs/mujoco/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/envs/mujoco/__init__.py -------------------------------------------------------------------------------- /gym_reinmav/envs/mujoco/assets/quadrotor_ground.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/envs/mujoco/assets/quadrotor_ground.xml -------------------------------------------------------------------------------- /gym_reinmav/envs/mujoco/assets/quadrotor_hovering.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/envs/mujoco/assets/quadrotor_hovering.xml -------------------------------------------------------------------------------- /gym_reinmav/envs/mujoco/assets/quadrotor_quat.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/envs/mujoco/assets/quadrotor_quat.xml -------------------------------------------------------------------------------- /gym_reinmav/envs/mujoco/mujoco_quad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/envs/mujoco/mujoco_quad.py -------------------------------------------------------------------------------- /gym_reinmav/envs/mujoco/mujoco_quad_hovering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/envs/mujoco/mujoco_quad_hovering.py -------------------------------------------------------------------------------- /gym_reinmav/envs/mujoco/mujoco_quad_quat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/envs/mujoco/mujoco_quad_quat.py -------------------------------------------------------------------------------- /gym_reinmav/envs/native/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/envs/native/__init__.py -------------------------------------------------------------------------------- /gym_reinmav/envs/native/quadrotor2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/envs/native/quadrotor2d.py -------------------------------------------------------------------------------- /gym_reinmav/envs/native/quadrotor2d_slungload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/envs/native/quadrotor2d_slungload.py -------------------------------------------------------------------------------- /gym_reinmav/envs/native/quadrotor3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/envs/native/quadrotor3d.py -------------------------------------------------------------------------------- /gym_reinmav/envs/native/quadrotor3d_slungload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/envs/native/quadrotor3d_slungload.py -------------------------------------------------------------------------------- /gym_reinmav/envs/native/reinmav_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/envs/native/reinmav_env.py -------------------------------------------------------------------------------- /gym_reinmav/example/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gym_reinmav/example/mujoco/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gym_reinmav/example/mujoco/control_quat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/example/mujoco/control_quat.py -------------------------------------------------------------------------------- /gym_reinmav/example/mujoco/control_rpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/example/mujoco/control_rpy.py -------------------------------------------------------------------------------- /gym_reinmav/example/mujoco/train_hovering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/example/mujoco/train_hovering.py -------------------------------------------------------------------------------- /gym_reinmav/plots/position_x.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/plots/position_x.png -------------------------------------------------------------------------------- /gym_reinmav/plots/vel_x.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/plots/vel_x.png -------------------------------------------------------------------------------- /gym_reinmav/plots/yaw.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/plots/yaw.png -------------------------------------------------------------------------------- /gym_reinmav/resources/mujoco_quad.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/resources/mujoco_quad.gif -------------------------------------------------------------------------------- /gym_reinmav/resources/native_slungload.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/resources/native_slungload.gif -------------------------------------------------------------------------------- /gym_reinmav/resources/reinmav-gym-mujoco.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/resources/reinmav-gym-mujoco.gif -------------------------------------------------------------------------------- /gym_reinmav/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/gym_reinmav/run.py -------------------------------------------------------------------------------- /gym_reinmav/wrapper/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | pyquaternion>0.9 2 | gym 3 | vpython 4 | matplotlib -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/setup.py -------------------------------------------------------------------------------- /test/test_quadrotor2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/test/test_quadrotor2d.py -------------------------------------------------------------------------------- /test/test_quadrotor2dslungload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/test/test_quadrotor2dslungload.py -------------------------------------------------------------------------------- /test/test_quadrotor3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/test/test_quadrotor3d.py -------------------------------------------------------------------------------- /test/test_quadrotor3dslungload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/test/test_quadrotor3dslungload.py -------------------------------------------------------------------------------- /test/test_reinmav.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethz-asl/reinmav-gym/HEAD/test/test_reinmav.py --------------------------------------------------------------------------------