kcg2015 / Vehicle-Detection-and-Tracking

Computer vision based vehicle detection and tracking using Tensorflow Object Detection API and Kalman-filtering
537 stars 193 forks source link

Tracking and counting #30

Closed JonathanB-96 closed 4 years ago

JonathanB-96 commented 4 years ago

Hi Kyle,

First of all I would like to thank you as I am grateful for your amazing work, I am a student conducting a research for my thesis on people counter using object detection and tracking and this was very helpful.

I used the SSD mobilenet pre-trained model and trained with a custom dataset I made from overhead images of people taken from an IP camera. I was able to detect the heads of people using tensorflow's object detection API after training while in real-time. With the help of your code and some arrangements to it I was also able to track the people's movement and assign an ID.

My next step, and this is where I am struggling, is to count the person entering a room i.e after crossing the ROI(region of interest) so it is important I know the direction of the movement (up or down) and when it crosses this ROI. I have spent a lot of time searching for something similar that could be useful to me but no luck. I was wondering if you had any feedback that could be of help to me as it would be greatly appreciated. TIA.

Below is a sample image of my work.

image