Awesome
Human detection and Tracking
Introduction
In this project we have worked on the problem of human detection,face detection, face recognition and tracking an individual. Our project is capable of detecting a human and its face in a given video and storing Local Binary Pattern Histogram (LBPH) features of the detected faces. LBPH features are the key points extracted from an image which is used to recognize and categorize images. Once a human is detected in video, we have tracked that person assigning him a label. We have used the stored LBPH features of individuals to recognize them in any other videos. After scanning through various videos our program gives output like- person labeled as subject1 is seen in video taken by camera1, subject1 is seen in video by camera2. In this way we have tracked an individual by recognizing him/her in the video taken by multiple cameras. Our whole work is based on the application of machine learning and image processing with the help of openCV.This code is built on opencv 3.1.1, python 3.4 and C++, other versions of opencv are NOT SUPPORTED.
Requirements
-
opencv [v3.1.1]
-
python3
- In Ubuntu python 3.4 can be installed via terminal with the command given below:
sudo apt-get install python3
- In Ubuntu python 3.4 can be installed via terminal with the command given below:
-
python libraries: Here is a list of all the python dependencies
- Python Image Library (PILLOW)
- Imutils
- numpy
-
C++
Approach
- The code follows the steps given below:
- First it reads a video and process each frame one by one.
- For each frame it tries to detect a human. If a human is detected it draws a rectangle around it.
- after completing step 2 it tries to detect human face.
- if a human face is detected it tries to recognize it with a pre-trained model file.
- If human face is recognized it puts the label on that human face else it moves to step 2 again for next frame
- The repository is structured as follows:
main.py
: This is the main python file that detects and recognizes humans.main.cpp
: This is the main C++ file that detects and recognizes humans.create_face_model.py
: This python script is used to create model file using the given data indata/
foldermodel.yaml
: This file contains trained model for given data. This trained model contains LBPH features of each and every face for given data.face_cascades/
: This directory contains sample data for testing our codes. This data is prepared by extracting face images of a praticular person from some videos.scripts/
: This directory contains some useful scripts that we used to work on different problems.video/
: This directory contains some of the videos that we used to while testing.
Installation
Python
Don't forget to install the necessary libraries described in the install paragraph above.
First you need to run the create_face_model.py file, which uses the images in /data to create a .yaml file
- In the project folder run
python create_face_model.py
- To run the python version of the code you have to put all the input videos in one folder and then provide the path of that folder as command line argument:
python3 main.py -v /path/to/input/videos/
Example- for our directory structure it is:
python3 main.py -v /video
C++
- To compile the C++ version of the code with openCV the command is:
g++ -ggdb `pkg-config --cflags opencv` -o `basename name_of_file.cpp .cpp` name_of_file.cpp `pkg-config --libs opencv`
Example- for our directory structure it is:
g++ -ggdb `pkg-config --cflags opencv` -o `basename main.cpp .cpp` main.cpp `pkg-config --libs opencv`
- To run the C++ version of the code you have to put all the input videos in one folder and then provide the path of that video as command line argument:
./name_of_file /path/to/input/video_file
Example- for our directory structure it is:
./main /video/2.mp4
- creating your own model file; just follow the steps given below to create your own model file:
- for each individual rename the images as
subjectx.y.jpg
for example for person 1 images should be named assubject01.0.jpg
,subject01.1.jpg
and so on. - put all the images of all the persons in a single folder for example you can see
data\
folder then run this command given below:python3 create_face_model.py -i /path/to/persons_images/
- for each individual rename the images as
Performance of code
- Since this is a computer vision project it requires a lot of computation power and performance of the code is kind of an issue here.
- The code was tested on two different machines to analyse performace. The input was 30fps 720p video.
- On a machine with AMD A4 dual-core processor we got an output of 4fps which is quite bad.
- on a machine with Intel i5 quad-core processor we got an output of 12fps.
Results
You can find project report here
To do
- improve the performance of the code
- improve the accuracy of the code and reducing the false positive rate.
- improve the face recognition accuracy to over 90 percent
Special Thanks to:
- Jignesh S. Bhatt - Thank you for mentoring this project
- Kamal Awasthi - Helped in testing the code