Skip to content

Tello drone object tracking using object detection (YOLO) and reinforcement learning (DDPG)

Notifications You must be signed in to change notification settings

soldierofhell/tello-rl-yolo

 
 

Repository files navigation

tello-rl-yolo

Tello drone implementation with YOLO and DDPG control.

This capstone project was realized in the context of the Udacity Machine Learning Nanodegree.

Dataset:

VOC2012 Train/val : http://pjreddie.com/media/files/VOCtrainval_11-May-2012.tar

VOC2012 Test : http://pjreddie.com/media/files/VOC2012test.tar

Inspired from :

How To

Requirements :

  • Python 3.X
  • Keras GPU
  • Keras-rl
  • OpenCV
  • Numpy
  • CUDA & NVIDA Drivers
  • OpenAI Gym

Make sure YOLO weight file VOC.h5 is in the root folder : https://drive.google.com/open?id=15oONh_eIdz3CkHdwybZeB49rDCpE0X9A

1- Start main.py

2- Once the video is on, it will take 30 seconds for the YOLO and DDPG to initialize (model creation, loading, etc..).

3- Take off using T

4- Drone should track bounding box in screen.

About

Tello drone object tracking using object detection (YOLO) and reinforcement learning (DDPG)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 79.3%
  • Jupyter Notebook 20.7%