[发明专利]一种基于头部检测的客流统计方法

说明书

本发明公开了一种基于头部检测的客流统计方法，包括：步骤1.基于One‑Stage的目标检测算法，结合Tiny‑yolo的网络结构与SSH中的SSH语境模块对输入的初始帧图片进行检测，得到一个或多个人头框坐标；为每一个人头框创建一条路径，用来保存该人头框中心点在后续帧中的位置信息，每一条路径分配一个id号；步骤2.输入当前帧图片进行人头检测得到当前帧人头框坐标；步骤3.以oks距离作为度量方式，以欧氏距离作为补充度量方式，用人头框的中心点坐标来计算目前存活的所有路径与当前帧检测出的所有人头框的oks距离矩阵与欧氏距离矩阵；步骤4.关联存活的所有路径与当前帧检测出的人头框；步骤5.进出客流量统计。

技术领域

本发明属于客流统计领域，具体是一种基于头部检测的客流统计方法。

背景技术

现阶段国内外客流统计技术主要包括以下四种：

红外线客流统计：依据人体的经过阻断红外线之间的对射从而统计客流，使用方便，数据传输量小，成本较低，安装方便，对于光线要求不高。但这种方法对于比较宽的门口，多人同时经过时易产生漏统计现象。

视频客流统计：主要依据人头计数，其精度较高，还可以根据图像分析追踪消费者在店内行动轨迹，同时可以和人脸识别系统相结合，进行CRM管理，成本不算高。但是这种算法较复杂，数据传输任务较重，共包括检测、跟踪和统计三个算法，人头检出率、多目标跟踪的精度以及统计过程的技巧都对客流统计准确率有重要的影响；

WIFI客流统计：采取智能手机mac地址进行人数统计，可以根据手机对店内的消费者的数量进行统计，且根据手机的IP进行消费者在店内行动轨迹的追踪以及判断新老顾客。但是，由于借助手机，存在一个人带多个手机、手机关闭无线信号或者关机等因素的影响，这种方法的统计数据会存在不准确的情况。

3D 传感客流计数器：图像识别与前两种相比要显得较为高端一些，同样是通过传感器检测，不同的是扫描人体轮廓进而得出立体的人体数据，再参照立体反射面的驼峰形状，进而用单片机计算，从而判断出人数。技术较为精确，但过高的成本限制了该方法的发展。

发明内容

为了解决现有技术中存在的上述技术问题，本发明提供了一种基于头部检测的客流统计方法，包括如下步骤：

步骤1.基于One-Stage的目标检测算法，结合Tiny-yolo的网络结构与SSH中的SSH语境模块对输入的初始帧图片进行检测，得到一个或多个人头框坐标；为每一个人头框创建一条路径，用来保存该人头框中心点在后续帧中的位置信息，每一条路径分配一个id号；

步骤2.输入当前帧图片进行人头检测得到当前帧人头框坐标；

步骤3.以oks距离作为度量方式，以欧氏距离作为补充度量方式，用人头框的中心点坐标来计算目前存活的所有路径与当前帧检测出的所有人头框的oks距离矩阵与欧氏距离矩阵；

步骤4.关联存活的所有路径与当前帧检测出的人头框；

步骤5. 进出客流量统计。

进一步的，步骤4具体包括如下步骤：

（1）关联操作：向融合匹配法输入距离矩阵，将当前帧检测出的人头框与路径进行匹配；

（2）对匹配上人头框的路径进行更新：存入被匹配上的人头框中心点坐标；每条路径最多匹配上一个人头框，假设第n条路径匹配上了第m个人头框，向第n条路径存入第m个人头框的中心点坐标；

（3）对当前帧中没有被匹配上的人头框进行初始化，创建新路径，并分配id号；假设当前帧中第k个与第q个人头框没有被任何一条路径匹配上，那么，初始化两条路径，分别用来存放这两个人头框中心点在后续帧中的坐标，并分配两个不同的id号，此时，总路径中新增了两条路径；

（4）删除连续n帧没有匹配上人头框的路径，未被删除的路径即为存活的路径。