[发明专利]头盔识别方法、装置、设备及存储介质

权利要求书

1.一种头盔识别方法，其特征在于，所述方法包括：

获取目标图像；

将所述目标图像输入预先训练的头盔识别网络，得到识别结果；所述头盔识别网络具有同时检测头盔区域和头肩区域的功能，所述识别结果包括所述目标图像中头盔区域的检测结果和头肩区域的检测结果；

基于所述头盔区域的检测结果和所述头肩区域的检测结果进行头盔识别；

所述头盔识别网络包括第一网络分支和第二网络分支；

所述第一网络分支用于进行目标检测，所述目标检测包括头盔的检测和头肩的检测；

所述第二网络分支用于进行属性检测，得到属性信息；所述属性信息用于辅助进行头盔识别；

相应地，所述识别结果包括所述目标图像的属性信息。

2.根据权利要求1所述的方法，其特征在于，所述基于所述头盔区域的检测结果和所述头肩区域的检测结果进行头盔识别，包括：

在所述头盔区域的检测结果指示存在所述头盔区域、且所述头肩区域的检测结果指示存在所述头肩区域的情况下，确定所述头盔区域与所述头肩区域之间的匹配关系，以确定所述头盔区域与所述头肩区域是否属于同一骑行人员；

在匹配出所述头盔区域与所述头肩区域属于同一骑行人员的情况下，确定所述头肩区域对应的骑行人员佩戴头盔。

3.根据权利要求1所述的方法，其特征在于，所述基于所述头盔区域的检测结果和所述头肩区域的检测结果进行头盔识别，包括：

按照所述属性信息确定所述目标图像中的车辆是否为预设类型，所述预设类型的车辆为需要骑行人员佩戴头盔的车辆；其中，所述属性信息包括车辆的类型属性、载人数量属性、头盔的颜色属性和/或头盔的款式属性；

在所述属性信息确定所述目标图像中的车辆为所述预设类型时，按照所述头盔区域的检测结果和所述头肩区域的检测结果确定所述头盔区域与所述头肩区域是否匹配，以确定所述头盔区域与所述头肩区域是否属于同一骑行人员；

在所述头盔区域与所述头肩区域匹配时，确定所述目标图像中所述头肩区域对应的骑行人员佩戴头盔。

4.根据权利要求3所述的方法，其特征在于，所述头盔识别网络基于中心点的检测网络CenterNet建立，所述头盔识别网络从前至后依次包括主干网络、分别与所述主干网络相连的所述第一网络分支和所述第二网络分支；

所述第一网络分支包括用于提取所述目标图像的热度图；并基于所述热度图进行目标检测；

所述第二网络分支包括深度可分离卷积网络，所述深度可分离卷积网络的网络层数小于第一阈值和/或参数数量小于第二阈值，所述第一阈值小于或等于现有的卷积网络的最小层数；所述第二阈值小于或等于现有的卷积网络的最小参数数量。

5.根据权利要求4所述的方法，其特征在于，所述主干网络分别与所述第一网络分支和所述第二网络分支进行局部残差连接。

6.根据权利要求1所述的方法，其特征在于，所述第二网络分支基于改进的focal loss函数训练得到，所述改进的focal loss函数通过下式表示：

FL(p_t)＝-α_t(1-p_t)^γlog(p_t)

其中，pt表示网络预测结果，FL(pt)表示所述改进的focal loss函数的函数值，γ为预设常数，α_t为平衡因子，并通过下式表示：

其中，w是训练数据中具有属性标签的样本所占样本总数的权重值。