SwapnilMad / Autonomous-Face-Tracking-Drone

Autonomous Face Tracking Drone using keras-rl

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Autonomous Face Tracking Drone

Reinforcement Learning applied on a Tello Drone to follow human faces using keras-rl. For this project DJI Tello Edu drone was used.

Agent is trained on a continuous state simulator to adjust velocity depending upon the position of the target using Deep Deterministic Policy Gradient (DDPG)

Click below image to redirect to youtube demo video

Youtube link

Getting Started

Clone or download the project to your system

Prequisites

Python

Install Python 3 or Anaconda

Installing

Install Dependencies

pip install -r requirements.txt

Running

Train the agent on a simulator and save weights

python train.py

Load the saved weights and test it on the drone

python drone_test.py

About

Autonomous Face Tracking Drone using keras-rl

License:MIT License


Languages

Language:Python 100.0%