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

说明书

本申请涉及一种头盔识别方法、装置、设备及存储介质，属于计算机技术领域，该方法包括：获取目标图像；将目标图像输入预先训练的头盔识别网络，得到识别结果；基于头盔区域的检测结果和头肩区域的检测结果进行头盔识别；可以解决现有的头盔识别效率较低的问题；头盔识别网络具有同时检测头盔区域和头肩区域的功能，这样，可以同时实现头盔区域和头肩区域的检测，并结合两个区域的检测结果识别头盔，既可以保证头盔识别的准确性，又可以提高头盔识别效率。

【技术领域】

本申请涉及一种头盔识别方法、装置、设备及存储介质，属于计算机技术领域。

【背景技术】

随着对交通安全关注度的提升，如何对摩托车和电动车的骑行者是否佩戴头盔进行识别是亟需解决的问题。

传统的头盔识别方式需要先识别骑行人员的头肩区域，然后对该头肩区域进行头盔识别。

然而，上述方式需要先识别出头肩区域，再对头肩区域进一步识别才能得到头盔识别结果，头盔识别的效率较低。

【发明内容】

本申请提供了一种头盔识别方法、装置、设备及存储介质，可以解决现有的头盔识别效率较低的问题。本申请提供如下技术方案：

第一方面，提供一种头盔识别方法，所述方法包括：

获取目标图像；

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

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

可选地，所述基于所述头盔区域的检测结果和所述头肩区域的检测结果进行头盔识别，包括：

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

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

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

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

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

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

可选地，所述基于所述头盔区域的检测结果和所述头肩区域的检测结果进行头盔识别，包括：

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

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

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

可选地，所述头盔识别网络基于中心点的检测网络CenterNet建立，所述头盔识别网络从前至后依次包括主干网络、分别与所述主干网络相连的所述第一网络分支和所述第二网络分支；

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