Student Projects

VE/VM450

Real-Time On-Device Flow Statistics Detection and Prediction

Sponsor: Allan Zhu, Allen Zhu, UM-SJTU Joint Institute
Team Members: Zhihao Ruan, Zhikai Chen, Jiayu Yi, Zekun Li, Zihao Shen
Instructor: Prof. Chengbin Ma

Project Video

Team Members

Team Members:

Zhihao Ruan
Zhikai Chen
Jiayu Yi
Zekun Li
Zihao Shen

Instructor:

Prof. Chengbin Ma

UM-SJTU JOINT INSTITUTE

Project Description

Problem Statement

A real-time on-device flow statistics detection and prediction model is brought forward to support social distancing considering the recent COVID-19 outbreak. The project sets out to provide insight into the flow statistics of a certain venue and help people avoid going to overcrowded locations, by collecting flow prediction statistics as well as an automated prediction of exits and entrances of the area based on human flow.

Fig. 1 Project demo [1]

Fig. 1 Project demo [1]

Concept Generation

The project can be broken down to the following sub-functions:
• Identify people in each frame.
• Track each personʼs trajectory
• Count people entering and exiting
• Store frames and bounding boxes
• Display counting statistics and video

The sequence between the above functions is shown in Fig.2.

Fig. 2 Morphological analysis

Fig. 2 Morphological analysis

Design Description

Our design consists of four parts:

1. Object Detection
Our project choose deep learning method for object detection. We did a benchmark among various object detection models. We finally choose MobileNet-SSD V2 320×320 as our object detector because of its detection speed.

2. Object Tracking
Our final object tracker design involves a tracker and an entrance detector. Kalman filter is used for tracker and DBSCAN (a density-based clustering algorithm) is used for entrance detector.

3. Backend Service
Our backend services include api configuration, image processing and data prediction. Python is used for most of the services and MongoDB is used for database.

4. Frontend Service
Our frontend service include data visualization, video demo display and development process display. We use React.js for web framework and ReCharts.js for interactive charts.

Fig.3 Benchmark of different models

Fig.3 Benchmark of different models

Fig.4 Object tracking mechanism

Fig.4 Object tracking mechanism

Validation

Validation metrics:
Since our project can be divided into four different functional parts, we use different engineering metrics for each part:
Object Detection: Detection accuracy and Detection speed
Object Tracking: Tracking accuracy, Tracking speed and Entrance detection accuracy
Backend Service: Server response latency
Frontend Service: Web page loading time
Validation Results:  
• Object detection accuracy: ≈ 100%
• Object detection speed: 5.2 FPS
• Object tracking accuracy: 88.9%
• Object tracking speed: 17.27 FPS
• Entrance detection accuracy: 80%
• Server response latency < 1000 ms • Web page loading time < 1000ms Compared with our customer requirements and engineering specifications, most requirements are met.[/et_pb_text][/et_pb_column][et_pb_column type="1_4" _builder_version="4.4.9"][et_pb_text _builder_version="4.4.9" text_line_height="1.4em" custom_margin="||-1px|||" custom_padding="60px||||false|false" custom_padding_tablet="" custom_padding_phone="20px|0px||0px|false|false" custom_padding_last_edited="on|phone" border_color_all="#5a7085"]

Fig.5 The MSE compared with the baseline

Fig.5 The MSE compared with the baseline

Conclusion

Our project starts from the background of COVID-19 pandemic. The basic task of our project is using object detection algorithm to get human traffic information, and then provide an elegant data visualization. We hope this project can contribute to fight against this pandemic.

Acknowledgement

Allan Zhu and Allen Zhu from UM-SJTU
Chengbin Ma, Mian Li, Chong Han, Jigang Wu, Shouhang Bo from UM-SJTU

Reference

[1] Original Video: https://www.pyimagesearch.com/2018/08/13/opencv-people-counter/