[发明专利]人体朝向检测方法、装置、电子设备和计算机存储介质

权利要求书

1.一种人体朝向检测方法，其特征在于，所述方法包括：

采用神经网络对待处理图像进行特征提取，得到所述待处理图像的特征；所述神经网络采用第一样本图像和第二样本图像训练得到；采用所述神经网络基于所述待处理图像的特征确定人体关键点和初步的人体朝向；根据确定的人体关键点和初步的人体朝向，确定最终的人体朝向；

所述神经网络采用第一样本图像和第二样本图像训练得到，包括：将所述第一样本图像和所述第二样本图像进行拼接，对拼接后的图像数据进行特征提取，得到拼接后的图像数据的特征；

按照所述第一样本图像和所述第二样本图像的拼接方式，将所述拼接后的图像数据的特征拆分为所述第一样本图像的特征和所述第二样本图像的特征；根据所述第一样本图像的特征进行行人关键点检测，得到所述第一样本图像的人体关键点；基于所述第二样本图像的特征进行朝向检测，得到所述第二样本图像的人体朝向；

根据检测的人体关键点、标注的人体关键点、检测的人体朝向和标注的人体朝向，调整所述神经网络的网络参数值。

2.根据权利要求1所述的方法，其特征在于，所述根据确定的人体关键点和初步的人体朝向，确定最终的人体朝向，包括：

响应于所述确定的人体关键点表征的人体朝向与所述初步的人体朝向相符的情况，将所述初步的人体朝向确定为所述最终的人体朝向。

3.根据权利要求1所述的方法，其特征在于，所述根据确定的人体关键点和初步的人体朝向，确定最终的人体朝向，包括：

响应于所述确定的人体关键点表征的人体朝向与所述初步的人体朝向不相符的情况，将所述确定的人体关键点表征的人体朝向确定为所述最终的人体朝向。

4.根据权利要求1-3任一所述的方法，其特征在于，所述第一样本图像中包括第一人体图像和标注的人体关键点，所述第二样本图像包括第二人体图像和标注的人体朝向。

5.根据权利要求1所述的方法，其特征在于，将所述第一样本图像和所述第二样本图像进行拼接，包括：

所述第一样本图像和所述第二样本图像沿批维度拼接；

在将所述第一样本图像和所述第二样本图像进行拼接之前，所述方法还包括：

将所述第一样本图像和所述第二样本图像在通道、高和宽三个维度上分别调整至相同。

6.根据权利要求1所述的方法，其特征在于，所述根据检测的人体关键点、标注的人体关键点、检测的人体朝向和标注的人体朝向，调整所述神经网络的网络参数值，包括：

根据所述检测的人体关键点和所述标注的人体关键点，得出所述神经网络的第一损失值，所述第一损失值表示所述检测的人体关键点和所述标注的人体关键点的差异；

根据所述检测的人体朝向和所述标注的人体朝向，得出所述神经网络的第二损失值，所述第二损失值表示所述检测的人体朝向和所述标注的人体朝向的差异；

根据所述第一损失值和所述第二损失值，调整所述神经网络的网络参数值。

7.一种人体朝向检测装置，其特征在于，所述装置包括提取模块和处理模块，其中，

提取模块，用于采用神经网络对待处理图像进行特征提取，得到所述待处理图像的特征；所述神经网络采用第一样本图像和第二样本图像训练得到；

处理模块，用于采用所述神经网络基于所述待处理图像的特征确定人体关键点和初步的人体朝向；根据确定的人体关键点和初步的人体朝向，确定最终的人体朝向；

训练模块，用于采用第一样本图像和第二样本图像训练所述神经网络，包括：将所述第一样本图像和所述第二样本图像进行拼接，对拼接后的图像数据进行特征提取，得到拼接后的图像数据的特征；

按照所述第一样本图像和所述第二样本图像的拼接方式，将所述拼接后的图像数据的特征拆分为所述第一样本图像的特征和所述第二样本图像的特征；根据所述第一样本图像的特征进行行人关键点检测，得到所述第一样本图像的人体关键点；基于所述第二样本图像的特征进行朝向检测，得到所述第二样本图像的人体朝向；

根据检测的人体关键点、标注的人体关键点、检测的人体朝向和标注的人体朝向，调整所述神经网络的网络参数值。