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