[发明专利]施工现场的安全帽佩戴检测方法、装置、设备及存储介质

权利要求书

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

根据施工现场图像样本以及与所述施工现场图像样本对应的标注信息训练安全帽佩戴检测模型；

将待检测图像输入所述安全帽佩戴检测模型，得到所述待检测图像中的人体头部位置信息以及与所述人体头部位置信息对应的安全帽类别信息；

输出所述人体头部位置信息和所述安全帽类别信息；

其中，所述标注信息包括所述施工现场图像样本上的人体头部位置标注以及与所述人体头部位置标注对应的安全帽类别标注，安全帽类别具体为佩戴安全帽类别或未佩戴安全帽类别；

所述根据施工现场图像样本以及与所述施工现场图像样本对应的标注信息训练安全帽佩戴检测模型，具体包括：

获取所述施工现场图像样本和所述标注信息；

将所述施工现场图像样本和所述标注信息输入SSD检测网络，训练所述SSD检测模型检测人体头部位置的能力以及识别所述人体头部位置处的安全帽类别的能力；

以训练好的所述SSD检测网络为所述安全帽佩戴检测模型；

所述将所述施工现场图像样本输入SSD检测网络，训练所述SSD检测模型检测人体头部位置的能力以及识别所述人体头部位置处的安全帽类别的能力，具体包括：将所述施工现场图像样本输入所述SSD检测网络；应用所述SSD检测网络提取所述施工现场图像样本中的特征，并应用注意力机制根据高层特征图的特征生成位置注意力特征图，以所述位置注意力特征图监督低层特征图的学习；其中，所述特征包括人体头部特征和安全帽类别特征；

所述应用注意力机制根据高层特征图的特征生成位置注意力特征图，以所述位置注意力特征图监督低层特征图的学习，具体为：将所述SSD检测网络中高层网络的语义信息向低层网络逐层传递，使各层网络中的语义信息以递归形式逐阶段整合至预设低层网络，以通过所述预设低层网络实现对安全帽的检测。

2.根据权利要求1所述的安全帽佩戴检测方法，其特征在于，在所述将所述施工现场图像样本和所述标注信息输入SSD检测网络，训练所述SSD检测模型检测人体头部位置的能力以及识别所述人体头部位置处的安全帽类别的能力之前，还包括：

将所述施工现场图像样本划分为训练集和测试集；

相应的，所述将所述施工现场图像样本输入SSD检测网络，训练所述SSD检测模型检测人体头部位置的能力以及识别所述人体头部位置处的安全帽类别的能力，具体包括：

利用所述训练集训练并更新所述SSD检测模型中的权值；

利用所述测试集测试所述SSD检测模型的泛化能力。

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

当检测到与所述人体头部位置信息对应的安全帽类别为所述未佩戴安全帽类别时，向与所述人体头部位置信息对应的地址发送报警信号。

4.根据权利要求1所述的安全帽佩戴检测方法，其特征在于，所述佩戴安全帽类别具体为佩戴蓝色安全帽、佩戴红色安全帽、佩戴黄色安全帽和佩戴白色安全帽中的任意一种。