AntoineA67 / ai-meysam

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Object Tracking AI-Robot

Read the article: Watch the video on Yotube:

Model files

The ML model used in this project is placed in 'all_models' directory inside parent directory.

Overview of the Project

Robot detects an object using a Machine Learning model 'MobileNet SSD v1 (COCO)' and TensorFlow Lite interpreter. The Robot follows the object and manoeuvres itself to get the object in the center of frame. While the robot is tracking / following the object, working of tracking algorithm and Robot's view can be accessed on a browser. Robot's view with information overlay is generated using OpenCV. The various overlays on a frame are shown in the picture below

When the object is present in the frame, information such as bounding boxes, center of the object, deviation of the object from center of the frame, robot direction and speed are updated as shown in picture below. In the below example, X and Y values denote the deviation of center of the object (the red dot) from center of the frame. Since the horizontal deviation i.e. value of 'X' is above the tolerance value, the code generated 'Move Left' command.

Python's micro Web Framework called "FLASK" is used for streaming the camera frame (or Robot's view) over LAN.

About


Languages

Language:Python 98.9%Language:HTML 0.6%Language:PHP 0.6%