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

说明书

本实施例公开了一种人体朝向检测方法、装置、电子设备和计算机存储介质，该方法包括：对待处理图像进行特征提取，得到所述待处理图像的特征；基于所述待处理图像的特征确定人体关键点和初步的人体朝向；根据确定的人体关键点和初步的人体朝向，确定最终的人体朝向。如此，在本公开实施例中，最终的人体朝向是在综合考虑人体关键点和初步的人体朝向的基础上得出的，因而，可以基于人体关键点提升最终的人体朝向的准确度和可利用性。

技术领域

本公开涉及计算机视觉处理技术，尤其涉及一种人体朝向检测方法、装置、电子设备和计算机存储介质。

背景技术

随着计算机视觉处理技术的发展，行人朝向检测逐渐称为计算机视觉领域重要的研究问题；行人朝向检测的方案可以是，对摄像机得到的图像进行处理，以预测图像中每一个人的身体和/或面部的朝向；然而，相关技术中，如何提高行人朝向检测的准确度和可利用性，是亟待解决的问题。

发明内容

本公开实施例期望提供人体朝向检测的技术方案。

本公开实施例提供了一种人体朝向检测方法，所述方法包括：

对待处理图像进行特征提取，得到所述待处理图像的特征；

基于所述待处理图像的特征确定人体关键点和初步的人体朝向；

根据确定的人体关键点和初步的人体朝向，确定最终的人体朝向。

可选地，所述根据确定的人体关键点和初步的人体朝向，确定最终的人体朝向，包括：

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

可选地，所述根据确定的人体关键点和初步的人体朝向，确定最终的人体朝向，包括：

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

可选地，对待处理图像进行特征提取得到所述待处理图像的特征，以及基于所述待处理图像的特征确定人体关键点和初步的人体朝向的步骤由神经网络执行，所述神经网络采用第一样本图像和第二样本图像训练得到，所述第一样本图像中包括第一人体图像和标注的人体关键点，所述第二样本图像包括第二人体图像和标注的人体朝向。

可选地，所述神经网络采用第一样本图像和第二样本图像训练得到，包括：

对所述第一样本图像和所述第二样本图像进行特征提取，得到所述第一样本图像和所述第二样本图像的特征；根据所述第一样本图像的特征进行行人关键点检测，得到所述第一类样本图像的人体关键点；基于所述第二样本图像的特征进行朝向检测，得到所述第二样本图像的人体朝向；

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

可选地，所述对所述第一样本图像和所述第二样本图像进行特征提取，得到所述第一样本图像和所述第二样本图像的特征，包括：

将所述第一样本图像和所述第二样本图像进行拼接，对拼接后的图像数据进行特征提取，得到拼接后的图像数据的特征；

按照所述第一样本图像和所述第二样本图像的拼接方式，将所述拼接后的图像数据的特征拆分为所述第一样本图像的特征和所述第二样本图像的特征。

可选地，将所述第一样本图像和所述第二样本图像进行拼接，包括：所述第一样本图像和所述第二样本图像沿批维度拼接；

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

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