RotorS Gym is a framework dedicated for robotics researchers for designing reinforcement learning-based controllers. It combines the following toolkits:
- Ubuntu 16.04 LTS
- ROS Kinetic
- gym-gazebo
- pytorch 1.0.1
With GUI
roslaunch rotors_gazebo mav.launch mav_name:=firefly world_name:=basic
Headless:
roslaunch rotors_gazebo mav.launch mav_name:=firefly world_name:=basic gui:=false headless:=true
Train the agents:
python train.py
Play the demo:
python demo.py