heinhtet123 / tello-face-recognition

An implementation of face detection and tracking on the dji Tello drone based on a HAAR Cascade using OpenCV and Python 3.6

Face Detection and Tracking with Dji Tello drone

This is an implementation of face detection and tracking on the dji Tello drone based on a HAAR Cascade using OpenCV and Python 3.6.

The current implementation allows the user to:

• Launch the drone through the command line using python main.py
• Receive video feed from the drone to the computer and visualize the face detection carried out by the drone

It allows the drone to:

• Detect multiple faces at any given frame
• Position the user at the center of any shot by deciding the best movement based on the users x, y and z coordinates

Note: Current implementation allows only tracking of 1 user.

Quick Start

To initialize your drone and get it up and running, simply clone the repository and download its dependencies with:

pip install -r requirements.txt

Afterwards, connect to the drones wifi and run:

python main.py

This will make the drone take off and initialize a video feed directly from the drone to your computer.

Next Steps

• Fix warning messages when executing the command and waitning for drone video feed
• Support drone centering with multiple users in a shot
• Optimize code for better video performance
• Implement trainable model for face recognition into the detection process (Follow just certain faces)