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

说明书

本申请提供了一种佩戴安全帽检测方法、设备以及计算机存储介质。该佩戴安全帽检测方法包括：获取摄像头采集的待检测图像；将待检测图像输入检测模型，获取待检测图像中的人头位置以及人头类型、人头置信度；将待检测图像输入分类模型，获取安全帽类型、安全帽置信度；将待检测图像输入关键点模型，获取安全帽关键点位置，以及利用安全帽关键点位置获取佩戴安全帽类型、佩戴安全帽置信度；利用人头类型、安全帽类型以及佩戴安全帽类型综合判断是否佩戴安全帽，以及利用人头置信度、安全帽置信度以及佩戴安全帽置信度计算判断结果的置信度。通过上述方式，本申请的佩戴安全帽检测方法融合了多种模型的判别结果，能够有效提升检测方法的准确率。

技术领域

本申请涉及计算机视觉技术领域，特别是涉及一种佩戴安全帽检测方法、设备以及计算机存储介质。

背景技术

安全性是工程发展中很重要的一部分，但楼宇建设或者工地生产中易有掉落物体的非安全作业区域，若工人安全不能保证，不仅会遭受人员伤亡和财产损失，正常的作业秩序亦无法维持。安全帽作为保护、防护的重要防范手段，各大相关企业均会在作业场地出入口设置闸机，用于识别进入场地的工人戴安全帽的情况。但是由于侥幸心理，有些工人认为安全帽戴着麻烦又不舒服，事故不会发生在自己身上，很多施工人员不带安全帽进入工地已经成为一种习惯。现在很多地方还是采取人力盯梢的方式，在闸机口设置人力，检查进出人员佩戴安全帽的情况，但巡查人员不可能全天候盯着，保证工人都戴上安全帽成为了工地监督管理的一大难题。这就需要借助科技手段来实现监督。

然而，目前对于工人是否佩戴好安全帽的检测手段单一且准确率无法保证，无法满足不同环境的安全帽检测需求。

发明内容

本申请提供了一种佩戴安全帽检测方法、设备以及计算机存储介质。

本申请提供了一种佩戴安全帽检测方法，所述佩戴安全帽检测方法包括：

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

将所述待检测图像输入检测模型，获取所述待检测图像中的人头位置以及人头类型、人头置信度；

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

将所述待检测图像输入关键点模型，获取安全帽关键点位置，以及利用所述安全帽关键点位置获取佩戴安全帽类型、佩戴安全帽置信度；

利用所述人头类型、所述安全帽类型以及所述佩戴安全帽类型综合判断是否佩戴所述安全帽，以及利用所述人头置信度、所述安全帽置信度以及所述佩戴安全帽置信度计算判断结果的置信度。

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

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

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

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

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

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

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

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