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

权利要求书

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

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

根据所述第一特征图中像素点之间的关联关系，对所述第一特征图进行编码，得到第二特征图，其中，所述像素点之间的关联关系包括位置关系和特征关系；

根据所述第二特征图进行人群定位，得到所述人群图像对应的目标定位图，其中，所述目标定位图用于指示所述人群图像中包括的人体的位置；

所述根据所述第二特征图进行人群定位，得到所述人群图像对应的目标定位图，包括：

根据所述第二特征图进行人群定位，得到第一定位概率图，其中，所述第一定位概率图用于指示所述人群图像中各像素点是人体的概率；

根据概率阈值，对所述第一定位概率图进行图像处理，得到所述目标定位图。

2.根据权利要求1所述的方法，其特征在于，所述根据所述第一特征图中像素点之间的关联关系，对所述第一特征图进行编码，得到第二特征图，包括：

对所述第一特征图进行维度转换，得到第一特征序列；

对所述第一特征序列进行编码，得到第二特征序列；

对所述第二特征序列进行维度转换，得到所述第二特征图，其中，所述第二特征图与所述第一特征图具有相同的尺寸。

3.根据权利要求2所述的方法，其特征在于，所述对所述第一特征序列进行编码，得到第二特征序列，包括：

利用多头注意力模块、前馈神经网络和归一化模块，对所述第一特征序列进行预设迭代次数的编码处理，得到所述第二特征序列。

4.根据权利要求3所述的方法，其特征在于，所述利用多头注意力模块、前馈神经网络和归一化模块，对所述第一特征序列进行预设迭代次数的编码处理，得到所述第二特征序列，包括：

利用所述归一化模块和所述多头注意力模块，对第t个第三特征序列进行处理，得到第t+1个第四特征序列，其中，t是大于或等于0的整数，在t＝0的情况下，第0个第三特征序列是所述第一特征序列；

利用所述归一化模块和所述前馈神经网络，对所述第t+1个第四特征序列进行处理，得到第t+1个第三特征序列；

在达到所述预设迭代次数的情况下，将所述第t+1个第三特征序列确定为所述第二特征序列。

5.根据权利要求1所述的方法，其特征在于，所述根据所述第二特征图进行人群定位，得到第一定位概率图，包括：

对所述第二特征图进行卷积处理，得到第三特征图；

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

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

6.根据权利要求1至5中任意一项所述的方法，其特征在于，所述根据概率阈值，对所述第一定位概率图进行图像处理，得到所述目标定位图，包括：

对所述第一定位概率图进行平均池化操作，得到均值池化图；

对所述均值池化图进行最大池化操作，得到最大池化图；

根据所述均值池化图和所述最大池化图，得到第二定位概率图；

根据所述概率阈值，对所述第二定位概率图进行阈值分割，得到所述目标定位图。

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

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

通过所述人群定位神经网络，确定所述人群样本图像对应的预测定位概率图，其中，所述预测定位概率图用于指示所述人群样本图像中各像素点是人体的概率；

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

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