uzh-rpg / agile_flight

Developing and Comparing Vision-based Algorithms for Vision-based Agile Flight
MIT License
137 stars 52 forks source link

Training script - RL #37

Open marleyshan21 opened 2 years ago

marleyshan21 commented 2 years ago

Hi Team,

The control modality choices are available on the ROS side of scripts and we wanted to use that for training an RL policy.

  1. Which is the ROS side of the script similar to run_vision_ppo in Python to get started with training a RL policy?
  2. Is there a corresponding VisionEnv_v1 training environment for the ROS side to train so that we could utilise the other control modalities. (Kindly let me know if such a thing is not reqd and this itself can be used too. I am new to it)

Thanks

yun-long commented 2 years ago

Hi @marleyshan21

Unfortunately, on the ROS side, there is no scripts for training RL policy. The RL algorithm only supports the body rate control modality.

Best,