/MARL-UAVs-Targets-Tracking

Attempt to reproduce and improve the implementation of the paper 'Improving Multi-Target Cooperative Tracking Guidance for UAV Swarms Using Multi-Agent Reinforcement Learning'.

Primary LanguagePython

MARL-UAVs-Targets-Tracking

The implement and improvement of the paper “Improving multi-target cooperative tracking guidance for UAV swarms using multi-agent reinforcement learning”.

Environment

You can simply use pip install to config the environment:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
pip install numpy matplotlib tqdm tensorboard scipy
pip install imageio[ffmpeg]

Run the code

cd src
python main.py

ToDo List

  • vanilla MAAC
    • Actor-Critic framework
  • MAAC-R
    • reciprocal reward (with PMI network)
  • MAAC-G
    • receive the global reward
  • 3D demo