ailia Tech BLOG

CrowdCounting : A Machine Learning Model for Counting People

This is an introduction to「CrowdCounting」, a machine learning model that can be used with ailia SDK. You can easily use this model to create AI applications using ailia SDK as well as many other ready-to-use ailia MODELS.


Overview

CrowdCountCascadedMtl is a machine learning model released in August 2017 that counts the number of people in an input image. It is suitable for counting attendance in large crowds such as concert halls or stadiums.

CNN-based Cascaded Multi-task Learning of High-level Prior and Density Estimation for Crowd…Estimating crowd count in densely crowded scenes is an extremely challenging task due to non-uniform scale variations…arxiv.org

svishwa/crowdcount-cascaded-mtlThis is implementation of the paper CNN-based Cascaded Multi-task Learning of High-level Prior and Density Estimation…github.com

Architecture

CrowdCounting calculates a DensityMap that shows the distribution of the crowd, and predicts the number of people by making estimation based on this Density Map.

Source:https://arxiv.org/pdf/1707.09605.pdf

The output of the high-level prior stage, which is trained using the Classifier model to count the number of people, is input to the Density estimation stage to increase the accuracy.

Source:https://arxiv.org/pdf/1707.09605.pdf

The Shanghai Tech dataset and UCF_CC_50 were used for training and evaluation, and both showed high performance.

Source:https://arxiv.org/pdf/1707.09605.pdf

Source:https://arxiv.org/pdf/1707.09605.pdf

Source:https://arxiv.org/pdf/1707.09605.pdf

Source:https://arxiv.org/pdf/1707.09605.pdf

Usage

To use CrowdCounting with the ailia SDK, use the following command to measure the number of people seen by the webcam.

$ python3 crowdcount-cascaded-mtl.py -v 0

ailia-ai/ailia-modelsAilia input shape: (1, 1, 480, 640) Automatically downloads the onnx and prototxt files on the first run. It is…github.com

An example of the processing result is shown below.


ailia Inc. has developed ailia SDK, which enables cross-platform, GPU-based rapid inference.

ailia Inc. provides a wide range of services from consulting and model creation, to the development of AI-based applications and SDKs. Feel free to contact us for any inquiry.