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

权利要求书

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

获取摄像头采集的待检测图像；

将所述待检测图像输入检测模型，获取所述待检测图像中的人头位置以及人头类型、人头置信度；其中，所述人头类型的值与所述检测模型检测到所述人头位置是否有遮挡相关；

将所述待检测图像输入分类模型，获取安全帽类型、安全帽置信度；

将所述待检测图像输入关键点模型，获取安全帽关键点，以及利用获取的各个安全帽关键点中任意一个安全帽关键点出现在所述待检测图像的左上角位置或者所述人头位置的左上角位置的情况，获取佩戴安全帽类型、佩戴安全帽置信度；利用所述人头类型、所述安全帽类型以及所述佩戴安全帽类型综合判断是否佩戴所述安全帽，以及利用所述人头置信度、所述安全帽置信度以及所述佩戴安全帽置信度计算判断结果的置信度；

所述利用所述安全帽关键点位置获取佩戴安全帽类型、佩戴安全帽置信度，包括：

判断多个安全帽关键点中是否存在任意一个安全帽关键点出现在所述待检测图像的左上角位置或者所述人头位置的左上角位置；

若是，则判定所述佩戴安全帽类型为无戴安全帽；

若否，则判定所述佩戴安全帽类型为有戴安全帽；

所述多个安全帽关键点包括第一安全帽关键点、第二安全帽关键点、第三安全帽关键点、第四安全帽关键点以及第五安全帽关键点；

所述判定所述佩戴安全帽类型为有戴安全帽，包括：

利用所述第一安全帽关键点和所述第三安全帽关键点建立第一直线表达式，以及利用所述第一安全帽关键点和所述第五安全帽关键点建立第二直线表达式；

将所述第二安全帽关键点的横坐标输入所述第一直线表达式，获取第一纵坐标，以及将所述第四安全帽关键点的横坐标输入所述第二直线表达式，获取第二纵坐标；

判断是否同时满足如下条件：所述第一纵坐标大于所述第二安全帽关键点的纵坐标，所述第二纵坐标大于所述第四安全帽关键点的纵坐标，第一直线的斜率以及第二直线的斜率均处于预设范围内；

若是，则判定所述佩戴安全帽类型为有戴安全帽；

若否，则判定所述佩戴安全帽类型为无戴安全帽。

2.根据权利要求1所述的佩戴安全帽检测方法，其特征在于，

所述获取所述待检测图像中的人头位置以及人头类型、人头置信度之后，包括：

利用所述检测模型检测所述待检测图像中的人手位置；

根据所述人手位置和所述人头位置的重叠区域获取人手类型；

输出所述人手位置、所述人手类型以及人手置信度。

3.根据权利要求2所述的佩戴安全帽检测方法，其特征在于，

所述利用所述人头类型、所述安全帽类型以及所述佩戴安全帽类型综合判断是否佩戴所述安全帽，包括：

当所述人头类型为有人头，所述安全帽类型为有安全帽，所述佩戴安全帽类型为有戴安全帽，且所述人手类型为无人手时，输出有佩戴安全帽的检测结果；

当所述人头类型为有人头，所述安全帽类型为有安全帽，所述佩戴安全帽类型为有戴安全帽，所述人手类型为有人手，且综合置信度大于等于预设阈值时，输出有佩戴安全帽的检测结果；

当所述人头类型为有人头，所述安全帽类型为有安全帽，所述佩戴安全帽类型为有戴安全帽，所述人手类型为有人手，且所述综合置信度小于所述预设阈值时，输出无佩戴安全帽的检测结果；

当不同时满足如下条件：所述人头类型为有人头，所述安全帽类型为有安全帽，所述佩戴安全帽类型为有戴安全帽时，输出无佩戴安全帽的检测结果；

其中，所述综合置信度通过所述人头置信度、所述安全帽置信度以及所述佩戴安全帽置信度按照预设数据公式计算得到。

4.根据权利要求3所述的佩戴安全帽检测方法，其特征在于，

所述获取安全帽类型、安全帽置信度，包括：

获取所述分类模型输出的安全帽类型、安全帽置信度、安全帽颜色类型以及安全帽颜色置信度；

所述利用所述人头类型、所述安全帽类型以及所述佩戴安全帽类型综合判断是否佩戴所述安全帽，包括：

判断所述安全帽颜色类型是否为其他；

若是，直接输出无佩戴安全帽的检测结果；

若否，利用所述人头类型、所述安全帽类型以及所述佩戴安全帽类型综合判断是否佩戴所述安全帽。