[发明专利]安全帽佩戴检测方法和装置、电子设备、计算机可读存储介质

权利要求书

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

对图像帧中的人体骨骼点进行检测，得到每组人体骨骼点的位置信息；

对所述图像帧中的安全帽进行检测，得到安全帽的位置信息；

通过匈牙利算法对每组人体骨骼点的位置信息与安全帽的位置信息进行匹配，得到人体骨骼点与安全帽的匹配对；

当所述匹配对符合预设条件时，则检测出符合预设条件的匹配对中的人员佩戴安全帽。

2.根据权利要求1所述的方法，其特征在于，所述通过匈牙利算法对每组人体骨骼点的位置信息与安全帽的位置信息进行匹配，得到人体骨骼点与安全帽的匹配对，包括：

对每组人体骨骼点的位置信息与安全帽的位置信息计算匹配值，所述每组人体骨骼点至少包括人体上鼻子、脖子对应的骨骼点；

将所述匹配值输入至匈牙利算法中计算出人体骨骼点与安全帽的匹配对。

3.根据权利要求2所述的方法，其特征在于，所述当所述匹配对符合预设条件时，则检测出符合预设条件的匹配对中的人员佩戴安全帽，包括：

判断所述匹配对的匹配值是否小于所述第一阈值；

若判断出所述匹配对的匹配值小于所述第一阈值，则检测出符合预设条件的匹配对中的人员佩戴安全帽。

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

当判断出所述匹配对的匹配值大于或等于所述第一阈值，则检测出符合预设条件的匹配对中的人员未佩戴安全帽。

5.根据权利要求1所述的方法，其特征在于，所述一组人体骨骼点至少包括人体上鼻子、脖子对应的骨骼点；所述当所述匹配对符合预设条件时，则检测出符合预设条件的匹配对中的人员佩戴安全帽，包括：

获取符合预设条件的匹配对，计算所述符合预设条件的匹配对中人体上脖子对应的骨骼点到鼻子对应的骨骼点的向量作为基准向量；

计算所述符合预设条件的匹配对中人体上鼻子对应的骨骼点到安全帽的中心点的向量；

将所述人体上鼻子对应的骨骼点到安全帽的中心点的向量分别与所述基准向量做点乘运算得到点乘运算结果，对所述点乘运算结果计算加权和；

当所述加权和小于第二阈值时，则检测出符合预设条件的匹配对中的人员佩戴安全帽。

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

当所述加权和大于或等于第二阈值时，则检测出符合预设条件的匹配对中的人员未佩戴安全帽。

7.根据权利要求4或6所述的方法，其特征在于，所述方法还包括：

获取未佩戴安全帽的人员的相关信息，所述相关信息包括位置信息、图像信息；

将所述未佩戴安全帽的人员的相关信息上传至服务器进行存储。

8.一种安全帽佩戴检测装置，其特征在于，包括：

人体骨骼点位置检测模块，用于对图像帧中的人体骨骼点进行检测，得到每组人体骨骼点的位置信息；

安全帽位置检测模块，用于对所述图像帧中的安全帽进行检测，得到安全帽的位置信息；

匹配对获取模块，用于通过匈牙利算法对每组人体骨骼点的位置信息与安全帽的位置信息进行匹配，得到人体骨骼点与安全帽的匹配对；

佩戴安全帽确定模块，用于当所述匹配对符合预设条件时，则检测出符合预设条件的匹配对中的人员佩戴安全帽。

9.一种电子设备，包括存储器及处理器，所述存储器中储存有计算机程序，其特征在于，所述计算机程序被所述处理器执行时，使得所述处理器执行如权利要求1至7中任一项所述的安全帽佩戴检测方法的步骤。

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