[发明专利]人群定位方法及装置、电子设备和存储介质

说明书

本公开涉及一种人群定位方法及装置、电子设备和存储介质，所述方法包括：对人群图像进行特征提取，得到至少一个第一特征图；对所述至少一个第一特征图进行人体关键点检测，确定所述人群图像对应的定位概率图，其中，所述定位概率图用于指示所述人群图像中各像素点是目标人体点的概率；根据所述定位概率图，对所述至少一个第一特征图进行加权处理，得到至少一个第二特征图；对所述至少一个第二特征图进行人体框检测，确定所述人群图像对应的目标人体框检测结果，其中，所述目标人体框检测结果中包括所述人群图像中各人体对应的目标人体框。本公开实施例可以提高了人群定位的准确率。

技术领域

本公开涉及计算机技术领域，尤其涉及一种人群定位方法及装置、电子设备和存储介质。

背景技术

随着人口的增长、城市化进程的加速，人群大量聚集的行为越来越多，且规模越来越大。人群分析对于公共安全、城市规划有着重要意义。常见的人群分析任务包括人群计数、群体行为解析、人群定位等，其中，人群定位是其它人群分析任务的基础。人群定位指的是，通过计算机视觉算法对图像或视频中包括的人体的位置进行估计，确定图像或视频中包括的人体的坐标，以为后续人群计数和群体行为解析等人群分析任务提供数据依据。人群定位的准确率直接影响人群计数的精度和人群行为解析的结果。因此，亟需一种准确率较高的人群定位方法。

发明内容

本公开提出了一种人群定位方法及装置、电子设备和存储介质的技术方案。

根据本公开的一方面，提供了一种人群定位方法，包括：对人群图像进行特征提取，得到至少一个第一特征图；对所述至少一个第一特征图进行人体关键点检测，确定所述人群图像对应的定位概率图，其中，所述定位概率图用于指示所述人群图像中各像素点是目标人体点的概率；根据所述定位概率图，对所述至少一个第一特征图进行加权处理，得到至少一个第二特征图；对所述至少一个第二特征图进行人体框检测，确定所述人群图像对应的目标人体框检测结果，其中，所述目标人体框检测结果中包括所述人群图像中各人体对应的目标人体框。

在一种可能的实现方式中，所述对所述至少一个第一特征图进行人体关键点检测，确定所述人群图像对应的定位概率图，包括：对目标第一特征图进行卷积处理，得到第三特征图，其中，所述目标第一特征图是所述至少一个第一特征图中的一个；对所述第三特征图进行转置卷积处理，得到第四特征图，其中，所述第四特征图和所述人群图像具有相同的尺寸；对所述第四特征图进行卷积处理，得到所述定位概率图。

在一种可能的实现方式中，所述根据所述定位概率图，对所述至少一个第一特征图进行加权处理，得到至少一个第二特征图，包括：根据所述定位概率图，对每个所述第一特征图进行像素级别的加权处理，得到每个所述第一特征图对应的所述第二特征图。

在一种可能的实现方式中，所述对所述至少一个第二特征图进行人体框检测，确定所述人群图像对应的目标人体框检测结果，包括：对所述至少一个第二特征图进行人体框检测，得到至少一个人体框预测结果；对所述至少一个人体框预测结果进行融合，得到所述目标人体框检测结果。

在一种可能的实现方式中，所述对所述至少一个人体框预测结果进行融合，得到所述目标人体框检测结果，包括：对所述至少一个人体框预测结果进行非极大值抑制处理，得到所述目标人体框检测结果。

在一种可能的实现方式中，所述方法还包括：根据所述定位概率图，对所述人群图像进行人群分析，得到所述人群图像对应的第一分析结果；根据所述目标人体框检测结果，对所述人群图像进行人群分析，得到所述人群图像对应的第二分析结果；对所述第一分析结果和所述第二分析结果进行融合，得到所述人群图像对应的目标分析结果。