[发明专利]一种检测安全帽佩戴情况的方法、计算机设备及存储介质

权利要求书

1.一种检测安全帽佩戴情况的方法，其特征在于，包括：

获取预设区域的时序图像；

检测每帧图像中的头部目标，得到每帧图像包含的头部目标及头部目标的类别、类别置信度以及所在图像区域，其中，所述类别包括：佩戴安全帽的头部目标和未佩戴安全帽的头部目标；

对检测到的所有头部目标进行多目标跟踪，将多帧图像中的同一所述头部目标进行关联；

对于每帧图像中检测出的每个所述头部目标，提取每个所述头部目标所在图像区域的预设图像特征；

对于每帧图像包含的每个所述头部目标，根据每个所述头部目标在当前图像和所述当前图像的前序图像中检测得到的类别和类别置信度，以及在所述当前图像中提取的所述预设图像特征，使用随机森林分类器，确定每个所述头部目标对应于所述当前图像的类别。

2.根据权利要求1所述的方法，其特征在于，还包括：

对于每帧图像检测出的每个所述头部目标，确定每个所述头部目标在所述当前图像中所在图像区域与其在所述当前图像的前序图像中所在图像区域之间的直线距离；

其中，对于每帧图像包含的每个所述头部目标，根据每个所述头部目标在当前图像和所述当前图像的前序图像中检测得到的类别和类别置信度、在所述当前图像中提取的所述预设图像特征，以及所述直线距离，使用随机森林分类器，确定每个所述头部目标对应于所述当前图像的类别。

3.根据权利要求1或2所述的方法，其特征在于，还包括：对于每帧图像中新检测出的所述头部目标，在后序图像中使用所述随机森林分类器确定所述新检测出的所述头部目标对应于所述后序图像的类别。

4.根据权利要求1或2所述的方法，其特征在于，所述预设图像特征包括：图像局部纹理特征和/或颜色直方图特征。

5.根据权利要求4所述的方法，其特征在于，使用LBP算子描述所述图像局部纹理特征。

6.根据权利要求1所述的方法，其特征在于，获取预设区域的时序图像，包括：

获取摄像头采集的视频，其中，所述摄像头的视野覆盖预设区域；

按照预设条件从所述视频中抽取多帧图像，得到所述预设区域的时序图像。

7.根据权利要求1至6中任一项所述的方法，其特征在于，还包括：检测每帧图像中的人体目标；根据每帧图像中头部目标的数量和人体目标的数量进行安全帽佩戴情况告警。

8.根据权利要求1至6中任一项所述的方法，其特征在于，还包括：

检测每帧图像中的人体，得到每帧图像包含的人体目标；

对于每帧图像，如果确定为未佩戴安全帽的头部目标的数量为0，且确定为佩戴安全帽的头部目标的数量不等于检测到的人体目标的数量，标记所述图像为异常帧；如果确定为未佩戴安全帽的头部目标的数量大于0，标记所述图像为告警帧；

根据连续多帧图像被标记为异常帧和/或告警帧的情况进行告警。

9.一种计算机设备，其特征在于，所述计算机设备包括：

存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序；

所述计算机程序被所述处理器执行时实现如权利要求1至8中任一项所述检测安全帽佩戴情况的方法的步骤。

10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质上存储有检测安全帽佩戴情况的程序，所述检测安全帽佩戴情况的程序被处理器执行时实现如权利要求1至8中任一项所述的检测安全帽佩戴情况的方法的步骤。