[发明专利]一种客流计数方法、装置、电子设备及存储介质

说明书

本发明实施例提供一种客流计数方法、装置、电子设备及存储介质，该方法包括以下步骤：获取待计数视频流，所述待计数视频流包括第一预设数量的检测帧；对所述待计数视频流的所述检测帧进行待计数人员的人头三维估计和身高估计，得到待计数人员的人头三维信息和身高信息；根据所述人头三维信息和身高信息对所述待计数人员进行轨迹跟踪，计算所述待计数人员的三维轨迹；判断所述待计数人员的三维轨迹是否穿过预设的门平面；若所述待计数人员的三维轨迹穿过所述预设的门平面，则对所述待计数人员进行计数。本发明能够提高客流统计的准确率。

技术领域

本发明涉及客流统计技术领域，尤其涉及一种客流计数方法、装置、电子设备及存储介质。

背景技术

在目前精准营销的商业模式下，客流计数、客流转化率、客户在店铺行为等信息已经成为商家十分关注的信息。目前市场上已有的客流相机大多为2D客流相机，即通过在二维像平面内画线、画框等设置来判断客流信息。比如，单目的2D计数主要是检测跟踪人头，判断人头是否从检测线的上方移动到下方。缺点在于画线在一些情况下无法计算到小孩子进出。且单一的计数逻辑无法同时兼容店铺场景的多变、行人进店路线多变、行人身高等等因素，造成客流统计的准确率较差。可见，现有客流计数方法存在客流统计的准确率较差的问题。

发明内容

本发明实施例提供一种客流计数方法，能够解决现有客流计数方法存在客流统计的准确率较差的问题。

第一方面，本发明实施例提供一种客流计数方法，所述客流计数方法包括：

获取待计数视频流，所述待计数视频流包括第一预设数量的检测帧；

对所述待计数视频流的所述检测帧进行待计数人员的人头三维估计和身高估计，得到待计数人员的人头三维信息和身高信息；

根据所述人头三维信息和身高信息对所述待计数人员进行轨迹跟踪，计算所述待计数人员的三维轨迹；

判断所述待计数人员的三维轨迹是否穿过预设的门平面；

若所述待计数人员的三维轨迹穿过所述预设的门平面，则对所述待计数人员进行计数。

可选的，所述对所述待计数视频流的所述检测帧进行待计数人员的人头三维估计和身高估计，得到待计数人员的人头三维信息和身高信息的步骤包括：

对所述检测帧进行人头检测以及人体检测，得到人头检测结果与人体检测结果；

基于所述人头检测结果进行人头跟踪，得到人头跟踪结果，所述人头跟踪结果包括人头中心点位置；

基于所述人体检测结果，进行人体关键点检测，得到人脚关键点位置；

根据所述人头中心点位置以及所述人脚关键点位置，对所述待计数人员进行身高估计，得到所述待计数人员的身高信息；

根据预设的地面方程、所述人头中心点位置以及所述人脚关键点位置进行人头三维估计，计算得到所述待计数人员的人头三维信息。

可选的，所述人头三维信息包括三维人头位置和人头ID，所述根据所述人头三维信息和身高信息对所述待计数人员进行轨迹跟踪，计算所述待计数人员的三维轨迹的步骤包括：

根据所述三维人头位置和所述人头ID，计算所述待计数视频流中所述待计数人员的三维轨迹。

可选的，所述待计数视频流还包括跟踪帧，相邻检测帧之间间隔有第二预设数量的跟踪帧，所述跟踪帧继承上一检测帧的身高信息，所述根据所述人头三维信息和身高信息对所述待计数人员进行轨迹跟踪，计算所述待计数人员的三维轨迹的步骤，包括：

对当前跟踪帧进行人头检测，得到人头检测结果；

基于所述人头检测结果进行人头跟踪，得到人头跟踪结果，所述人头跟踪结果包括人头中心点位置；