Skip to content

A naive Algorithm that uses People Detection and Convex Hull as subroutines to solve this problem: "Given an image of people standing randomly, how many people can be a part of some queue {Q}. A queue {Q} is basically a line(y = mx + c) that minimises a custom loss function. "

jalotra/Queue-Detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

25 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Queue-Detection

Objective and Approach:

  1. To find a possible queue vector which is basically a line or (y = mx + c) using People Detection as a subroutine and then answer this question : "How many people are standing in the queue".

  2. Assumption Taken : "Whatever be the camera angle is the queue is always a planar surface in a 3D world".

  3. Finds the bounding boxes around people using Yolo.

  4. Of all those bounding boxes the code tries to find the best params(m, c) such that maximum people lie on or some delta(eps) along this line.

  5. Takes 90% of all the points to remove outliers.

  6. Finding params(m, c) is done by solving an optimisation problem that uses Convex-Hull as a subroutine.

For more details refer the Presentation that I made : Presentation

Current Tasks

Yolo from here : https://github.com/ultralytics/yolov5

  1. To use the YOLO Algorithm to detect all the bounding boxes that are persons.
  2. Reading the Source Code of Darknet, learning about Yolo Layers and the implementation in Pytorch.
  3. Maybe Fine Tune the Model to only output bounding boxes around humans. Something like this : https://www.codeproject.com/Articles/5283660/AI-Queue-Length-Detection-Counting-the-Number-of-P

Setting Up the Development Arena

  1. You must have Conda installed, since it provides containerisation we dont have to worry about building and running on Different Platforms/ OSes.
  2. First run "bash setup_env.sh" -> This will create a Queue_Detection conda environment.
  3. Run conda activate Queue_Detection to activate the newly made env.
  4. Next run "bash setup.sh" -> This will install all the related dependencies.
  5. In case of error, read the *sh files and figure it out yourself.

Average Timings on Still Images

  1. Yolov5s -
  2. Yolov5m -
  3. Yolov5l -
  4. Yolov5x -
  5. Yolov5x + TTA -

Datasets to Check On

  1. Shanghai DataSet : https://www.kaggle.com/tthien/shanghaitech

Readings :

Ques : Why are there different versions of the same paper ?

  1. V1 : https://arxiv.org/pdf/1506.02640v1.pdf
  2. V2 : Somrthing Here
  3. V3 : Something Here
  4. V4 : https://arxiv.org/pdf/1506.02640v4.pdf
  5. https://blog.paperspace.com/how-to-implement-a-yolo-object-detector-in-pytorch/

Results :

Picture 1 :

Indian Queue

Picture 2 :

Indian Queue

Picture 3 :

Indian Queue

Picture 4 :

Indian Queue

Picture 5 :

Not Indian Queue

About

A naive Algorithm that uses People Detection and Convex Hull as subroutines to solve this problem: "Given an image of people standing randomly, how many people can be a part of some queue {Q}. A queue {Q} is basically a line(y = mx + c) that minimises a custom loss function. "

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published