zzzzzyh111 / gazebo_uav_rl

obstacle detection and avoidance for UAV via gazebo

UAV Obstacle Avoidance-DQN-Gazebo

Based on the Gazebo simulation, this repository focuses on the UAV obstacle avoidance problem, which leverages visual inputs and the DQN algorithm.
If you have your own gazebo world file, please run the code after roslaunch! From my side, the roslaunch command is

roslaunch hector_quadrotor_gazebo quadrotor_empty_world.launch

Please make sure that the relevant dependencies have been installed before the training process

sudo apt-get update
sudo apt-get upgrade
pip install rospkg
pip install catkin-tools
pip install opencv-python
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 # I'm using torch2.0.1+cuda11.7

As for the other dependencies, please install them according to your error messages.

After that, you can start training straight away by running

python3 main.py

Moreover, you can define your own goal/start/obstacle positions in config.py. Definitions regarding the simulation environment, such as resetting the environment, can be found in env.py. ddqn.py describes the DQN algorithm and its derivatives. validate.py is used to test the parameters.

Since the training process requires real physical time to give the drone time to perform the maneuvers, the training time is long (about 10 hours or so to see the REWARD noticeably rise phenomenon), so please be patient. If you need to do gazebo simulation acceleration, please refer to this website. The training curves are shown below.

The collision-free trajectories are shown below.

This work has been published in [AIAA Aviation 2023 Forum](https://arc.aiaa.org/doi/abs/10.2514/6.2023-3813). Please kindly refer to the following format for citations if needed

@inproceedings{zhang2023partially,
  title={Partially-Observable Monocular Autonomous Navigation for UAV through Deep Reinforcement Learning},
  author={Zhang, Yuhang and Low, Kin Huat and Lyu, Chen},
  booktitle={AIAA AVIATION 2023 Forum},
  pages={3813},
  year={2023}
}

gazebo_uav_rl's Issues

Is there any documentation on this?

Is there any documentation on this?

CPU memory increasing

Hello, first of all, thank you very much for your kind works! I do have a question, though. When I am training, the CPU memory keeps increasing continuously, and it doesn't decrease much at the start of each new episode. Do you have any solutions to this issue? Looking forward to your reply, and once again, thank you for your work.

Request for Information about Gazebo Environment Used

Hi

Could you please provide details about the Gazebo environment and model used in this project?

i cant run your code with hector gazebo, i get many error