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

权利要求书

1.一种人群定位方法，其特征在于，包括：

对人群图像进行特征提取，得到至少一个第一特征图；

对所述至少一个第一特征图进行人体关键点检测，确定所述人群图像对应的定位概率图，其中，所述定位概率图用于指示所述人群图像中各像素点是目标人体点的概率；

根据所述定位概率图，对所述至少一个第一特征图进行加权处理，得到至少一个第二特征图；

对所述至少一个第二特征图进行人体框检测，确定所述人群图像对应的目标人体框检测结果，其中，所述目标人体框检测结果中包括所述人群图像中各人体对应的目标人体框。

2.根据权利要求1所述的方法，其特征在于，所述对所述至少一个第一特征图进行人体关键点检测，确定所述人群图像对应的定位概率图，包括：

对目标第一特征图进行卷积处理，得到第三特征图，其中，所述目标第一特征图是所述至少一个第一特征图中的一个；

对所述第三特征图进行转置卷积处理，得到第四特征图，其中，所述第四特征图和所述人群图像具有相同的尺寸；

对所述第四特征图进行卷积处理，得到所述定位概率图。

3.根据权利要求1或2所述的方法，其特征在于，所述根据所述定位概率图，对所述至少一个第一特征图进行加权处理，得到至少一个第二特征图，包括：

根据所述定位概率图，对每个所述第一特征图进行像素级别的加权处理，得到每个所述第一特征图对应的所述第二特征图。

4.根据权利要求1至3中任意一项所述的方法，其特征在于，所述对所述至少一个第二特征图进行人体框检测，确定所述人群图像对应的目标人体框检测结果，包括：

对所述至少一个第二特征图进行人体框检测，得到至少一个人体框预测结果；

对所述至少一个人体框预测结果进行融合，得到所述目标人体框检测结果。

5.根据权利要求4所述的方法，其特征在于，所述对所述至少一个人体框预测结果进行融合，得到所述目标人体框检测结果，包括：

对所述至少一个人体框预测结果进行非极大值抑制处理，得到所述目标人体框检测结果。

6.根据权利要求1至5中任意一项所述的方法，其特征在于，所述方法还包括：

根据所述定位概率图，对所述人群图像进行人群分析，得到所述人群图像对应的第一分析结果；

根据所述目标人体框检测结果，对所述人群图像进行人群分析，得到所述人群图像对应的第二分析结果；

对所述第一分析结果和所述第二分析结果进行融合，得到所述人群图像对应的目标分析结果。

7.根据权利要求1至6中任意一项所述的方法，其特征在于，所述人群定位方法通过人群定位神经网络实现，所述人群定位神经网络的训练样本包括：人群样本图像、所述人群样本图像对应的真实定位图、以及所述人群样本图像对应的真实人体框检测结果；

所述人群定位神经网络的训练方法如下：

通过所述人群定位神经网络，确定所述人群样本图像对应的预测定位概率图，以及所述人群样本图像对应的预测人体框检测结果，其中，所述预测定位概率图用于指示所述人群样本图像中各像素点是目标人体点的概率，所述预测人体框检测结果中包括所述人群样本图像中各人体对应的预测人体框；

基于所述预测定位概率图和所述真实定位图，确定第一定位损失；

基于所述预测人体框检测结果和所述真实人体框检测结果，确定第二定位损失；

基于所述第一定位损失和所述第二定位损失，优化所述人群定位神经网络。

8.根据权利要求7所述的方法，其特征在于，所述基于所述第一定位损失和所述第二定位损失，优化所述人群定位神经网络，包括：

对所述第一定位损失和所述第二定位损失进行求和，得到目标定位损失；

基于所述目标定位损失，优化所述人群定位神经网络。